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
|
import pytest
import aiohttp
from aiohttp import web
from aiohttp.test_utils import TestClient as _TestClient
from aiohttp.test_utils import TestServer as _TestServer
@pytest.fixture
def app(loop):
async def handler(request):
return web.Response(body=b"OK")
app = web.Application(loop=loop)
app.router.add_route('*', '/', handler)
return app
async def test_server_context_manager(app, loop):
async with _TestServer(app) as server:
async with aiohttp.ClientSession(loop=loop) as client:
async with client.head(server.make_url('/')) as resp:
assert resp.status == 200
@pytest.mark.parametrize("method", [
"head", "get", "post", "options", "post", "put", "patch", "delete"
])
async def test_client_context_manager_response(method, app, loop):
async with _TestClient(app) as client:
async with getattr(client, method)('/') as resp:
assert resp.status == 200
if method != 'head':
text = await resp.text()
assert "OK" in text
|
