update test db core link

zcemycl · Oct 27, 2023 · 499c113 · 499c113
1 parent 82799f7
commit 499c113
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 6 deletions.
diff --git a/src/example_package/dataclasses/__init__.py b/src/example_package/dataclasses/__init__.py
@@ -21,15 +21,15 @@
     Column(
         "parent_person_id",
         Integer,
-        ForeignKey("person.id"),
+        ForeignKey("person.id", ondelete="CASCADE"),
         primary_key=True,
         index=True,
         unique=False,
     ),
     Column(
         "child_person_id",
         Integer,
-        ForeignKey("person.id"),
+        ForeignKey("person.id", ondelete="CASCADE"),
         primary_key=True,
         index=True,
         unique=False,
@@ -45,15 +45,15 @@
     Column(
         "person_id",
         Integer,
-        ForeignKey("person.id"),
+        ForeignKey("person.id", ondelete="CASCADE"),
         primary_key=True,
         index=True,
         unique=False,
     ),
     Column(
         "skill_id",
         Integer,
-        ForeignKey("skill.id"),
+        ForeignKey("skill.id", ondelete="CASCADE"),
         primary_key=True,
         index=True,
         unique=False,

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -35,6 +35,7 @@ def get_engine() -> Engine:
 
 @pytest.fixture(autouse=True)
 def reset(get_engine: Engine):
+    print("1")
     engine = get_engine
     metadata.drop_all(bind=engine)
     metadata.create_all(bind=engine)

diff --git a/tests/test_db.py b/tests/test_db.py
@@ -1,10 +1,10 @@
 import json
 from pathlib import Path
 
-from sqlalchemy.sql import insert, select
+from sqlalchemy.sql import delete, insert, select
 
 import example_package.dataclasses.orm as d
-from example_package.dataclasses import person
+from example_package.dataclasses import friendship, person
 
 
 def test_db(get_engine):
@@ -18,6 +18,37 @@ def test_db(get_engine):
         _ = conn.execute(stmt)
 
 
+def test_db_core_link(get_engine):
+    engine = get_engine
+
+    with engine.begin() as conn:
+        _ = conn.execute(
+            insert(person).values(
+                [
+                    {
+                        "firstname": "1",
+                        "lastname": "1",
+                        "others": {},
+                        "role": "developer",
+                    },
+                    {
+                        "firstname": "2",
+                        "lastname": "2",
+                        "others": {},
+                        "role": "developer",
+                    },
+                ]
+            )
+        )
+        _ = conn.execute(
+            insert(friendship).values(
+                [{"parent_person_id": 1, "child_person_id": 2}]
+            )
+        )
+        d = delete(person).where(person.c.id == 1)
+        conn.execute(d)
+
+
 def test_orm_link(test_session2):
     p1 = d.person(firstname="1", lastname="1", others={}, role="developer")
     s1 = d.skill(name="python", persons=[p1])