def test_tg_get_receiver_coroutine_function_error(self):
mngr = TGManager("token")
async def request(self, method, **kwargs):
if method == "getMe":
return TGResponse(True, (), {})
mngr.request = types.MethodType(request, mngr)
with self.assertRaises(ValueError):
self.loop.run_until_complete(
mngr.get_receiver_coroutine_function())
self.loop.run_until_complete(mngr.dispose())
def test_tg_get_receiver_coroutine_function(self):
mngr = TGManager("token")
async def request(self, method, **kwargs):
if method == "getMe":
return TGResponse(False, (), {
"first_name": "A",
"last_name": "B",
"username": "******"
})
mngr.request = types.MethodType(request, mngr)
receiver = self.loop.run_until_complete(
mngr.get_receiver_coroutine_function())
self.assertEqual(receiver, mngr.receiver)
self.loop.run_until_complete(mngr.dispose())