def test_unregistered(self, mock_agent, mock_context, mock_factory, mock_delete_queue):
url = 'test-url'
queue = 'test-queue'
consumer_id = 'abc'
consumer = {'id': consumer_id}
mock_consumer_manager = Mock()
mock_consumer_manager.get_consumer = Mock(return_value=consumer)
mock_factory.consumer_manager = Mock(return_value=mock_consumer_manager)
mock_context.return_value = Mock(url=url, address=queue)
# test manager
agent_manager = AgentManager()
agent_manager.unregister(consumer_id)
# validations
task_tags = [
tags.resource_tag(tags.ACTION_AGENT_QUEUE_DELETE, consumer['id'])
]
mock_context.assert_called_with(consumer)
mock_agent.unregister.assert_called_with(mock_context.return_value)
mock_delete_queue.apply_async.assert_called_once_with(
args=[url, queue, consumer_id], countdown=QUEUE_DELETE_DELAY, tags=task_tags)
def test_unregistered(self, mock_agent, mock_context, mock_factory, mock_delete_queue):
url = 'test-url'
queue = 'test-queue'
consumer_id = 'abc'
consumer = {'id': consumer_id}
mock_consumer_manager = Mock()
mock_consumer_manager.get_consumer = Mock(return_value=consumer)
mock_factory.consumer_manager = Mock(return_value=mock_consumer_manager)
mock_context.return_value = Mock(url=url, address=queue)
# test manager
agent_manager = AgentManager()
agent_manager.unregister(consumer_id)
# validations
task_tags = [
tags.resource_tag(tags.ACTION_AGENT_QUEUE_DELETE, consumer['id'])
]
mock_context.assert_called_with(consumer)
mock_agent.unregister.assert_called_with(mock_context.return_value)
mock_delete_queue.apply_async.assert_called_once_with(
args=[url, queue, consumer_id], countdown=QUEUE_DELETE_DELAY, tags=task_tags)
