def test_install_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.install_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.install_content(consumer['id'], [unit], options)
# validations
task_tags = [
tags.resource_tag(tags.RESOURCE_CONSUMER_TYPE, consumer['id']),
tags.action_tag(tags.ACTION_AGENT_UNIT_INSTALL)
]
mock_consumer_manager.get_consumer.assert_called_with(consumer['id'])
mock_task_status.assert_called_with(task_id=task_id,
worker_name='agent',
tags=task_tags)
mock_context.assert_called_with(consumer,
task_id=task_id,
consumer_id=consumer['id'])
mock_profiler.install_units.assert_called_with(consumer, [unit],
options, {}, ANY)
mock_agent.install.assert_called_with(mock_context.return_value,
[unit], options)
mock_factory.consumer_history_manager(
).record_event.assert_called_with(consumer['id'],
'content_unit_installed',
{'units': [unit]})
def test_install_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.install_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.install_content(consumer['id'], [unit], options)
# validations
task_tags = [
tags.resource_tag(tags.RESOURCE_CONSUMER_TYPE, consumer['id']),
tags.action_tag(tags.ACTION_AGENT_UNIT_INSTALL)
]
mock_consumer_manager.get_consumer.assert_called_with(consumer['id'])
mock_task_status.assert_called_with(task_id=task_id, worker_name='agent', tags=task_tags)
mock_context.assert_called_with(consumer, task_id=task_id, consumer_id=consumer['id'])
mock_profiler.install_units.assert_called_with(consumer, [unit], options, {}, ANY)
mock_agent.install.assert_called_with(mock_context.return_value, [unit], options)
mock_factory.consumer_history_manager().record_event.assert_called_with(
consumer['id'], 'content_unit_installed', {'units': [unit]})
def test_install_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_manager = 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.install_units = Mock(return_value=[unit])
mock_get_profiler.return_value = (mock_profiler, {})
task_id = '2345'
mock_task = {'task_id': task_id}
mock_task_status_manager.create_task_status = Mock(return_value=mock_task)
mock_context.return_value = {}
mock_uuid.return_value = task_id
# test manager
options = {'a': 1}
agent_manager = AgentManager()
task = agent_manager.install_content(consumer['id'], [unit], options)
# validations
tags = [
resource_tag(dispatch_constants.RESOURCE_CONSUMER_TYPE, consumer['id']),
action_tag(ACTION_AGENT_UNIT_INSTALL)
]
self.assertEqual(task, mock_task)
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_manager.create_task_status.assert_called_with(task_id, 'agent', tags=tags)
mock_profiler.install_units.assert_called_with(consumer, [unit], options, {}, ANY)
mock_agent.install.assert_called_with(mock_context.return_value, [unit], options)
