get_by_batchref in abstract+real repo, and a new integration test for it [get_by_batchref]

hjwp · hjwp · commit 3be8292590b9 · 2019-07-26T16:31:50.000+01:00
diff --git a/src/allocation/repository.py b/src/allocation/repository.py
@@ -1,6 +1,7 @@
 from typing import Set
 import abc
-from allocation import model
+from allocation import model, orm
+
 
 class AbstractRepository(abc.ABC):
 
@@ -17,6 +18,12 @@ def get(self, sku):
             self.seen.add(p)
         return p
 
+    def get_by_batchref(self, batchref):
+        p = self._get_by_batchref(batchref)
+        if p:
+            self.seen.add(p)
+        return p
+
     @abc.abstractmethod
     def _add(self, product):
         raise NotImplementedError
@@ -25,6 +32,11 @@ def _add(self, product):
     def _get(self, sku):
         raise NotImplementedError
 
+    @abc.abstractmethod
+    def _get_by_batchref(self, batchref):
+        raise NotImplementedError
+
+
 
 
 class SqlAlchemyRepository(AbstractRepository):
@@ -39,3 +51,8 @@ def _add(self, product):
     def _get(self, sku):
         return self.session.query(model.Product).filter_by(sku=sku).first()
 
+    def _get_by_batchref(self, batchref):
+        return self.session.query(model.Product).join(model.Batch).filter(
+            orm.batches.c.reference == batchref,
+        ).first()
+
diff --git a/tests/integration/test_repository.py b/tests/integration/test_repository.py
@@ -0,0 +1,14 @@
+from allocation import model, repository
+
+def test_get_by_batchref(session):
+    repo = repository.SqlAlchemyRepository(session)
+    b1 = model.Batch(ref='b1', sku='sku1', qty=100, eta=None)
+    b2 = model.Batch(ref='b2', sku='sku1', qty=100, eta=None)
+    b3 = model.Batch(ref='b3', sku='sku2', qty=100, eta=None)
+    p1 = model.Product(sku='sku1', batches=[b1, b2])
+    p2 = model.Product(sku='sku2', batches=[b3])
+    repo.add(p1)
+    repo.add(p2)
+    assert repo.get_by_batchref('b2') == p1
+    assert repo.get_by_batchref('b3') == p2
+
