1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
from typing import Optional, Type
import pytest
from litestar import Controller, HttpMethod, Litestar, Request, Router, get
from litestar.handlers.http_handlers.base import HTTPRouteHandler
RouterRequest: Type[Request] = type("RouterRequest", (Request,), {})
ControllerRequest: Type[Request] = type("ControllerRequest", (Request,), {})
AppRequest: Type[Request] = type("AppRequest", (Request,), {})
HandlerRequest: Type[Request] = type("HandlerRequest", (Request,), {})
@pytest.mark.parametrize(
"handler_request_class, controller_request_class, router_request_class, app_request_class, has_default_app_class, expected",
(
(HandlerRequest, ControllerRequest, RouterRequest, AppRequest, True, HandlerRequest),
(None, ControllerRequest, RouterRequest, AppRequest, True, ControllerRequest),
(None, None, RouterRequest, AppRequest, True, RouterRequest),
(None, None, None, AppRequest, True, AppRequest),
(None, None, None, None, True, Request),
(None, None, None, None, False, Request),
),
ids=(
"Custom class for all layers",
"Custom class for all above handler layer",
"Custom class for all above controller layer",
"Custom class for all above router layer",
"No custom class for layers",
"No default class in app",
),
)
def test_request_class_resolution_of_layers(
handler_request_class: Optional[Type[Request]],
controller_request_class: Optional[Type[Request]],
router_request_class: Optional[Type[Request]],
app_request_class: Optional[Type[Request]],
has_default_app_class: bool,
expected: Type[Request],
) -> None:
class MyController(Controller):
@get()
def handler(self, request: Request) -> None:
assert type(request) is expected
if controller_request_class:
MyController.request_class = ControllerRequest
router = Router(path="/", route_handlers=[MyController])
if router_request_class:
router.request_class = router_request_class
app = Litestar(route_handlers=[router])
if app_request_class or not has_default_app_class:
app.request_class = app_request_class # type: ignore[assignment]
route_handler: HTTPRouteHandler = app.route_handler_method_map["/"][HttpMethod.GET] # type: ignore[assignment]
if handler_request_class:
route_handler.request_class = handler_request_class
request_class = route_handler.resolve_request_class()
assert request_class is expected
|
