aiogram/tests/test_bot/test_api.py

import pytest

from aiogram.bot.api import check_token
from aiogram.utils.exceptions import ValidationError

VALID_TOKEN = "123456789:AABBCCDDEEFFaabbccddeeff-1234567890"
INVALID_TOKENS = [
    "123456789:AABBCCDDEEFFaabbccddeeff 123456789",  # space is exists
    "ABC:AABBCCDDEEFFaabbccddeeff123456789",  # left part is not digit
    ":AABBCCDDEEFFaabbccddeeff123456789",  # there is no left part
    "123456789:",  # there is no right part
    "ABC AABBCCDDEEFFaabbccddeeff123456789",  # there is no ':' separator
    None,  # is None
    12345678,  # is digit
    {},  # is dict
    [],  # is dict
]


@pytest.fixture(params=INVALID_TOKENS, name="invalid_token")
def invalid_token_fixture(request):
    return request.param


class TestCheckToken:
    @staticmethod
    def test_valid():
        if check_token(VALID_TOKEN) is not True:
            raise AssertionError

    @staticmethod
    def test_invalid_token(invalid_token):
        with pytest.raises(ValidationError):
            check_token(invalid_token)