split tests into packages

tests/test_model_methods/test_save_related.py

@@ -0,0 +1,213 @@
+from typing import List
+
+import databases
+import pytest
+import sqlalchemy
+
+import ormar
+from tests.settings import DATABASE_URL
+
+database = databases.Database(DATABASE_URL, force_rollback=True)
+metadata = sqlalchemy.MetaData()
+
+
+class CringeLevel(ormar.Model):
+    class Meta:
+        tablename = "levels"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100)
+
+
+class NickName(ormar.Model):
+    class Meta:
+        tablename = "nicks"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100, nullable=False, name="hq_name")
+    is_lame: bool = ormar.Boolean(nullable=True)
+    level: CringeLevel = ormar.ForeignKey(CringeLevel)
+
+
+class NicksHq(ormar.Model):
+    class Meta:
+        tablename = "nicks_x_hq"
+        metadata = metadata
+        database = database
+
+
+class HQ(ormar.Model):
+    class Meta:
+        tablename = "hqs"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100, nullable=False, name="hq_name")
+    nicks: List[NickName] = ormar.ManyToMany(NickName, through=NicksHq)
+
+
+class Company(ormar.Model):
+    class Meta:
+        tablename = "companies"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100, nullable=False, name="company_name")
+    founded: int = ormar.Integer(nullable=True)
+    hq: HQ = ormar.ForeignKey(HQ, related_name="companies")
+
+
+@pytest.fixture(autouse=True, scope="module")
+def create_test_database():
+    engine = sqlalchemy.create_engine(DATABASE_URL)
+    metadata.drop_all(engine)
+    metadata.create_all(engine)
+    yield
+    metadata.drop_all(engine)
+
+
+@pytest.mark.asyncio
+async def test_saving_related_fk_rel():
+    async with database:
+        async with database.transaction(force_rollback=True):
+            hq = await HQ.objects.create(name="Main")
+            comp = await Company.objects.create(name="Banzai", founded=1988, hq=hq)
+            assert comp.saved
+
+            count = await comp.save_related()
+            assert count == 0
+
+            comp.hq.name = "Suburbs"
+            assert not comp.hq.saved
+            assert comp.saved
+
+            count = await comp.save_related()
+            assert count == 1
+            assert comp.hq.saved
+
+
+@pytest.mark.asyncio
+async def test_saving_many_to_many():
+    async with database:
+        async with database.transaction(force_rollback=True):
+            nick1 = await NickName.objects.create(name="BazingaO", is_lame=False)
+            nick2 = await NickName.objects.create(name="Bazinga20", is_lame=True)
+
+            hq = await HQ.objects.create(name="Main")
+            assert hq.saved
+
+            await hq.nicks.add(nick1)
+            assert hq.saved
+            await hq.nicks.add(nick2)
+            assert hq.saved
+
+            count = await hq.save_related()
+            assert count == 0
+
+            count = await hq.save_related(save_all=True)
+            assert count == 2
+
+            hq.nicks[0].name = "Kabucha"
+            hq.nicks[1].name = "Kabucha2"
+            assert not hq.nicks[0].saved
+            assert not hq.nicks[1].saved
+
+            count = await hq.save_related()
+            assert count == 2
+            assert hq.nicks[0].saved
+            assert hq.nicks[1].saved
+
+
+@pytest.mark.asyncio
+async def test_saving_reversed_relation():
+    async with database:
+        async with database.transaction(force_rollback=True):
+            hq = await HQ.objects.create(name="Main")
+            await Company.objects.create(name="Banzai", founded=1988, hq=hq)
+
+            hq = await HQ.objects.select_related("companies").get(name="Main")
+            assert hq.saved
+            assert hq.companies[0].saved
+
+            hq.companies[0].name = "Konichiwa"
+            assert not hq.companies[0].saved
+            count = await hq.save_related()
+            assert count == 1
+            assert hq.companies[0].saved
+
+            await Company.objects.create(name="Joshua", founded=1888, hq=hq)
+
+            hq = await HQ.objects.select_related("companies").get(name="Main")
+            assert hq.saved
+            assert hq.companies[0].saved
+            assert hq.companies[1].saved
+
+            hq.companies[0].name = hq.companies[0].name + "20"
+            assert not hq.companies[0].saved
+            # save only if not saved so now only one
+            count = await hq.save_related()
+            assert count == 1
+            assert hq.companies[0].saved
+
+            hq.companies[0].name = hq.companies[0].name + "20"
+            hq.companies[1].name = hq.companies[1].name + "30"
+            assert not hq.companies[0].saved
+            assert not hq.companies[1].saved
+            count = await hq.save_related()
+            assert count == 2
+            assert hq.companies[0].saved
+            assert hq.companies[1].saved
+
+
+@pytest.mark.asyncio
+async def test_saving_nested():
+    async with database:
+        async with database.transaction(force_rollback=True):
+            level = await CringeLevel.objects.create(name="High")
+            level2 = await CringeLevel.objects.create(name="Low")
+            nick1 = await NickName.objects.create(
+                name="BazingaO", is_lame=False, level=level
+            )
+            nick2 = await NickName.objects.create(
+                name="Bazinga20", is_lame=True, level=level2
+            )
+
+            hq = await HQ.objects.create(name="Main")
+            assert hq.saved
+
+            await hq.nicks.add(nick1)
+            assert hq.saved
+            await hq.nicks.add(nick2)
+            assert hq.saved
+
+            count = await hq.save_related()
+            assert count == 0
+
+            hq.nicks[0].level.name = "Medium"
+            assert not hq.nicks[0].level.saved
+            assert hq.nicks[0].saved
+
+            count = await hq.save_related(follow=True)
+            assert count == 1
+            assert hq.nicks[0].saved
+            assert hq.nicks[0].level.saved
+
+            hq.nicks[0].level.name = "Low"
+            hq.nicks[1].level.name = "Medium"
+            assert not hq.nicks[0].level.saved
+            assert not hq.nicks[1].level.saved
+            assert hq.nicks[0].saved
+            assert hq.nicks[1].saved
+
+            count = await hq.save_related(follow=True)
+            assert count == 2
+            assert hq.nicks[0].saved
+            assert hq.nicks[0].level.saved
+            assert hq.nicks[1].saved
+            assert hq.nicks[1].level.saved