60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from pydantic import BaseModel
|
|
from typing import Optional
|
|
from aiohttp import web
|
|
from aiohttp_pydantic import PydanticView
|
|
|
|
|
|
class ArticleModel(BaseModel):
|
|
name: str
|
|
nb_page: Optional[int]
|
|
|
|
|
|
class ArticleView(PydanticView):
|
|
async def post(self, article: ArticleModel):
|
|
return web.json_response(article.dict())
|
|
|
|
|
|
async def test_post_an_article_without_required_field_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.post("/article", json={})
|
|
assert resp.status == 400
|
|
assert resp.content_type == "application/json"
|
|
assert await resp.json() == [
|
|
{"loc": ["name"], "msg": "field required", "type": "value_error.missing"}
|
|
]
|
|
|
|
|
|
async def test_post_an_article_with_wrong_type_field_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.post("/article", json={"name": "foo", "nb_page": "foo"})
|
|
assert resp.status == 400
|
|
assert resp.content_type == "application/json"
|
|
assert await resp.json() == [
|
|
{
|
|
"loc": ["nb_page"],
|
|
"msg": "value is not a valid integer",
|
|
"type": "type_error.integer",
|
|
}
|
|
]
|
|
|
|
|
|
async def test_post_a_valid_article_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.post("/article", json={"name": "foo", "nb_page": 3})
|
|
assert resp.status == 200
|
|
assert resp.content_type == "application/json"
|
|
assert await resp.json() == {"name": "foo", "nb_page": 3}
|