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)