aiogram/tests/test_utils/test_link.py

from typing import Any, Dict

import pytest

from aiogram.utils.link import BRANCH, create_telegram_link, create_tg_link, docs_url


class TestLink:
    @pytest.mark.parametrize(
        "base,params,result",
        [["user", dict(id=42), "tg://user?id=42"]],
    )
    def test_create_tg_link(self, base: str, params: Dict[str, Any], result: str):
        assert create_tg_link(base, **params) == result

    @pytest.mark.parametrize(
        "base,params,result",
        [
            ["username", dict(), "https://t.me/username"],
            ["username", dict(start="test"), "https://t.me/username?start=test"],
        ],
    )
    def test_create_telegram_link(self, base: str, params: Dict[str, Any], result: str):
        assert create_telegram_link(base, **params) == result

    def test_fragment(self):
        assert (
            docs_url("test.html", fragment_="test")
            == f"https://docs.aiogram.dev/en/{BRANCH}/test.html#test"
        )

    def test_docs(self):
        assert docs_url("test.html") == f"https://docs.aiogram.dev/en/{BRANCH}/test.html"