46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
from aiohttp import web
|
|
from aiohttp_pydantic import PydanticView
|
|
|
|
|
|
class ArticleView(PydanticView):
|
|
|
|
async def get(self, with_comments: bool):
|
|
return web.json_response({'with_comments': with_comments})
|
|
|
|
|
|
async def test_get_article_without_required_qs_should_return_an_error_message(aiohttp_client, loop):
|
|
app = web.Application()
|
|
app.router.add_view('/article', ArticleView)
|
|
|
|
client = await aiohttp_client(app)
|
|
resp = await client.get('/article')
|
|
assert resp.status == 400
|
|
assert resp.content_type == 'application/json'
|
|
assert await resp.json() == [{'loc': ['with_comments'],
|
|
'msg': 'field required',
|
|
'type': 'value_error.missing'}]
|
|
|
|
|
|
async def test_get_article_with_wrong_qs_type_should_return_an_error_message(aiohttp_client, loop):
|
|
app = web.Application()
|
|
app.router.add_view('/article', ArticleView)
|
|
|
|
client = await aiohttp_client(app)
|
|
resp = await client.get('/article', params={'with_comments': 'foo'})
|
|
assert resp.status == 400
|
|
assert resp.content_type == 'application/json'
|
|
assert await resp.json() == [{'loc': ['with_comments'],
|
|
'msg': 'value could not be parsed to a boolean',
|
|
'type': 'type_error.bool'}]
|
|
|
|
|
|
async def test_get_article_with_valid_qs_should_return_the_parsed_type(aiohttp_client, loop):
|
|
app = web.Application()
|
|
app.router.add_view('/article', ArticleView)
|
|
|
|
client = await aiohttp_client(app)
|
|
resp = await client.get('/article', params={'with_comments': 'yes'})
|
|
assert resp.status == 200
|
|
assert resp.content_type == 'application/json'
|
|
assert await resp.json() == {'with_comments': True}
|