test_bot.py 文件源码

def test_wait_event(self):
        bot = AB.Bot()
        test_event = {'unittest': True}
        waiter = bot.wait_for_event(**test_event)
        bot.event_to_chat = mock_tornado()
        bot._get_next_event = mock_tornado(
            side_effect=[gen.maybe_future(test_event), TestException])
        try:
            yield bot.start()
        except TestException:
            pass

        event = yield waiter
        self.assertEquals(event, test_event)