strip out orm leaving only repo tests

hjwp · hjwp · commit cedfae3deced · 2021-02-24T21:58:12.000Z
diff --git a/conftest.py b/conftest.py
@@ -1,8 +1,9 @@
+# pytest: disable=redefined-outer-name
 import pytest
 from sqlalchemy import create_engine
-from sqlalchemy.orm import sessionmaker, clear_mappers
+from sqlalchemy.orm import sessionmaker
 
-from orm import metadata, start_mappers
+from db_tables import metadata
 
 
 @pytest.fixture
@@ -14,6 +15,4 @@ def in_memory_db():
 
 @pytest.fixture
 def session(in_memory_db):
-    start_mappers()
     yield sessionmaker(bind=in_memory_db)()
-    clear_mappers()
diff --git a/db_tables.py b/db_tables.py
@@ -1,8 +1,4 @@
 from sqlalchemy import Table, MetaData, Column, Integer, String, Date, ForeignKey
-from sqlalchemy.orm import mapper, relationship
-
-import model
-
 
 metadata = MetaData()
 
@@ -32,16 +28,3 @@
     Column("orderline_id", ForeignKey("order_lines.id")),
     Column("batch_id", ForeignKey("batches.id")),
 )
-
-
-def start_mappers():
-    lines_mapper = mapper(model.OrderLine, order_lines)
-    mapper(
-        model.Batch,
-        batches,
-        properties={
-            "_allocations": relationship(
-                lines_mapper, secondary=allocations, collection_class=set,
-            )
-        },
-    )
diff --git a/repository.py b/repository.py
@@ -17,10 +17,7 @@ def __init__(self, session):
         self.session = session
 
     def add(self, batch):
-        self.session.add(batch)
+        pass
 
-    def get(self, reference):
-        return self.session.query(model.Batch).filter_by(reference=reference).one()
-
-    def list(self):
-        return self.session.query(model.Batch).all()
+    def get(self, reference) -> model.Batch:
+        pass
diff --git a/test_orm.py b/test_orm.py
