fix issue #68

tests/test_saving_related.py

@@ -0,0 +1,53 @@
+from typing import Union
+
+import databases
+import ormar
+import pytest
+import sqlalchemy as sa
+
+from tests.settings import DATABASE_URL
+
+engine = sa.create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
+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
+def create_test_database():
+    metadata.create_all(engine)
+    yield
+    metadata.drop_all(engine)
+
+
+@pytest.mark.asyncio
+async def test_model_relationship(create_test_database):
+    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"

tests/test_signals.py

@@ -78,6 +78,7 @@ def test_passing_not_callable():
 
 def test_passing_callable_without_kwargs():
     with pytest.raises(SignalDefinitionError):
+
         @pre_save(Album)
         def trigger(sender, instance):  # pragma: no cover
             pass
@@ -87,6 +88,7 @@ def test_passing_callable_without_kwargs():
 async def test_signal_functions(cleanup):
     async with database:
         async with database.transaction(force_rollback=True):
+
             @pre_save(Album)
             async def before_save(sender, instance, **kwargs):
                 await AuditLog(
@@ -170,9 +172,9 @@ async def test_signal_functions(cleanup):
             assert len(audits) == 2
             assert audits[0].event_type == "PRE_DELETE_album"
             assert (
-                    audits[0].event_log.get("id")
-                    == audits[1].event_log.get("id")
-                    == album.id
+                audits[0].event_log.get("id")
+                == audits[1].event_log.get("id")
+                == album.id
             )
             assert audits[1].event_type == "POST_DELETE_album"
 
@@ -186,6 +188,7 @@ async def test_signal_functions(cleanup):
 async def test_multiple_signals(cleanup):
     async with database:
         async with database.transaction(force_rollback=True):
+
             @pre_save(Album)
             async def before_save(sender, instance, **kwargs):
                 await AuditLog(
@@ -216,6 +219,7 @@ async def test_multiple_signals(cleanup):
 async def test_static_methods_as_signals(cleanup):
     async with database:
         async with database.transaction(force_rollback=True):
+
             class AlbumAuditor:
                 event_type = "ALBUM_INSTANCE"
 
@@ -240,6 +244,7 @@ async def test_static_methods_as_signals(cleanup):
 async def test_methods_as_signals(cleanup):
     async with database:
         async with database.transaction(force_rollback=True):
+
             class AlbumAuditor:
                 def __init__(self):
                     self.event_type = "ALBUM_INSTANCE"
@@ -265,6 +270,7 @@ async def test_methods_as_signals(cleanup):
 async def test_multiple_senders_signal(cleanup):
     async with database:
         async with database.transaction(force_rollback=True):
+
             @pre_save([Album, Cover])
             async def before_save(sender, instance, **kwargs):
                 await AuditLog(
@@ -292,6 +298,7 @@ async def test_multiple_senders_signal(cleanup):
 async def test_modifing_the_instance(cleanup):
     async with database:
         async with database.transaction(force_rollback=True):
+
             @pre_update(Album)
             async def before_update(sender, instance, **kwargs):
                 if instance.play_count > 50 and not instance.is_best_seller: