split tests into packages

tests/test_model_methods/test_load_all.py

@@ -0,0 +1,251 @@
+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 BaseMeta(ormar.ModelMeta):
+    database = database
+    metadata = metadata
+
+
+class Language(ormar.Model):
+    class Meta(BaseMeta):
+        tablename = "languages"
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100)
+    level: str = ormar.String(max_length=150, default="Beginner")
+
+
+class CringeLevel(ormar.Model):
+    class Meta(BaseMeta):
+        tablename = "levels"
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100)
+    language = ormar.ForeignKey(Language)
+
+
+class NickName(ormar.Model):
+    class Meta(BaseMeta):
+        tablename = "nicks"
+
+    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 HQ(ormar.Model):
+    class Meta(BaseMeta):
+        tablename = "hqs"
+
+    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)
+
+
+class Company(ormar.Model):
+    class Meta(BaseMeta):
+        tablename = "companies"
+
+    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_load_all_fk_rel():
+    async with database:
+        async with database.transaction(force_rollback=True):
+            hq = await HQ.objects.create(name="Main")
+            company = await Company.objects.create(name="Banzai", founded=1988, hq=hq)
+
+            hq = await HQ.objects.get(name="Main")
+            await hq.load_all()
+
+            assert hq.companies[0] == company
+            assert hq.companies[0].name == "Banzai"
+            assert hq.companies[0].founded == 1988
+
+            hq2 = await HQ.objects.select_all().get(name="Main")
+            assert hq2.companies[0] == company
+            assert hq2.companies[0].name == "Banzai"
+            assert hq2.companies[0].founded == 1988
+
+
+@pytest.mark.asyncio
+async def test_load_all_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")
+            await hq.nicks.add(nick1)
+            await hq.nicks.add(nick2)
+
+            hq = await HQ.objects.get(name="Main")
+            await hq.load_all()
+
+            assert hq.nicks[0] == nick1
+            assert hq.nicks[0].name == "BazingaO"
+
+            assert hq.nicks[1] == nick2
+            assert hq.nicks[1].name == "Bazinga20"
+
+            hq2 = await HQ.objects.select_all().get(name="Main")
+            assert hq2.nicks[0] == nick1
+            assert hq2.nicks[0].name == "BazingaO"
+            assert hq2.nicks[1] == nick2
+            assert hq2.nicks[1].name == "Bazinga20"
+
+
+@pytest.mark.asyncio
+async def test_load_all_with_order():
+    async with database:
+        async with database.transaction(force_rollback=True):
+            nick1 = await NickName.objects.create(name="Barry", is_lame=False)
+            nick2 = await NickName.objects.create(name="Joe", is_lame=True)
+            hq = await HQ.objects.create(name="Main")
+            await hq.nicks.add(nick1)
+            await hq.nicks.add(nick2)
+
+            hq = await HQ.objects.get(name="Main")
+            await hq.load_all(order_by="-nicks__name")
+
+            assert hq.nicks[0] == nick2
+            assert hq.nicks[0].name == "Joe"
+
+            assert hq.nicks[1] == nick1
+            assert hq.nicks[1].name == "Barry"
+
+            await hq.load_all()
+            assert hq.nicks[0] == nick1
+            assert hq.nicks[1] == nick2
+
+            hq2 = (
+                await HQ.objects.select_all().order_by("-nicks__name").get(name="Main")
+            )
+            assert hq2.nicks[0] == nick2
+            assert hq2.nicks[1] == nick1
+
+            hq3 = await HQ.objects.select_all().get(name="Main")
+            assert hq3.nicks[0] == nick1
+            assert hq3.nicks[1] == nick2
+
+
+@pytest.mark.asyncio
+async def test_loading_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)
+
+            company = await Company.objects.get(name="Banzai")
+            await company.load_all()
+
+            assert company.hq == hq
+
+            company2 = await Company.objects.select_all().get(name="Banzai")
+            assert company2.hq == hq
+
+
+@pytest.mark.asyncio
+async def test_loading_nested():
+    async with database:
+        async with database.transaction(force_rollback=True):
+            language = await Language.objects.create(name="English")
+            level = await CringeLevel.objects.create(name="High", language=language)
+            level2 = await CringeLevel.objects.create(name="Low", language=language)
+            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")
+            await hq.nicks.add(nick1)
+            await hq.nicks.add(nick2)
+
+            hq = await HQ.objects.get(name="Main")
+            await hq.load_all(follow=True)
+
+            assert hq.nicks[0] == nick1
+            assert hq.nicks[0].name == "BazingaO"
+            assert hq.nicks[0].level.name == "High"
+            assert hq.nicks[0].level.language.name == "English"
+
+            assert hq.nicks[1] == nick2
+            assert hq.nicks[1].name == "Bazinga20"
+            assert hq.nicks[1].level.name == "Low"
+            assert hq.nicks[1].level.language.name == "English"
+
+            hq2 = await HQ.objects.select_all(follow=True).get(name="Main")
+            assert hq2.nicks[0] == nick1
+            assert hq2.nicks[0].name == "BazingaO"
+            assert hq2.nicks[0].level.name == "High"
+            assert hq2.nicks[0].level.language.name == "English"
+
+            assert hq2.nicks[1] == nick2
+            assert hq2.nicks[1].name == "Bazinga20"
+            assert hq2.nicks[1].level.name == "Low"
+            assert hq2.nicks[1].level.language.name == "English"
+
+            hq5 = await HQ.objects.select_all().get(name="Main")
+            assert len(hq5.nicks) == 2
+            await hq5.nicks.select_all(follow=True).all()
+            assert hq5.nicks[0] == nick1
+            assert hq5.nicks[0].name == "BazingaO"
+            assert hq5.nicks[0].level.name == "High"
+            assert hq5.nicks[0].level.language.name == "English"
+            assert hq5.nicks[1] == nick2
+            assert hq5.nicks[1].name == "Bazinga20"
+            assert hq5.nicks[1].level.name == "Low"
+            assert hq5.nicks[1].level.language.name == "English"
+
+            await hq.load_all(follow=True, exclude="nicks__level__language")
+            assert len(hq.nicks) == 2
+            assert hq.nicks[0].level.language is None
+            assert hq.nicks[1].level.language is None
+
+            hq3 = (
+                await HQ.objects.select_all(follow=True)
+                .exclude_fields("nicks__level__language")
+                .get(name="Main")
+            )
+            assert len(hq3.nicks) == 2
+            assert hq3.nicks[0].level.language is None
+            assert hq3.nicks[1].level.language is None
+
+            await hq.load_all(follow=True, exclude="nicks__level__language__level")
+            assert len(hq.nicks) == 2
+            assert hq.nicks[0].level.language is not None
+            assert hq.nicks[0].level.language.level is None
+            assert hq.nicks[1].level.language is not None
+            assert hq.nicks[1].level.language.level is None
+
+            await hq.load_all(follow=True, exclude="nicks__level")
+            assert len(hq.nicks) == 2
+            assert hq.nicks[0].level is None
+            assert hq.nicks[1].level is None
+
+            await hq.load_all(follow=True, exclude="nicks")
+            assert len(hq.nicks) == 0

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

tests/test_model_methods/test_saving_related.py

@@ -0,0 +1,76 @@
+from typing import Union
+
+import databases
+import pytest
+import sqlalchemy as sa
+from sqlalchemy import create_engine
+
+import ormar
+from ormar.exceptions import ModelPersistenceError
+from tests.settings import DATABASE_URL
+
+metadata = sa.MetaData()
+db = databases.Database(DATABASE_URL)
+
+
+class Category(ormar.Model):
+    class Meta:
+        tablename = "categories"
+        metadata = metadata
+        database = db
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=50, unique=True, index=True)
+    code: int = ormar.Integer()
+
+
+class Workshop(ormar.Model):
+    class Meta:
+        tablename = "workshops"
+        metadata = metadata
+        database = db
+
+    id: int = ormar.Integer(primary_key=True)
+    topic: str = ormar.String(max_length=255, index=True)
+    category: Union[ormar.Model, Category] = ormar.ForeignKey(
+        Category, related_name="workshops", nullable=False
+    )
+
+
+@pytest.fixture(autouse=True, scope="module")
+def create_test_database():
+    engine = create_engine(DATABASE_URL)
+    metadata.create_all(engine)
+    yield
+    metadata.drop_all(engine)
+
+
+@pytest.mark.asyncio
+async def test_model_relationship():
+    async with db:
+        async with db.transaction(force_rollback=True):
+            cat = await Category(name="Foo", code=123).save()
+            ws = await Workshop(topic="Topic 1", category=cat).save()
+
+            assert ws.id == 1
+            assert ws.topic == "Topic 1"
+            assert ws.category.name == "Foo"
+
+            ws.topic = "Topic 2"
+            await ws.update()
+
+            assert ws.id == 1
+            assert ws.topic == "Topic 2"
+            assert ws.category.name == "Foo"
+
+
+@pytest.mark.asyncio
+async def test_model_relationship_with_not_saved():
+    async with db:
+        async with db.transaction(force_rollback=True):
+            cat = Category(name="Foo", code=123)
+            with pytest.raises(ModelPersistenceError):
+                await Workshop(topic="Topic 1", category=cat).save()
+
+            with pytest.raises(ModelPersistenceError):
+                await Workshop.objects.create(topic="Topic 1", category=cat)