add additional test for loading the related model after returning fastapi pk dummy model

This commit is contained in:
collerek
2020-11-09 16:03:22 +01:00
parent 3b3f0445f4
commit d06a88fb6f

View File

@ -1,3 +1,4 @@
import asyncio
from typing import List, Optional from typing import List, Optional
import databases import databases
@ -128,6 +129,11 @@ def test_all_endpoints():
response = client.get("/items/raw/") response = client.get("/items/raw/")
items = [Item(**item) for item in response.json()] items = [Item(**item) for item in response.json()]
assert items[0].name == "New name" assert items[0].name == "New name"
assert items[0].category.name is None
loop = asyncio.get_event_loop()
loop.run_until_complete(items[0].category.load())
assert items[0].category.name is not None
response = client.get(f"/items/{item.pk}") response = client.get(f"/items/{item.pk}")
new_item = Item(**response.json()) new_item = Item(**response.json())