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