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
66
67
68
69
70
71
72
|
import pytest
from lia import LitestarRequestAdapter
pytestmark = [pytest.mark.litestar]
@pytest.mark.asyncio
async def test_litestar_adapter() -> None:
from litestar import Litestar, post, Request
from litestar.testing import TestClient
adapter_result = None
@post("/test") # type: ignore[misc]
async def handler(request: Request) -> dict[str, str]:
nonlocal adapter_result
adapter_result = LitestarRequestAdapter(request)
return {"status": "ok"}
app = Litestar([handler])
with TestClient(app=app) as client:
client.cookies["session"] = "123"
client.post("/test?query=test", json={"key": "value"})
assert adapter_result is not None
assert dict(adapter_result.query_params) == {"query": "test"}
assert adapter_result.method == "POST"
assert adapter_result.headers["content-type"] == "application/json"
assert adapter_result.content_type == "application/json"
assert "test" in adapter_result.url
assert dict(adapter_result.cookies) == {"session": "123"}
@pytest.mark.asyncio
async def test_litestar_adapter_form_data() -> None:
from litestar import Litestar, post, Request
from litestar.testing import TestClient
adapter_result = None
@post("/test") # type: ignore[misc]
async def handler(request: Request) -> dict[str, str]:
nonlocal adapter_result
adapter_result = LitestarRequestAdapter(request)
return {"status": "ok"}
app = Litestar([handler])
with TestClient(app=app) as client:
client.cookies["session"] = "123"
client.post(
"/test?query=test",
data={"form": "data"},
files={"file": ("test.txt", b"upload")},
)
assert adapter_result is not None
assert dict(adapter_result.query_params) == {"query": "test"}
assert adapter_result.method == "POST"
assert adapter_result.headers[
"content-type"
] is not None and adapter_result.headers["content-type"].startswith(
"multipart/form-data"
)
assert (
adapter_result.content_type is not None
and adapter_result.content_type.startswith("multipart/form-data")
)
assert "test" in adapter_result.url
assert dict(adapter_result.cookies) == {"session": "123"}
|
