fake repo get_by_batchref [fakerepo_get_by_batchref]

hjwp · hjwp · commit 8f67133de3e3 · 2021-02-24T17:03:23.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):
     def __init__(self):
