Telethon/client/tests/client_test.py

import os
import random

from pytest import mark

from telethon import Client
from telethon import _tl as tl


@mark.api
@mark.net
async def test_ping_pong() -> None:
    api_id = os.getenv("TG_ID")
    api_hash = os.getenv("TG_HASH")
    assert api_id and api_id.isdigit()
    assert api_hash
    client = Client(None, int(api_id), api_hash)
    assert not client.connected
    await client.connect()
    assert client.connected

    ping_id = random.randrange(-(2**63), 2**63)
    pong = await client(tl.mtproto.functions.ping(ping_id=ping_id))
    assert isinstance(pong, tl.mtproto.types.Pong)
    assert pong.ping_id == ping_id

    await client.disconnect()