isolated test for a handler [test_handler_in_isolation]

hjwp · hjwp · commit 1f21ddfa2983 · 2024-03-01T15:05:12.000Z
diff --git a/tests/unit/test_handlers.py b/tests/unit/test_handlers.py
@@ -137,3 +137,28 @@ def __init__(self):
     def collect_new_events(self):
         self.events_published += super().collect_new_events()
         return []
+
+
+def test_reallocates_if_necessary_isolated():
+    uow = FakeUnitOfWorkWithFakeMessageBus()
+
+    # test setup as before
+    event_history = [
+        events.BatchCreated("batch1", "INDIFFERENT-TABLE", 50, None),
+        events.BatchCreated("batch2", "INDIFFERENT-TABLE", 50, date.today()),
+        events.AllocationRequired("order1", "INDIFFERENT-TABLE", 20),
+        events.AllocationRequired("order2", "INDIFFERENT-TABLE", 20),
+    ]
+    for e in event_history:
+        messagebus.handle(e, uow)
+    [batch1, batch2] = uow.products.get(sku="INDIFFERENT-TABLE").batches
+    assert batch1.available_quantity == 10
+    assert batch2.available_quantity == 50
+
+    messagebus.handle(events.BatchQuantityChanged("batch1", 25), uow)
+
+    # assert on new events emitted rather than downstream side-effects
+    [reallocation_event] = uow.events_published
+    assert isinstance(reallocation_event, events.AllocationRequired)
+    assert reallocation_event.orderid in {"order1", "order2"}
+    assert reallocation_event.sku == "INDIFFERENT-TABLE"
