1- from typing import Awaitable , Callable
2-
31from aiohttp import web
4- from aiohttp .test_utils import TestClient
2+ from aiohttp .pytest_plugin import AiohttpClient
53from aiohttp_remotes import ForwardedRelaxed , ForwardedStrict , setup as _setup
64
7- _Client = Callable [[web .Application ], Awaitable [TestClient ]]
8-
95
10- async def test_forwarded_relaxed_ok (aiohttp_client : _Client ) -> None :
6+ async def test_forwarded_relaxed_ok (aiohttp_client : AiohttpClient ) -> None :
117 async def handler (request : web .Request ) -> web .Response :
128 assert request .host == "example.com"
139 assert request .scheme == "https"
@@ -25,7 +21,7 @@ async def handler(request: web.Request) -> web.Response:
2521 assert resp .status == 200
2622
2723
28- async def test_forwarded_relaxed_no_for (aiohttp_client : _Client ) -> None :
24+ async def test_forwarded_relaxed_no_for (aiohttp_client : AiohttpClient ) -> None :
2925 async def handler (request : web .Request ) -> web .Response :
3026 assert request .host == "example.com"
3127 assert request .scheme == "https"
@@ -43,7 +39,7 @@ async def handler(request: web.Request) -> web.Response:
4339 assert resp .status == 200
4440
4541
46- async def test_forwarded_relaxed_no_proto (aiohttp_client : _Client ) -> None :
42+ async def test_forwarded_relaxed_no_proto (aiohttp_client : AiohttpClient ) -> None :
4743 async def handler (request : web .Request ) -> web .Response :
4844 assert request .host == "example.com"
4945 assert request .scheme == "http"
@@ -61,7 +57,7 @@ async def handler(request: web.Request) -> web.Response:
6157 assert resp .status == 200
6258
6359
64- async def test_forwarded_relaxed_no_host (aiohttp_client : _Client ) -> None :
60+ async def test_forwarded_relaxed_no_host (aiohttp_client : AiohttpClient ) -> None :
6561 async def handler (request : web .Request ) -> web .Response :
6662 url = cl .make_url ("/" )
6763 assert url .host is not None
@@ -82,7 +78,7 @@ async def handler(request: web.Request) -> web.Response:
8278 assert resp .status == 200
8379
8480
85- async def test_forwarded_relaxed_many_hosts (aiohttp_client : _Client ) -> None :
81+ async def test_forwarded_relaxed_many_hosts (aiohttp_client : AiohttpClient ) -> None :
8682 async def handler (request : web .Request ) -> web .Response :
8783 assert request .host == "example.com"
8884 assert request .scheme == "https"
@@ -102,7 +98,7 @@ async def handler(request: web.Request) -> web.Response:
10298 assert resp .status == 200
10399
104100
105- async def test_forwarded_strict_ok (aiohttp_client : _Client ) -> None :
101+ async def test_forwarded_strict_ok (aiohttp_client : AiohttpClient ) -> None :
106102 async def handler (request : web .Request ) -> web .Response :
107103 assert request .host == "example.com"
108104 assert request .scheme == "https"
@@ -120,7 +116,7 @@ async def handler(request: web.Request) -> web.Response:
120116 assert resp .status == 200
121117
122118
123- async def test_forwarded_strict_no_proto (aiohttp_client : _Client ) -> None :
119+ async def test_forwarded_strict_no_proto (aiohttp_client : AiohttpClient ) -> None :
124120 async def handler (request : web .Request ) -> web .Response :
125121 assert request .host == "example.com"
126122 assert request .scheme == "http"
@@ -137,7 +133,7 @@ async def handler(request: web.Request) -> web.Response:
137133 assert resp .status == 200
138134
139135
140- async def test_forwarded_strict_no_host (aiohttp_client : _Client ) -> None :
136+ async def test_forwarded_strict_no_host (aiohttp_client : AiohttpClient ) -> None :
141137 async def handler (request : web .Request ) -> web .Response :
142138 assert request .host .startswith ("127.0.0.1:" )
143139 assert request .scheme == "https"
@@ -154,7 +150,7 @@ async def handler(request: web.Request) -> web.Response:
154150 assert resp .status == 200
155151
156152
157- async def test_forwarded_strict_too_many_protos (aiohttp_client : _Client ) -> None :
153+ async def test_forwarded_strict_too_many_protos (aiohttp_client : AiohttpClient ) -> None :
158154 async def handler (request : web .Request ) -> web .Response :
159155 return web .Response ()
160156
@@ -169,7 +165,7 @@ async def handler(request: web.Request) -> web.Response:
169165 assert resp .status == 400
170166
171167
172- async def test_forwarded_strict_too_many_for (aiohttp_client : _Client ) -> None :
168+ async def test_forwarded_strict_too_many_for (aiohttp_client : AiohttpClient ) -> None :
173169 async def handler (request : web .Request ) -> web .Response :
174170 return web .Response ()
175171
@@ -181,7 +177,7 @@ async def handler(request: web.Request) -> web.Response:
181177 assert resp .status == 400
182178
183179
184- async def test_forwarded_strict_untrusted_ip (aiohttp_client : _Client ) -> None :
180+ async def test_forwarded_strict_untrusted_ip (aiohttp_client : AiohttpClient ) -> None :
185181 async def handler (request : web .Request ) -> web .Response :
186182 return web .Response ()
187183
@@ -193,7 +189,7 @@ async def handler(request: web.Request) -> web.Response:
193189 assert resp .status == 400
194190
195191
196- async def test_forwarded_strict_whitelist (aiohttp_client : _Client ) -> None :
192+ async def test_forwarded_strict_whitelist (aiohttp_client : AiohttpClient ) -> None :
197193 async def handler (request : web .Request ) -> web .Response :
198194 assert request .remote == "127.0.0.1"
199195 return web .Response ()
@@ -206,7 +202,7 @@ async def handler(request: web.Request) -> web.Response:
206202 assert resp .status == 200
207203
208204
209- async def test_forwarded_strict_no_for (aiohttp_client : _Client ) -> None :
205+ async def test_forwarded_strict_no_for (aiohttp_client : AiohttpClient ) -> None :
210206 async def handler (request : web .Request ) -> web .Response :
211207 return web .Response ()
212208
0 commit comments