use bootstrap in service layer tests [bootstrap_tests]

Author: hjwp

tests/unit/test_handlers.py

@@ -1,10 +1,12 @@
 # pylint: disable=no-self-use
+from __future__ import annotations
 from datetime import date
 from unittest import mock
 import pytest
+from allocation import bootstrap
 from allocation.adapters import repository
-from allocation.domain import commands, events
-from allocation.service_layer import handlers, messagebus, unit_of_work
+from allocation.domain import commands
+from allocation.service_layer import handlers, unit_of_work
 
 
 class FakeRepository(repository.AbstractRepository):
@@ -37,94 +39,96 @@ def rollback(self):
         pass
 
 
+def bootstrap_test_app():
+    return bootstrap.bootstrap(
+        start_orm=False,
+        uow=FakeUnitOfWork(),
+        send_mail=lambda *args: None,
+        publish=lambda *args: None,
+    )
+
+
 class TestAddBatch:
     def test_for_new_product(self):
-        uow = FakeUnitOfWork()
-        messagebus.handle(
-            commands.CreateBatch("b1", "CRUNCHY-ARMCHAIR", 100, None), uow
-        )
-        assert uow.products.get("CRUNCHY-ARMCHAIR") is not None
-        assert uow.committed
+        bus = bootstrap_test_app()
+        bus.handle(commands.CreateBatch("b1", "CRUNCHY-ARMCHAIR", 100, None))
+        assert bus.uow.products.get("CRUNCHY-ARMCHAIR") is not None
+        assert bus.uow.committed
 
     def test_for_existing_product(self):
-        uow = FakeUnitOfWork()
-        messagebus.handle(commands.CreateBatch("b1", "GARISH-RUG", 100, None), uow)
-        messagebus.handle(commands.CreateBatch("b2", "GARISH-RUG", 99, None), uow)
-        assert "b2" in [b.reference for b in uow.products.get("GARISH-RUG").batches]
-
-
-@pytest.fixture(autouse=True)
-def fake_redis_publish():
-    with mock.patch("allocation.adapters.redis_eventpublisher.publish"):
-        yield
+        bus = bootstrap_test_app()
+        bus.handle(commands.CreateBatch("b1", "GARISH-RUG", 100, None))
+        bus.handle(commands.CreateBatch("b2", "GARISH-RUG", 99, None))
+        assert "b2" in [
+            b.reference for b in bus.uow.products.get("GARISH-RUG").batches
+        ]
 
 
 class TestAllocate:
     def test_allocates(self):
-        uow = FakeUnitOfWork()
-        messagebus.handle(
-            commands.CreateBatch("batch1", "COMPLICATED-LAMP", 100, None), uow
-        )
-        messagebus.handle(commands.Allocate("o1", "COMPLICATED-LAMP", 10), uow)
-        [batch] = uow.products.get("COMPLICATED-LAMP").batches
+        bus = bootstrap_test_app()
+        bus.handle(commands.CreateBatch("batch1", "COMPLICATED-LAMP", 100, None))
+        bus.handle(commands.Allocate("o1", "COMPLICATED-LAMP", 10))
+        [batch] = bus.uow.products.get("COMPLICATED-LAMP").batches
         assert batch.available_quantity == 90
 
     def test_errors_for_invalid_sku(self):
-        uow = FakeUnitOfWork()
-        messagebus.handle(commands.CreateBatch("b1", "AREALSKU", 100, None), uow)
+        bus = bootstrap_test_app()
+        bus.handle(commands.CreateBatch("b1", "AREALSKU", 100, None))
 
         with pytest.raises(handlers.InvalidSku, match="Invalid sku NONEXISTENTSKU"):
-            messagebus.handle(commands.Allocate("o1", "NONEXISTENTSKU", 10), uow)
+            bus.handle(commands.Allocate("o1", "NONEXISTENTSKU", 10))
 
     def test_commits(self):
-        uow = FakeUnitOfWork()
-        messagebus.handle(
-            commands.CreateBatch("b1", "OMINOUS-MIRROR", 100, None), uow
-        )
-        messagebus.handle(commands.Allocate("o1", "OMINOUS-MIRROR", 10), uow)
-        assert uow.committed
+        bus = bootstrap_test_app()
+        bus.handle(commands.CreateBatch("b1", "OMINOUS-MIRROR", 100, None))
+        bus.handle(commands.Allocate("o1", "OMINOUS-MIRROR", 10))
+        assert bus.uow.committed
 
     def test_sends_email_on_out_of_stock_error(self):
-        uow = FakeUnitOfWork()
-        messagebus.handle(
-            commands.CreateBatch("b1", "POPULAR-CURTAINS", 9, None), uow
-        )
+        emails = []
 
-        with mock.patch("allocation.adapters.email.send") as mock_send_mail:
-            messagebus.handle(commands.Allocate("o1", "POPULAR-CURTAINS", 10), uow)
-            assert mock_send_mail.call_args == mock.call(
-                "[email protected]", f"Out of stock for POPULAR-CURTAINS"
-            )
+        def fake_send_mail(*args):
+            emails.append(args)
+
+        bus = bootstrap.bootstrap(
+            start_orm=False,
+            uow=FakeUnitOfWork(),
+            send_mail=fake_send_mail,
+            publish=lambda *args: None,
+        )
+        bus.handle(commands.CreateBatch("b1", "POPULAR-CURTAINS", 9, None))
+        bus.handle(commands.Allocate("o1", "POPULAR-CURTAINS", 10))
+        assert emails == [
+            ("[email protected]", f"Out of stock for POPULAR-CURTAINS"),
+        ]
 
 
 class TestChangeBatchQuantity:
     def test_changes_available_quantity(self):
-        uow = FakeUnitOfWork()
-        messagebus.handle(
-            commands.CreateBatch("batch1", "ADORABLE-SETTEE", 100, None), uow
-        )
-        [batch] = uow.products.get(sku="ADORABLE-SETTEE").batches
+        bus = bootstrap_test_app()
+        bus.handle(commands.CreateBatch("batch1", "ADORABLE-SETTEE", 100, None))
+        [batch] = bus.uow.products.get(sku="ADORABLE-SETTEE").batches
         assert batch.available_quantity == 100
 
-        messagebus.handle(commands.ChangeBatchQuantity("batch1", 50), uow)
-
+        bus.handle(commands.ChangeBatchQuantity("batch1", 50))
         assert batch.available_quantity == 50
 
     def test_reallocates_if_necessary(self):
-        uow = FakeUnitOfWork()
+        bus = bootstrap_test_app()
         history = [
             commands.CreateBatch("batch1", "INDIFFERENT-TABLE", 50, None),
             commands.CreateBatch("batch2", "INDIFFERENT-TABLE", 50, date.today()),
             commands.Allocate("order1", "INDIFFERENT-TABLE", 20),
             commands.Allocate("order2", "INDIFFERENT-TABLE", 20),
         ]
         for msg in history:
-            messagebus.handle(msg, uow)
-        [batch1, batch2] = uow.products.get(sku="INDIFFERENT-TABLE").batches
+            bus.handle(msg)
+        [batch1, batch2] = bus.uow.products.get(sku="INDIFFERENT-TABLE").batches
         assert batch1.available_quantity == 10
         assert batch2.available_quantity == 50
 
-        messagebus.handle(commands.ChangeBatchQuantity("batch1", 25), uow)
+        bus.handle(commands.ChangeBatchQuantity("batch1", 25))
 
         # order1 or order2 will be deallocated, so we'll have 25 - 20
         assert batch1.available_quantity == 5