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
|
from typing import Tuple
import pytest
from aiohttp import web
from aiohttp.test_utils import make_mocked_request
from aiohttp.typedefs import Handler
from aiohttp_session import (
SESSION_KEY,
SimpleCookieStorage,
get_session,
session_middleware,
)
from .typedefs import AiohttpClient
def create_app(*handlers: Tuple[str, Handler]) -> web.Application:
middleware = session_middleware(SimpleCookieStorage())
app = web.Application(middlewares=[middleware])
for url, handler in handlers:
app.router.add_route("GET", url, handler)
return app
async def test_stream_response(aiohttp_client: AiohttpClient) -> None:
async def stream_response(request: web.Request) -> web.StreamResponse:
session = await get_session(request)
session["will_not"] = "show up"
return web.StreamResponse()
client = await aiohttp_client(create_app(("/stream", stream_response)))
resp = await client.get("/stream")
assert resp.status == 200
assert SESSION_KEY.upper() not in resp.cookies
async def test_bad_response_type(aiohttp_client: AiohttpClient) -> None:
async def bad_response(request: web.Request): # type: ignore[no-untyped-def]
return ""
middleware = session_middleware(SimpleCookieStorage())
req = make_mocked_request("GET", "/")
with pytest.raises(RuntimeError):
await middleware(req, bad_response)
async def test_prepared_response_type(aiohttp_client: AiohttpClient) -> None:
async def prepared_response(request: web.Request) -> web.StreamResponse:
resp = web.Response()
await resp.prepare(request)
return resp
middleware = session_middleware(SimpleCookieStorage())
req = make_mocked_request("GET", "/")
with pytest.raises(RuntimeError):
await middleware(req, prepared_response)
|
