check modyfing annotations hack for mypy

This commit is contained in:
collerek
2021-05-25 13:33:52 +02:00
parent b643c884ac
commit 88a49e67dd

View File

@ -43,16 +43,21 @@ def create_test_database():
metadata.drop_all(engine) metadata.drop_all(engine)
@app.post("/categories/", response_model=Category) async def create_category(category: Category):
async def create_category(category: Category.get_pydantic(exclude={"id"})): # type: ignore
return await Category(**category.dict()).save() return await Category(**category.dict()).save()
create_category.__annotations__["category"] = Category.get_pydantic(exclude={"id"})
app.post("/categories/", response_model=Category)(create_category)
@app.post( @app.post(
"/selfrefs/", "/selfrefs/",
response_model=SelfRef.get_pydantic(exclude={"parent", "children__name"}), response_model=SelfRef.get_pydantic(exclude={"parent", "children__name"}),
) )
async def create_selfref(selfref: SelfRef.get_pydantic(exclude={"children__name"})): # type: ignore async def create_selfref(
selfref: SelfRef.get_pydantic(exclude={"children__name"}), # type: ignore
):
selfr = SelfRef(**selfref.dict()) selfr = SelfRef(**selfref.dict())
await selfr.save() await selfr.save()
if selfr.children: if selfr.children:
@ -80,7 +85,7 @@ def test_read_main():
test_selfref = dict(id=1, name="test") test_selfref = dict(id=1, name="test")
test_selfref2 = dict(id=2, name="test2", parent={"id": 1}) test_selfref2 = dict(id=2, name="test2", parent={"id": 1})
test_selfref3 = dict(id=3, name="test3", children=[{"id": 1}]) test_selfref3 = dict(id=3, name="test3", children=[{"name": "aaa"}])
response = client.post("/selfrefs/", json=test_selfref) response = client.post("/selfrefs/", json=test_selfref)
assert response.status_code == 200 assert response.status_code == 200
@ -104,14 +109,24 @@ def test_read_main():
assert self_ref.id == 3 assert self_ref.id == 3
assert self_ref.name == "test3" assert self_ref.name == "test3"
assert self_ref.parent is None assert self_ref.parent is None
assert self_ref.children[0].dict() == {"id": 1} assert self_ref.children[0].dict() == {"id": 4}
response = client.get("/selfrefs/3/") response = client.get("/selfrefs/3/")
assert response.status_code == 200 assert response.status_code == 200
check_children = SelfRef(**response.json()) check_children = SelfRef(**response.json())
assert check_children.children[0].dict() == { assert check_children.children[0].dict() == {
"children": [], "children": [],
"id": 1, "id": 4,
"name": "test", "name": "selfref",
"parent": {"id": 3, "name": "test3"}, "parent": {"id": 3, "name": "test3"},
} }
response = client.get("/selfrefs/2/")
assert response.status_code == 200
check_children = SelfRef(**response.json())
assert check_children.dict() == {
"children": [],
"id": 2,
"name": "test2",
"parent": {"id": 1},
}