def test_vk_update_longpoll_data_exception(self): mngr = VKManager("token") async def raw_request(self, method, **kwargs): return VKResponse(True, (), "") mngr.raw_request = types.MethodType(raw_request, mngr) with self.assertRaises(ValueError): self.loop.run_until_complete(mngr.update_longpoll_data()) self.loop.run_until_complete(mngr.dispose()) self.assertEqual(mngr.longpoll, None)
def test_vk_update_longpoll_data(self): mngr = VKManager("token") async def raw_request(self, method, **kwargs): return VKResponse(False, (), { "key": "key", "server": "server", "ts": "0" }) mngr.raw_request = types.MethodType(raw_request, mngr) self.loop.run_until_complete(mngr.update_longpoll_data()) self.loop.run_until_complete(mngr.dispose()) self.assertEqual(mngr.longpoll, { "key": "key", "server": "server", "ts": "0" })
def test_vk_get_receiver_coroutine_function(self): mngr = VKManager("token") async def raw_request(self, method, **kwargs): return VKResponse(False, (), [{"id": 1, "name": "keks"}]) mngr.raw_request = types.MethodType(raw_request, mngr) async def update_longpoll_data(self): self.longpoll = "updated" mngr.update_longpoll_data = types.MethodType(update_longpoll_data, mngr) receiver = self.loop.run_until_complete( mngr.get_receiver_coroutine_function()) self.assertEqual(mngr.longpoll, "updated") self.assertEqual(mngr.receiver, receiver) self.assertEqual(mngr.group_id, 1) self.loop.run_until_complete(mngr.dispose())