def test_agent_sources(self):
source_agent1 = TestAgent.new('Source Agent 1')
source_agent2 = TestAgent.new('Source Agent 2')
mute_agent = MuteAgent.new('Mute Agent')
agent = TestAgent.new('Test Agent',
source_agents=[source_agent1, source_agent2, mute_agent])
self.assertEqual(SourceAgent.get_source_agents(agent),
[source_agent1, source_agent2])
def test_agent_sources(self):
source_agent1 = TestAgent.new('Source Agent 1')
source_agent2 = TestAgent.new('Source Agent 2')
mute_agent = MuteAgent.new('Mute Agent')
agent = TestAgent.new(
'Test Agent',
source_agents=[source_agent1, source_agent2, mute_agent])
self.assertEqual(SourceAgent.get_source_agents(agent),
[source_agent1, source_agent2])
def test_agent_events(self):
source_agent = Agent.new('Source Agent')
listening_agent = Agent.new('Listening Agent',
source_agents=[source_agent])
listening_agent_2 = Agent.new('Listening Agent 2', source_agents=None)
source_agent.add_event({'event_field': 'event_value'})
source_agent._put_events_queue()
self.assertIn(listening_agent,
SourceAgent.get_listening_agents(source_agent))
events = listening_agent.receive_events()
self.assertEqual(len(events), 1)
events = listening_agent_2.receive_events()
self.assertEqual(len(events), 0)
def test_agent_events(self):
source_agent = Agent.new('Source Agent')
listening_agent = Agent.new('Listening Agent',
source_agents=[source_agent])
listening_agent_2 = Agent.new('Listening Agent 2',
source_agents=None)
source_agent.add_event({'event_field': 'event_value'})
source_agent._put_events_queue()
self.assertIn(listening_agent,
SourceAgent.get_listening_agents(source_agent))
events = listening_agent.receive_events()
self.assertEqual(len(events), 1)
events = listening_agent_2.receive_events()
self.assertEqual(len(events), 0)
