async def test_allow_head(aiohttp_client, loop):
app = web.Application(loop=loop)
s = SwaggerDocs(app, "/docs")
async def handler(request, param_id: int):
"""
---
parameters:
- name: param_id
in: path
required: true
schema:
type: integer
responses:
'200':
description: OK.
"""
return web.json_response({"param_id": param_id})
s.add_get("/r1/{param_id}", handler)
s.add_get("/r2/{param_id}", handler, allow_head=False)
client = await aiohttp_client(app)
resp = await client.head("/r1/1")
assert resp.status == 200
resp = await client.head("/r2/1")
assert resp.status == 405
async def test_all_methods(aiohttp_client, loop):
app = web.Application(loop=loop)
s = SwaggerDocs(app, "/docs")
class View(web.View):
async def get(self):
"""
---
responses:
'200':
description: OK.
"""
return web.json_response()
async def handler(request):
"""
---
responses:
'200':
description: OK.
"""
return web.json_response()
s.add_get("/r", handler, allow_head=False),
s.add_head("/r", handler),
s.add_put("/r", handler),
s.add_patch("/r", handler),
s.add_post("/r", handler),
s.add_delete("/r", handler),
s.add_options("/r", handler),
s.add_view("/r2", View),
client = await aiohttp_client(app)
for method in (
hdrs.METH_GET,
hdrs.METH_HEAD,
hdrs.METH_POST,
hdrs.METH_PUT,
hdrs.METH_PATCH,
hdrs.METH_DELETE,
hdrs.METH_OPTIONS,
):
resp = await getattr(client, method.lower())("/r")
assert resp.status == 200
resp = await client.get("/r2")
assert resp.status == 200