fake repo get_by_batchref [fakerepo_get_by_batchref]

hjwp · hjwp · commit 679a763ae6b8 · 2020-01-03T03:01:00.000Z
diff --git a/tests/unit/test_handlers.py b/tests/unit/test_handlers.py
@@ -19,6 +19,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):
 
