add missing types for generic

aio-libs · Nov 27, 2024 · ff409c1 · ff409c1
1 parent 9415db5
commit ff409c1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_redis_storage.py b/tests/test_redis_storage.py
@@ -34,7 +34,7 @@ def create_app(
 
 
 async def make_cookie(
-    client: TestClient, redis: aioredis.Redis, data: dict[Any, Any]
+    client: TestClient[web.Request, web.Application], redis: aioredis.Redis, data: dict[Any, Any]
 ) -> None:
     session_data = {"session": data, "created": int(time.time())}
     value = json.dumps(session_data)
@@ -43,13 +43,13 @@ async def make_cookie(
     client.session.cookie_jar.update_cookies({"AIOHTTP_SESSION": key})
 
 
-async def make_cookie_with_bad_value(client: TestClient, redis: aioredis.Redis) -> None:
+async def make_cookie_with_bad_value(client: TestClient[web.Request, web.Application], redis: aioredis.Redis) -> None:
     key = uuid.uuid4().hex
     await redis.set("AIOHTTP_SESSION_" + key, "")
     client.session.cookie_jar.update_cookies({"AIOHTTP_SESSION": key})
 
 
-async def load_cookie(client: TestClient, redis: aioredis.Redis) -> Any:
+async def load_cookie(client: TestClient[web.Request, web.Application], redis: aioredis.Redis) -> Any:
     cookies = client.session.cookie_jar.filter_cookies(client.make_url("/"))
     key = cookies["AIOHTTP_SESSION"]
     value_bytes = await redis.get("AIOHTTP_SESSION_" + key.value)

diff --git a/tests/typedefs.py b/tests/typedefs.py
@@ -3,4 +3,4 @@
 from aiohttp import web
 from aiohttp.test_utils import TestClient
 
-AiohttpClient = Callable[[web.Application], Awaitable[TestClient]]
+AiohttpClient = Callable[[web.Application], Awaitable[TestClient[web.Request, web.Application]]]