hacky way for messagebus to return results

hjwp · hjwp · commit 691de96b6a1e · 2019-08-13T07:30:16.000+01:00
diff --git a/src/allocation/messagebus.py b/src/allocation/messagebus.py
@@ -4,10 +4,15 @@
 
 
 def handle(events_: List[events.Event], uow: unit_of_work.AbstractUnitOfWork):
+    results = []
     while events_:
         event = events_.pop(0)
+        print('handling message', event, flush=True)
         for handler in HANDLERS[type(event)]:
-            handler(event, uow=uow)
+            r = handler(event, uow=uow)
+            print('got result', r, flush=True)
+            results.append(r)
+    return results
 
 
 HANDLERS = {
diff --git a/tests/unit/test_handlers.py b/tests/unit/test_handlers.py
@@ -69,11 +69,11 @@ class TestAllocate:
     @staticmethod
     def test_returns_allocation():
         uow = FakeUnitOfWork()
-        result = messagebus.handle([
+        results = messagebus.handle([
             events.BatchCreated("b1", "COMPLICATED-LAMP", 100, None),
-            events.AllocationRequest("o1", "COMPLICATED-LAMP", 10 )
+            events.AllocationRequest("o1", "COMPLICATED-LAMP", 10),
         ], uow)
-        assert result == "b1"
+        assert results.pop() == "b1"
 
     @staticmethod
     def test_errors_for_invalid_sku():
@@ -85,7 +85,6 @@ def test_errors_for_invalid_sku():
                 events.AllocationRequest("o1", "NONEXISTENTSKU", 10)
             ], uow)
 
-
     @staticmethod
     def test_commits():
         uow = FakeUnitOfWork()
