def test_update_content(self, *mocks):
mock_agent = mocks[0]
mock_context = mocks[1]
mock_factory = mocks[2]
mock_get_profiler = mocks[3]
mock_get_profiled_consumer = mocks[4]
mock_task_status = mocks[5]
mock_uuid = mocks[6]
unit = {'type_id': 'xyz', 'unit_key': {}}
consumer = {'id': '1234'}
mock_consumer_manager = Mock()
mock_consumer_manager.get_consumer = Mock(return_value=consumer)
mock_factory.consumer_manager = Mock(
return_value=mock_consumer_manager)
mock_get_profiled_consumer.return_value = consumer
mock_profiler = Mock()
mock_profiler.update_units = Mock(return_value=[unit])
mock_get_profiler.return_value = (mock_profiler, {})
task_id = '2345'
mock_context.return_value = {}
mock_uuid.return_value = task_id
# test manager
options = {'a': 1}
agent_manager = AgentManager()
agent_manager.update_content(consumer['id'], [unit], options)
# validations
task_tags = [
tags.resource_tag(tags.RESOURCE_CONSUMER_TYPE, consumer['id']),
tags.action_tag(tags.ACTION_AGENT_UNIT_UPDATE)
]
mock_consumer_manager.get_consumer.assert_called_with(consumer['id'])
mock_context.assert_called_with(consumer,
task_id=task_id,
consumer_id=consumer['id'])
mock_task_status.assert_called_with(task_id=task_id,
worker_name='agent',
tags=task_tags)
mock_profiler.update_units.assert_called_with(consumer, [unit],
options, {}, ANY)
mock_agent.update.assert_called_with(mock_context.return_value, [unit],
options)
def test_update_content(self, *mocks):
mock_agent = mocks[0]
mock_context = mocks[1]
mock_factory = mocks[2]
mock_get_profiler = mocks[3]
mock_get_profiled_consumer = mocks[4]
mock_task_status = mocks[5]
mock_uuid = mocks[6]
unit = {'type_id': 'xyz', 'unit_key': {}}
consumer = {'id': '1234'}
mock_consumer_manager = Mock()
mock_consumer_manager.get_consumer = Mock(return_value=consumer)
mock_factory.consumer_manager = Mock(return_value=mock_consumer_manager)
mock_get_profiled_consumer.return_value = consumer
mock_profiler = Mock()
mock_profiler.update_units = Mock(return_value=[unit])
mock_get_profiler.return_value = (mock_profiler, {})
task_id = '2345'
mock_context.return_value = {}
mock_uuid.return_value = task_id
# test manager
options = {'a': 1}
agent_manager = AgentManager()
agent_manager.update_content(consumer['id'], [unit], options)
# validations
task_tags = [
tags.resource_tag(tags.RESOURCE_CONSUMER_TYPE, consumer['id']),
tags.action_tag(tags.ACTION_AGENT_UNIT_UPDATE)
]
mock_consumer_manager.get_consumer.assert_called_with(consumer['id'])
mock_context.assert_called_with(consumer, task_id=task_id, consumer_id=consumer['id'])
mock_task_status.assert_called_with(task_id=task_id, worker_name='agent', tags=task_tags)
mock_profiler.update_units.assert_called_with(consumer, [unit], options, {}, ANY)
mock_agent.update.assert_called_with(mock_context.return_value, [unit], options)
