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_inbound(self): logging.info('*** on_inbound ***') listener = Listener(engine=self.engine) listener.DEFER_DURATION = 0.0 with self.assertRaises(AssertionError): listener.on_inbound(my_message) with self.assertRaises(AssertionError): listener.on_inbound(my_private_message) with self.assertRaises(AssertionError): listener.on_inbound(my_join) with self.assertRaises(AssertionError): listener.on_inbound(my_leave) listener.on_inbound(my_event) with mock.patch.object(self.engine, 'dispatch', return_value=None) as mocked: listener.on_inbound(my_event) self.assertTrue(mocked.called)