def test_inject_strategy(self, managers): consumer_id = 'test-consumer' strategy = 'test-strategy' notes = {constants.STRATEGY_NOTE_KEY: strategy} manager = Mock() manager.get_consumer.return_value = {'notes': notes} managers.consumer_manager.return_value = manager # test options = {} NodeProfiler._inject_strategy(consumer_id, options) # validation manager.get_consumer.assert_called_once_with(consumer_id) self.assertEqual(options[constants.STRATEGY_KEYWORD], strategy)
def test_inject_strategy(self, managers): consumer_id = 'test-consumer' strategy = 'test-strategy' notes = {constants.STRATEGY_NOTE_KEY: strategy} manager = Mock() manager.get_consumer.return_value = {'notes': notes} managers.consumer_manager.return_value = manager # test options = {} NodeProfiler._inject_strategy(consumer_id, options) # validation manager.get_consumer.assert_called_once_with(consumer_id) self.assertEqual(options[constants.STRATEGY_KEYWORD], strategy)
def test_update_units(self): consumer = Mock(id='123') units = [1, 2, 3] options = {} config = Mock() conduit = Mock() # test profiler = NodeProfiler() profiler._inject_parent_settings = Mock() profiler._inject_strategy = Mock() # validation _units = profiler.update_units(consumer, units, options, config, conduit) profiler._inject_parent_settings.assert_called_once_with(options) profiler._inject_strategy.assert_called_once_with(consumer.id, options) self.assertEqual(_units, units)
def test_update_units(self): consumer = Mock(id='123') units = [1, 2, 3] options = {} config = Mock() conduit = Mock() # test profiler = NodeProfiler() profiler._inject_parent_settings = Mock() profiler._inject_strategy = Mock() # validation _units = profiler.update_units(consumer, units, options, config, conduit) profiler._inject_parent_settings.assert_called_once_with(options) profiler._inject_strategy.assert_called_once_with(consumer.id, options) self.assertEqual(_units, units)