tests change to use bus [handler_tests]

hjwp · hjwp · commit 0bc35ef48fcd · 2020-02-21T12:05:48.000Z
diff --git a/tests/unit/test_handlers.py b/tests/unit/test_handlers.py
@@ -1,8 +1,10 @@
 # pylint: disable=no-self-use
 from unittest import mock
 import pytest
+
 from allocation.adapters import repository
-from allocation.service_layer import services, unit_of_work
+from allocation.domain import events
+from allocation.service_layer import handlers, messagebus, unit_of_work
 
 
 class FakeRepository(repository.AbstractRepository):
@@ -36,49 +38,64 @@ class TestAddBatch:
 
     def test_for_new_product(self):
         uow = FakeUnitOfWork()
-        services.add_batch("b1", "CRUNCHY-ARMCHAIR", 100, None, uow)
+        messagebus.handle(
+            events.BatchCreated("b1", "CRUNCHY-ARMCHAIR", 100, None), uow
+        )
         assert uow.products.get("CRUNCHY-ARMCHAIR") is not None
         assert uow.committed
 
 
     def test_for_existing_product(self):
         uow = FakeUnitOfWork()
-        services.add_batch("b1", "GARISH-RUG", 100, None, uow)
-        services.add_batch("b2", "GARISH-RUG", 99, None, uow)
+        messagebus.handle(events.BatchCreated("b1", "GARISH-RUG", 100, None), uow)
+        messagebus.handle(events.BatchCreated("b2", "GARISH-RUG", 99, None), uow)
         assert "b2" in [b.reference for b in uow.products.get("GARISH-RUG").batches]
 
 
+
 class TestAllocate:
 
     def test_returns_allocation(self):
         uow = FakeUnitOfWork()
-        services.add_batch("batch1", "COMPLICATED-LAMP", 100, None, uow)
-        result = services.allocate("o1", "COMPLICATED-LAMP", 10, uow)
+        messagebus.handle(
+            events.BatchCreated("batch1", "COMPLICATED-LAMP", 100, None), uow
+        )
+        result = messagebus.handle(
+            events.AllocationRequired("o1", "COMPLICATED-LAMP", 10), uow
+        )
         assert result == "batch1"
 
 
     def test_errors_for_invalid_sku(self):
         uow = FakeUnitOfWork()
-        services.add_batch("b1", "AREALSKU", 100, None, uow)
-
-        with pytest.raises(services.InvalidSku, match="Invalid sku NONEXISTENTSKU"):
-            services.allocate("o1", "NONEXISTENTSKU", 10, uow)
+        messagebus.handle(events.BatchCreated("b1", "AREALSKU", 100, None), uow)
 
+        with pytest.raises(handlers.InvalidSku, match="Invalid sku NONEXISTENTSKU"):
+            messagebus.handle(
+                events.AllocationRequired("o1", "NONEXISTENTSKU", 10), uow
+            )
 
     def test_commits(self):
         uow = FakeUnitOfWork()
-        services.add_batch("b1", "OMINOUS-MIRROR", 100, None, uow)
-        services.allocate("o1", "OMINOUS-MIRROR", 10, uow)
+        messagebus.handle(
+            events.BatchCreated("b1", "OMINOUS-MIRROR", 100, None), uow
+        )
+        messagebus.handle(
+            events.AllocationRequired("o1", "OMINOUS-MIRROR", 10), uow
+        )
         assert uow.committed
 
 
     def test_sends_email_on_out_of_stock_error(self):
         uow = FakeUnitOfWork()
-        services.add_batch("b1", "POPULAR-CURTAINS", 9, None, uow)
+        messagebus.handle(
+            events.BatchCreated("b1", "POPULAR-CURTAINS", 9, None), uow
+        )
 
-        with mock.patch("allocation.adapters.email.send_mail") as mock_send_mail:
-            services.allocate("o1", "POPULAR-CURTAINS", 10, uow)
+        with mock.patch("allocation.adapters.email.send") as mock_send_mail:
+            messagebus.handle(
+                events.AllocationRequired("o1", "POPULAR-CURTAINS", 10), uow
+            )
             assert mock_send_mail.call_args == mock.call(
-                "stock@made.com",
-                f"Out of stock for POPULAR-CURTAINS",
+                "stock@made.com", f"Out of stock for POPULAR-CURTAINS"
             )
