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)