aiogram/tests/test_utils/test_auth_widget.py

import pytest

from aiogram.utils.auth_widget import (check_integrity, check_token,
                                       generate_hash)

TOKEN = "123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11"


@pytest.fixture
def data():
    return {
        "id": "42",
        "first_name": "John",
        "last_name": "Smith",
        "username": "username",
        "photo_url": "https://t.me/i/userpic/320/picname.jpg",
        "auth_date": "1565810688",
        "hash": "c303db2b5a06fe41d23a9b14f7c545cfc11dcc7473c07c9c5034ae60062461ce",
    }


def test_generate_hash(data):
    res = generate_hash(data, TOKEN)
    if res != data["hash"]:
        raise AssertionError


class Test_check_token:
    """
    This case gonna be deleted
    """

    @staticmethod
    def test_ok(data):
        if check_token(data, TOKEN) is not True:
            raise AssertionError

    @staticmethod
    def test_fail(data):
        data.pop("username")
        if check_token(data, TOKEN) is not False:
            raise AssertionError


class Test_check_integrity:
    @staticmethod
    def test_ok(data):
        if check_integrity(TOKEN, data) is not True:
            raise AssertionError

    @staticmethod
    def test_fail(data):
        data.pop("username")
        if check_integrity(TOKEN, data) is not False:
            raise AssertionError