50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
from aiohttp_pydantic.injectors import _parse_func_signature
|
|
from pydantic import BaseModel
|
|
from uuid import UUID
|
|
|
|
|
|
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})
|
|
|