test_deep_linking.py

import pytest

from aiogram.utils.deep_linking import (
    decode_payload,
    encode_payload,
    get_start_link,
    get_startgroup_link,
)
from tests.types import dataset


PAYLOADS = [
    'foo',
    'AAbbCCddEEff1122334455',
    'aaBBccDDeeFF5544332211',
    -12345678901234567890,
    12345678901234567890,
]

WRONG_PAYLOADS = [
    '@BotFather',
    "Some:special$characters#=",
    'spaces spaces spaces',
    1234567890123456789.0,
]
USERNAME = dataset.USER["username"]


@pytest.fixture(params=PAYLOADS, name='payload')
def payload_fixture(request):
    return request.param


@pytest.fixture(params=WRONG_PAYLOADS, name='wrong_payload')
def wrong_payload_fixture(request):
    return request.param


@pytest.fixture(autouse=True)
def get_bot_user_fixture(monkeypatch):
    """ Monkey patching of bot.me calling. """
    from aiogram.utils import deep_linking

    async def get_bot_user_mock():
        from aiogram.types import User
        return User(**dataset.USER)

    monkeypatch.setattr(deep_linking, '_get_bot_user', get_bot_user_mock)


class TestDeepLinking:
    async def test_get_start_link(self, payload):
        link = await get_start_link(payload)
        assert link == f'https://t.me/{USERNAME}?start={payload}'

    async def test_wrong_symbols(self, wrong_payload):
        with pytest.raises(ValueError):
            await get_start_link(wrong_payload)

    async def test_get_startgroup_link(self, payload):
        link = await get_startgroup_link(payload)
        assert link == f'https://t.me/{USERNAME}?startgroup={payload}'

    async def test_filter_encode_and_decode(self, payload):
        encoded = encode_payload(payload)
        decoded = decode_payload(encoded)
        assert decoded == str(payload)

    async def test_get_start_link_with_encoding(self, wrong_payload):
        # define link
        link = await get_start_link(wrong_payload, encode=True)

        # define reference link
        encoded_payload = encode_payload(wrong_payload)

        assert link == f'https://t.me/{USERNAME}?start={encoded_payload}'

    async def test_64_len_payload(self):
        payload = "p" * 64
        link = await get_start_link(payload)
        assert link

    async def test_too_long_payload(self):
        payload = "p" * 65
        print(payload, len(payload))
        with pytest.raises(ValueError):
            await get_start_link(payload)