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)
    assert res == data["hash"]


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

    def test_ok(self, data):
        assert check_token(data, TOKEN) is True

    def test_fail(self, data):
        data.pop("username")
        assert check_token(data, TOKEN) is False


class Test_check_integrity:
    def test_ok(self, data):
        assert check_integrity(TOKEN, data) is True

    def test_fail(self, data):
        data.pop("username")
        assert check_integrity(TOKEN, data) is False