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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
from __future__ import annotations
import asyncio
import pytest
from quart import Blueprint
from quart import g
from quart import Quart
from quart import render_template_string
from quart import Response
from quart import ResponseReturnValue
from quart import session
from quart import stream_template_string
@pytest.fixture(scope="function")
def app() -> Quart:
app = Quart(__name__)
app.secret_key = "secret"
return app
@pytest.fixture(scope="function")
def blueprint() -> Blueprint:
blueprint = Blueprint("blueprint", __name__)
@blueprint.route("/")
def index() -> str:
return ""
return blueprint
async def test_template_render(app: Quart) -> None:
async with app.app_context():
rendered = await render_template_string("{{ foo }}", foo="bar")
assert rendered == "bar"
async def test_default_template_context(app: Quart) -> None:
async with app.app_context():
g.foo = "bar"
rendered = await render_template_string("{{ g.foo }}")
assert rendered == "bar"
async with app.test_request_context("/"):
session["foo"] = "bar"
rendered = await render_template_string(
"{{ request.method }} {{ request.path }} {{ session.foo }}"
)
assert rendered == "GET / bar"
async def test_template_context_processors(app: Quart, blueprint: Blueprint) -> None:
@blueprint.context_processor
async def blueprint_context() -> dict:
await asyncio.sleep(0.01) # Test the ability to await
return {"context": "foo"}
@blueprint.app_context_processor
async def app_blueprint_context() -> dict:
return {"global_context": "boo"}
@app.context_processor
async def app_context() -> dict:
return {"context": "bar"}
app.register_blueprint(blueprint)
async with app.app_context():
rendered = await render_template_string("{{ context }}")
assert rendered == "bar"
async with app.test_request_context("/"):
rendered = await render_template_string("{{ context }} {{ global_context }}")
assert rendered == "foo boo"
async with app.test_request_context("/other"):
rendered = await render_template_string("{{ context }} {{ global_context }}")
assert rendered == "bar boo"
async def test_template_globals(app: Quart, blueprint: Blueprint) -> None:
@blueprint.app_template_global()
def blueprint_global(value: str) -> str:
return value.upper()
@app.template_global()
def app_global(value: str) -> str:
return value.lower()
app.register_blueprint(blueprint)
async with app.app_context():
rendered = await render_template_string(
"{{ app_global('BAR') }} {{ blueprint_global('foo') }}"
)
assert rendered == "bar FOO"
async def test_template_filters(app: Quart, blueprint: Blueprint) -> None:
@blueprint.app_template_filter()
def blueprint_filter(value: str) -> str:
return value.upper()
@app.template_filter()
def app_filter(value: str) -> str:
return value.lower()
app.register_blueprint(blueprint)
async with app.app_context():
rendered = await render_template_string("{{ 'App' | app_filter }}")
assert rendered == "app"
async with app.test_request_context("/"):
rendered = await render_template_string("{{ 'App' | blueprint_filter }}")
assert rendered == "APP"
async def test_template_tests(app: Quart, blueprint: Blueprint) -> None:
@blueprint.app_template_test()
def blueprint_test(value: int) -> bool:
return value == 5
@app.template_test()
def app_test(value: int) -> bool:
return value == 3
app.register_blueprint(blueprint)
async with app.app_context():
rendered = await render_template_string("{% if 3 is app_test %}foo{% endif %}")
assert rendered == "foo"
async with app.test_request_context("/"):
rendered = await render_template_string(
"{% if 5 is blueprint_test %}bar{% endif %}"
)
assert rendered == "bar"
async def test_simple_stream(app: Quart) -> None:
@app.get("/")
async def index() -> ResponseReturnValue:
return await stream_template_string("{{ config }}", config=42) # type: ignore
test_client = app.test_client()
response = await test_client.get("/")
assert (await response.data) == b"42"
@app.get("/2")
async def index2() -> ResponseReturnValue:
return Response(await stream_template_string("{{ config }}", config=43))
test_client = app.test_client()
response = await test_client.get("/2")
assert (await response.data) == b"43"
|
