def test_unregistered(self, mock_agent, mock_context, mock_factory):
consumer = {'id': 'xyz'}
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 = {}
# test manager
agent_manager = AgentManager()
consumer_id = 'abc'
agent_manager.unregistered(consumer_id)
# validations
mock_context.assert_called_with(consumer)
mock_agent.unregistered.assert_called_with(mock_context.return_value)
def test_unregistered(self, mock_agent, mock_context, mock_factory):
consumer = {'id': 'xyz'}
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 = {}
# test manager
agent_manager = AgentManager()
consumer_id = 'abc'
agent_manager.unregistered(consumer_id)
# validations
mock_context.assert_called_with(consumer)
mock_agent.unregistered.assert_called_with(mock_context.return_value)
