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)
