from typing import Generic, TypeVar

import pytest

from apischema.typing import Annotated, generic_mro, resolve_type_hints

T = TypeVar("T")
U = TypeVar("U")


class A(Generic[T, U]):
    t: T
    u: U


class B(A[int, T]):
    v: T


class C(B[str]):
    pass


class D(C):
    d: Annotated[int, ""]


test_cases = [
    (A, [A], {"t": T, "u": U}),
    (A[int, str], [A[int, str]], {"t": int, "u": str}),
    (A[int, T], [A[int, T]], {"t": int, "u": T}),  # type: ignore
    (B, [B, A[int, T]], {"t": int, "u": T, "v": T}),  # type: ignore
    (B[U], [B[U], A[int, U]], {"t": int, "u": U, "v": U}),  # type: ignore
    (B[str], [B[str], A[int, str]], {"t": int, "u": str, "v": str}),
    (C, [C, B[str], A[int, str]], {"t": int, "u": str, "v": str}),
    (
        D,
        [D, C, B[str], A[int, str]],
        {"t": int, "u": str, "v": str, "d": Annotated[int, ""]},
    ),
]


@pytest.mark.parametrize("tp, result, _", test_cases)
def test_generic_mro(tp, result, _):
    assert generic_mro(tp) == (*result, Generic, object)


@pytest.mark.parametrize("tp, _, result", test_cases)
def test_resolve_type_hints(tp, _, result):
    assert resolve_type_hints(tp) == result