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)