sort-of fake messagebus on fake-uow. [fake_messagebus]

hjwp · hjwp · commit f8ed253f3326 · 2019-11-20T21:12:36.000Z
diff --git a/tests/unit/test_handlers.py b/tests/unit/test_handlers.py
@@ -1,5 +1,6 @@
 from datetime import date
 from unittest import mock
+from typing import List
 import pytest
 from allocation import events, exceptions, messagebus, repository, unit_of_work
 
@@ -122,3 +123,15 @@ def test_reallocates_if_necessary():
         assert batch1.available_quantity == 5
         # and 20 will be reallocated to the next batch
         assert batch2.available_quantity == 30
+
+
+class FakeUnitOfWorkWithFakeMessageBus(FakeUnitOfWork):
+
+    def __init__(self):
+        super().__init__()
+        self.events_published = []  # type: List[events.Event]
+
+    def publish_events(self):
+        for product in self.products.seen:
+            while product.events:
+                self.events_published.append(product.events.pop(0))
