add fastapi tests with inheritance and relations, more docstrings in queryset

tests/test_inheritance_concrete_fastapi.py

@@ -6,7 +6,17 @@ from fastapi import FastAPI
 from starlette.testclient import TestClient
 
 from tests.settings import DATABASE_URL
-from tests.test_inheritance_concrete import Category, Subject, metadata, db as database  # type: ignore
+from tests.test_inheritance_concrete import (  # type: ignore
+    Category,
+    Subject,
+    Person,
+    Bus,
+    Truck,
+    Bus2,
+    Truck2,
+    db as database,
+    metadata,
+)
 
 app = FastAPI()
 app.state.database = database
@@ -37,6 +47,56 @@ async def create_category(category: Category):
     return category
 
 
+@app.post("/buses/", response_model=Bus)
+async def create_bus(bus: Bus):
+    await bus.save()
+    return bus
+
+
+@app.get("/buses/{item_id}", response_model=Bus)
+async def get_bus(item_id: int):
+    bus = await Bus.objects.select_related(["owner", "co_owner"]).get(pk=item_id)
+    return bus
+
+
+@app.post("/trucks/", response_model=Truck)
+async def create_truck(truck: Truck):
+    await truck.save()
+    return truck
+
+
+@app.post("/persons/", response_model=Person)
+async def create_person(person: Person):
+    await person.save()
+    return person
+
+
+@app.post("/buses2/", response_model=Bus2)
+async def create_bus2(bus: Bus2):
+    await bus.save()
+    return bus
+
+
+@app.post("/buses2/{item_id}/add_coowner/", response_model=Bus2)
+async def add_bus_coowner(item_id: int, person: Person):
+    bus = await Bus2.objects.select_related(["owner", "co_owners"]).get(pk=item_id)
+    await bus.co_owners.add(person)
+    return bus
+
+
+@app.post("/trucks2/", response_model=Truck2)
+async def create_truck2(truck: Truck2):
+    await truck.save()
+    return truck
+
+
+@app.post("/trucks2/{item_id}/add_coowner/", response_model=Truck2)
+async def add_truck_coowner(item_id: int, person: Person):
+    truck = await Truck2.objects.select_related(["owner", "co_owners"]).get(pk=item_id)
+    await truck.co_owners.add(person)
+    return truck
+
+
 @pytest.fixture(autouse=True, scope="module")
 def create_test_database():
     engine = sqlalchemy.create_engine(DATABASE_URL)
@@ -73,3 +133,87 @@ def test_read_main():
         assert sub.name == "Bar"
         assert sub.category.pk == cat.pk
         assert isinstance(sub.updated_date, datetime.datetime)
+
+
+def test_inheritance_with_relation():
+    client = TestClient(app)
+    with client as client:
+        sam = Person(**client.post("/persons/", json={"name": "Sam"}).json())
+        joe = Person(**client.post("/persons/", json={"name": "Joe"}).json())
+
+        truck_dict = dict(
+            name="Shelby wanna be",
+            max_capacity=1400,
+            owner=sam.dict(),
+            co_owner=joe.dict(),
+        )
+        bus_dict = dict(
+            name="Unicorn", max_persons=50, owner=sam.dict(), co_owner=joe.dict()
+        )
+        unicorn = Bus(**client.post("/buses/", json=bus_dict).json())
+        shelby = Truck(**client.post("/trucks/", json=truck_dict).json())
+
+        assert shelby.name == "Shelby wanna be"
+        assert shelby.owner.name == "Sam"
+        assert shelby.co_owner.name == "Joe"
+        assert shelby.co_owner == joe
+        assert shelby.max_capacity == 1400
+
+        assert unicorn.name == "Unicorn"
+        assert unicorn.owner == sam
+        assert unicorn.owner.name == "Sam"
+        assert unicorn.co_owner.name == "Joe"
+        assert unicorn.max_persons == 50
+
+        unicorn2 = Bus(**client.get(f"/buses/{unicorn.pk}").json())
+        assert unicorn2.name == "Unicorn"
+        assert unicorn2.owner == sam
+        assert unicorn2.owner.name == "Sam"
+        assert unicorn2.co_owner.name == "Joe"
+        assert unicorn2.max_persons == 50
+
+
+def test_inheritance_with_m2m_relation():
+    client = TestClient(app)
+    with client as client:
+        sam = Person(**client.post("/persons/", json={"name": "Sam"}).json())
+        joe = Person(**client.post("/persons/", json={"name": "Joe"}).json())
+        alex = Person(**client.post("/persons/", json={"name": "Alex"}).json())
+
+        truck_dict = dict(name="Shelby wanna be", max_capacity=2000, owner=sam.dict())
+        bus_dict = dict(name="Unicorn", max_persons=80, owner=sam.dict())
+
+        unicorn = Bus2(**client.post("/buses2/", json=bus_dict).json())
+        shelby = Truck2(**client.post("/trucks2/", json=truck_dict).json())
+
+        unicorn = Bus2(
+            **client.post(f"/buses2/{unicorn.pk}/add_coowner/", json=joe.dict()).json()
+        )
+        unicorn = Bus2(
+            **client.post(f"/buses2/{unicorn.pk}/add_coowner/", json=alex.dict()).json()
+        )
+
+        assert shelby.name == "Shelby wanna be"
+        assert shelby.owner.name == "Sam"
+        assert len(shelby.co_owners) == 0
+        assert shelby.max_capacity == 2000
+
+        assert unicorn.name == "Unicorn"
+        assert unicorn.owner == sam
+        assert unicorn.owner.name == "Sam"
+        assert unicorn.co_owners[0].name == "Joe"
+        assert unicorn.co_owners[1] == alex
+        assert unicorn.max_persons == 80
+
+        client.post(f"/trucks2/{shelby.pk}/add_coowner/", json=alex.dict())
+
+        shelby = Truck2(
+            **client.post(f"/trucks2/{shelby.pk}/add_coowner/", json=joe.dict()).json()
+        )
+
+        assert shelby.name == "Shelby wanna be"
+        assert shelby.owner.name == "Sam"
+        assert len(shelby.co_owners) == 2
+        assert shelby.co_owners[0] == alex
+        assert shelby.co_owners[1] == joe
+        assert shelby.max_capacity == 2000