fake repo get_by_batchref [fakerepo_get_by_batchref]

hjwp · hjwp · commit 6b767c18ad28 · 2019-07-26T16:31:50.000+01:00
diff --git a/tests/unit/test_services.py b/tests/unit/test_services.py
@@ -16,6 +16,12 @@ def _add(self, product):
     def _get(self, sku):
         return next((p for p in self._products if p.sku == sku), None)
 
+    def _get_by_batchref(self, batchref):
+        return next((
+            p for p in self._products for b in p.batches
+            if b.reference == batchref
+        ), None)
+
 
 class FakeUnitOfWork(unit_of_work.AbstractUnitOfWork):
 
