21
21
22
22
def create_app (
23
23
handler : Handler ,
24
- redis : aioredis .Redis [ bytes ] ,
24
+ redis : aioredis .Redis ,
25
25
max_age : Optional [int ] = None ,
26
26
key_factory : Callable [[], str ] = lambda : uuid .uuid4 ().hex ,
27
27
) -> web .Application :
@@ -34,7 +34,7 @@ def create_app(
34
34
35
35
36
36
async def make_cookie (
37
- client : TestClient , redis : aioredis .Redis [ bytes ] , data : Dict [Any , Any ]
37
+ client : TestClient , redis : aioredis .Redis , data : Dict [Any , Any ]
38
38
) -> None :
39
39
session_data = {"session" : data , "created" : int (time .time ())}
40
40
value = json .dumps (session_data )
@@ -43,21 +43,21 @@ async def make_cookie(
43
43
client .session .cookie_jar .update_cookies ({"AIOHTTP_SESSION" : key })
44
44
45
45
46
- async def make_cookie_with_bad_value (client : TestClient , redis : aioredis .Redis [ bytes ] ) -> None :
46
+ async def make_cookie_with_bad_value (client : TestClient , redis : aioredis .Redis ) -> None :
47
47
key = uuid .uuid4 ().hex
48
48
await redis .set ("AIOHTTP_SESSION_" + key , "" )
49
49
client .session .cookie_jar .update_cookies ({"AIOHTTP_SESSION" : key })
50
50
51
51
52
- async def load_cookie (client : TestClient , redis : aioredis .Redis [ bytes ] ) -> Any :
52
+ async def load_cookie (client : TestClient , redis : aioredis .Redis ) -> Any :
53
53
cookies = client .session .cookie_jar .filter_cookies (client .make_url ("/" ))
54
54
key = cookies ["AIOHTTP_SESSION" ]
55
55
value_bytes = await redis .get ("AIOHTTP_SESSION_" + key .value )
56
56
return None if value_bytes is None else json .loads (value_bytes )
57
57
58
58
59
59
async def test_create_new_session (
60
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
60
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
61
61
) -> None :
62
62
async def handler (request : web .Request ) -> web .StreamResponse :
63
63
session = await get_session (request )
@@ -73,7 +73,7 @@ async def handler(request: web.Request) -> web.StreamResponse:
73
73
74
74
75
75
async def test_load_existing_session (
76
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
76
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
77
77
) -> None :
78
78
async def handler (request : web .Request ) -> web .StreamResponse :
79
79
session = await get_session (request )
@@ -90,7 +90,7 @@ async def handler(request: web.Request) -> web.StreamResponse:
90
90
91
91
92
92
async def test_load_bad_session (
93
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
93
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
94
94
) -> None :
95
95
async def handler (request : web .Request ) -> web .StreamResponse :
96
96
session = await get_session (request )
@@ -107,7 +107,7 @@ async def handler(request: web.Request) -> web.StreamResponse:
107
107
108
108
109
109
async def test_change_session (
110
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
110
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
111
111
) -> None :
112
112
async def handler (request : web .Request ) -> web .StreamResponse :
113
113
session = await get_session (request )
@@ -134,7 +134,7 @@ async def handler(request: web.Request) -> web.StreamResponse:
134
134
135
135
136
136
async def test_clear_cookie_on_session_invalidation (
137
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
137
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
138
138
) -> None :
139
139
async def handler (request : web .Request ) -> web .StreamResponse :
140
140
session = await get_session (request )
@@ -155,7 +155,7 @@ async def handler(request: web.Request) -> web.StreamResponse:
155
155
156
156
157
157
async def test_create_cookie_in_handler (
158
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
158
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
159
159
) -> None :
160
160
async def handler (request : web .Request ) -> web .StreamResponse :
161
161
session = await get_session (request )
@@ -182,7 +182,7 @@ async def handler(request: web.Request) -> web.StreamResponse:
182
182
183
183
184
184
async def test_set_ttl_on_session_saving (
185
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
185
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
186
186
) -> None :
187
187
async def handler (request : web .Request ) -> web .StreamResponse :
188
188
session = await get_session (request )
@@ -202,7 +202,7 @@ async def handler(request: web.Request) -> web.StreamResponse:
202
202
203
203
204
204
async def test_set_ttl_manually_set (
205
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
205
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
206
206
) -> None :
207
207
async def handler (request : web .Request ) -> web .StreamResponse :
208
208
session = await get_session (request )
@@ -223,7 +223,7 @@ async def handler(request: web.Request) -> web.StreamResponse:
223
223
224
224
225
225
async def test_create_new_session_if_key_doesnt_exists_in_redis (
226
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
226
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
227
227
) -> None :
228
228
async def handler (request : web .Request ) -> web .StreamResponse :
229
229
session = await get_session (request )
@@ -237,7 +237,7 @@ async def handler(request: web.Request) -> web.StreamResponse:
237
237
238
238
239
239
async def test_create_storage_with_custom_key_factory (
240
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
240
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
241
241
) -> None :
242
242
async def handler (request : web .Request ) -> web .StreamResponse :
243
243
session = await get_session (request )
@@ -260,7 +260,7 @@ def key_factory() -> str:
260
260
261
261
262
262
async def test_redis_session_fixation (
263
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
263
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
264
264
) -> None :
265
265
async def login (request : web .Request ) -> web .StreamResponse :
266
266
session = await get_session (request )
@@ -289,9 +289,9 @@ async def test_redis_from_create_pool(redis_url: str) -> None:
289
289
async def handler (request : web .Request ) -> web .Response : # type: ignore[empty-body]
290
290
"""Dummy handler"""
291
291
292
- redis = aioredis .from_url (redis_url )
292
+ redis = aioredis .from_url (redis_url ) # type: ignore[no-untyped-call]
293
293
create_app (handler = handler , redis = redis )
294
- await redis .close ()
294
+ await redis .aclose ()
295
295
296
296
297
297
async def test_not_redis_provided_to_storage () -> None :
@@ -321,7 +321,7 @@ async def handler(request: web.Request) -> web.Response: # type: ignore[empty-b
321
321
322
322
323
323
async def test_load_session_dont_load_expired_session (
324
- aiohttp_client : AiohttpClient , redis : aioredis .Redis [ bytes ]
324
+ aiohttp_client : AiohttpClient , redis : aioredis .Redis
325
325
) -> None :
326
326
async def handler (request : web .Request ) -> web .StreamResponse :
327
327
session = await get_session (request )
0 commit comments