def test_on_message_fan(self): logging.info('*** on_message/fan ***') self.engine.set('bot.id', '*not*for*me') class MyFan(object): def __init__(self): self.called = False def put(self, arguments): self.called = True self.bot.fan = MyFan() listener = Listener(engine=self.engine) listener.DEFER_DURATION = 0.0 listener.on_message(my_message) self.assertFalse(self.bot.fan.called) label = 'fan.' + my_message.channel_id logging.debug(u"- stamping '{}'".format(label)) self.engine.set(label, time.time()) listener.on_message(my_message) self.assertTrue(self.bot.fan.called)
def test_process(self): logging.info('*** process ***') listener = Listener(engine=self.engine) listener.DEFER_DURATION = 0.0 self.engine.set('listener.counter', 22) with self.assertRaises(AssertionError): listener.process('hello world') self.assertEqual(self.engine.get('listener.counter'), 23) listener.on_message = mock.Mock() listener.process(str(my_message)) self.assertEqual(self.engine.get('listener.counter'), 24) self.assertTrue(listener.on_message.called) listener.on_message = mock.Mock() listener.process(str(my_private_message)) self.assertEqual(self.engine.get('listener.counter'), 25) self.assertTrue(listener.on_message.called) listener.on_join = mock.Mock() listener.process(str(my_join)) self.assertEqual(self.engine.get('listener.counter'), 26) self.assertTrue(listener.on_join.called) listener.on_leave = mock.Mock() listener.process(str(my_leave)) self.assertEqual(self.engine.get('listener.counter'), 27) self.assertTrue(listener.on_leave.called) listener.on_inbound = mock.Mock() listener.process(str(my_event)) self.assertEqual(self.engine.get('listener.counter'), 28) self.assertTrue(listener.on_inbound.called)
def test_on_message(self): logging.info('*** on_message ***') listener = Listener(engine=self.engine) listener.DEFER_DURATION = 0.0 listener.on_message(my_message) listener.on_message(my_private_message) with self.assertRaises(AssertionError): listener.on_message(my_join) with self.assertRaises(AssertionError): listener.on_message(my_leave) with self.assertRaises(AssertionError): listener.on_message(my_event) with mock.patch.object(self.engine, 'dispatch', return_value=None) as mocked: listener.on_message(my_message) self.assertTrue(mocked.called)