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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
import pytest
from litestar import Litestar
from litestar.config.csrf import CSRFConfig
from litestar.openapi.config import OpenAPIConfig
from litestar.openapi.plugins import RapidocRenderPlugin, ScalarRenderPlugin, SwaggerRenderPlugin
from litestar.testing import TestClient
rapidoc_fragment = ".addEventListener('before-try',"
swagger_fragment = "requestInterceptor:"
def test_rapidoc_csrf() -> None:
app = Litestar(
csrf_config=CSRFConfig(secret="litestar"),
openapi_config=OpenAPIConfig(
title="Litestar Example",
version="0.0.1",
render_plugins=[RapidocRenderPlugin()],
),
)
with TestClient(app=app) as client:
resp = client.get("/schema/rapidoc")
assert resp.status_code == 200
assert resp.headers["content-type"] == "text/html; charset=utf-8"
assert rapidoc_fragment in resp.text
def test_swagger_ui_csrf() -> None:
app = Litestar(
csrf_config=CSRFConfig(secret="litestar"),
openapi_config=OpenAPIConfig(
title="Litestar Example",
version="0.0.1",
render_plugins=[SwaggerRenderPlugin()],
),
)
with TestClient(app=app) as client:
resp = client.get("/schema/swagger")
assert resp.status_code == 200
assert resp.headers["content-type"] == "text/html; charset=utf-8"
assert swagger_fragment in resp.text
def test_plugins_csrf_httponly() -> None:
app = Litestar(
csrf_config=CSRFConfig(secret="litestar", cookie_httponly=True),
openapi_config=OpenAPIConfig(
title="Litestar Example",
version="0.0.1",
render_plugins=[RapidocRenderPlugin(), SwaggerRenderPlugin()],
),
)
with TestClient(app=app) as client:
resp = client.get("/schema/rapidoc")
assert resp.status_code == 200
assert rapidoc_fragment not in resp.text
resp = client.get("/schema/swagger")
assert resp.status_code == 200
assert swagger_fragment not in resp.text
@pytest.mark.parametrize(
"scalar_config",
[
{"showSidebar": False},
],
)
@pytest.mark.parametrize(
"expected_config_render",
[
"document.getElementById('api-reference').dataset.configuration = '{\"showSidebar\":false}'",
],
)
def test_openapi_scalar_options(scalar_config: dict, expected_config_render: str) -> None:
app = Litestar(
openapi_config=OpenAPIConfig(
title="Litestar Example",
version="0.0.1",
render_plugins=[ScalarRenderPlugin(options=scalar_config)],
)
)
with TestClient(app=app) as client:
resp = client.get("/schema/scalar")
assert resp.status_code == 200
assert expected_config_render in resp.text
|
