FakeRepository [fake_repo]

hjwp · hjwp · commit 08651a5104e8 · 2020-02-21T12:02:54.000Z
diff --git a/test_services.py b/test_services.py
@@ -1,9 +1,24 @@
 import pytest
-
 import model
+import repository
 import services
 
 
+class FakeRepository(repository.AbstractRepository):
+
+    def __init__(self, batches):
+        self._batches = set(batches)
+
+    def add(self, batch):
+        self._batches.add(batch)
+
+    def get(self, reference):
+        return next(b for b in self._batches if b.reference == reference)
+
+    def list(self):
+        return list(self._batches)
+
+
 def test_returns_allocation():
     line = model.OrderLine("o1", "COMPLICATED-LAMP", 10)
     batch = model.Batch("b1", "COMPLICATED-LAMP", 100, eta=None)
