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())
