implement .seen on repository [repository_tracks_seen]

hjwp · hjwp · commit 651f813dcafc · 2019-07-18T22:23:22.000+01:00
diff --git a/src/allocation/repository.py b/src/allocation/repository.py
@@ -1,31 +1,41 @@
+from typing import Set
 import abc
 from allocation import model
 
 class AbstractRepository(abc.ABC):
 
-    @abc.abstractmethod
+    def __init__(self):
+        self.seen = set()  # type: Set[model.Product]
+
     def add(self, product):
+        self._add(product)
+        self.seen.add(product)
+
+    def get(self, sku):
+        p = self._get(sku)
+        if p:
+            self.seen.add(p)
+        return p
+
+    @abc.abstractmethod
+    def _add(self, product):
         raise NotImplementedError
 
     @abc.abstractmethod
-    def get(self, sku):
+    def _get(self, sku):
         raise NotImplementedError
 
 
 
 class SqlAlchemyRepository(AbstractRepository):
 
     def __init__(self, session):
+        super().__init__()
         self.session = session
-        self.seen = set()
 
-    def add(self, product):
-        self.seen.add(product)
+    def _add(self, product):
         self.session.add(product)
 
-    def get(self, sku):
-        p = self.session.query(model.Product).filter_by(sku=sku).first()
-        if p:
-            self.seen.add(p)
-        return p
+    def _get(self, sku):
+        return self.session.query(model.Product).filter_by(sku=sku).first()
 
diff --git a/tests/unit/test_services.py b/tests/unit/test_services.py
@@ -6,12 +6,13 @@
 class FakeRepository(repository.AbstractRepository):
 
     def __init__(self, products):
+        super().__init__()
         self._products = set(products)
 
-    def add(self, product):
+    def _add(self, product):
         self._products.add(product)
 
-    def get(self, sku):
+    def _get(self, sku):
         return next((p for p in self._products if p.sku == sku), None)
 
 
