from __future__ import annotations

from uuid import UUID

from pydantic import BaseModel

from aiohttp_pydantic.injectors import _parse_func_signature


class User(BaseModel):
    firstname: str
    lastname: str


def test_parse_func_signature():
    def body_only(self, user: User):
        pass

    def path_only(self, id: str, /):
        pass

    def qs_only(self, page: int):
        pass

    def header_only(self, *, auth: UUID):
        pass

    def path_and_qs(self, id: str, /, page: int):
        pass

    def path_and_header(self, id: str, /, *, auth: UUID):
        pass

    def qs_and_header(self, page: int, *, auth: UUID):
        pass

    def path_qs_and_header(self, id: str, /, page: int, *, auth: UUID):
        pass

    def path_body_qs_and_header(self, id: str, /, user: User, page: int, *, auth: UUID):
        pass

    assert _parse_func_signature(body_only) == ({}, {"user": User}, {}, {}, {})
    assert _parse_func_signature(path_only) == ({"id": str}, {}, {}, {}, {})
    assert _parse_func_signature(qs_only) == ({}, {}, {"page": int}, {}, {})
    assert _parse_func_signature(header_only) == ({}, {}, {}, {"auth": UUID}, {})
    assert _parse_func_signature(path_and_qs) == (
        {"id": str},
        {},
        {"page": int},
        {},
        {},
    )
    assert _parse_func_signature(path_and_header) == (
        {"id": str},
        {},
        {},
        {"auth": UUID},
        {},
    )
    assert _parse_func_signature(qs_and_header) == (
        {},
        {},
        {"page": int},
        {"auth": UUID},
        {},
    )
    assert _parse_func_signature(path_qs_and_header) == (
        {"id": str},
        {},
        {"page": int},
        {"auth": UUID},
        {},
    )
    assert _parse_func_signature(path_body_qs_and_header) == (
        {"id": str},
        {"user": User},
        {"page": int},
        {"auth": UUID},
        {},
    )