fix view tests to use bootstrap. [bootstrap_view_tests]

Author: hjwp

tests/integration/test_views.py

@@ -1,35 +1,48 @@
+# pylint: disable=redefined-outer-name
 from datetime import date
-from allocation import views
+from sqlalchemy.orm import clear_mappers
+import pytest
+from allocation import bootstrap, views
 from allocation.domain import commands
-from allocation.service_layer import messagebus, unit_of_work
+from allocation.service_layer import unit_of_work
 
 today = date.today()
 
 
-def test_allocations_view(sqlite_session_factory):
-    uow = unit_of_work.SqlAlchemyUnitOfWork(sqlite_session_factory)
-    messagebus.handle(commands.CreateBatch("sku1batch", "sku1", 50, None), uow)
-    messagebus.handle(commands.CreateBatch("sku2batch", "sku2", 50, today), uow)
-    messagebus.handle(commands.Allocate("order1", "sku1", 20), uow)
-    messagebus.handle(commands.Allocate("order1", "sku2", 20), uow)
+@pytest.fixture
+def sqlite_bus(sqlite_session_factory):
+    bus = bootstrap.bootstrap(
+        start_orm=True,
+        uow=unit_of_work.SqlAlchemyUnitOfWork(sqlite_session_factory),
+        send_mail=lambda *args: None,
+        publish=lambda *args: None,
+    )
+    yield bus
+    clear_mappers()
+
+
+def test_allocations_view(sqlite_bus):
+    sqlite_bus.handle(commands.CreateBatch("sku1batch", "sku1", 50, None))
+    sqlite_bus.handle(commands.CreateBatch("sku2batch", "sku2", 50, today))
+    sqlite_bus.handle(commands.Allocate("order1", "sku1", 20))
+    sqlite_bus.handle(commands.Allocate("order1", "sku2", 20))
     # add a spurious batch and order to make sure we're getting the right ones
-    messagebus.handle(commands.CreateBatch("sku1batch-later", "sku1", 50, today), uow)
-    messagebus.handle(commands.Allocate("otherorder", "sku1", 30), uow)
-    messagebus.handle(commands.Allocate("otherorder", "sku2", 10), uow)
+    sqlite_bus.handle(commands.CreateBatch("sku1batch-later", "sku1", 50, today))
+    sqlite_bus.handle(commands.Allocate("otherorder", "sku1", 30))
+    sqlite_bus.handle(commands.Allocate("otherorder", "sku2", 10))
 
-    assert views.allocations("order1", uow) == [
+    assert views.allocations("order1", sqlite_bus.uow) == [
         {"sku": "sku1", "batchref": "sku1batch"},
         {"sku": "sku2", "batchref": "sku2batch"},
     ]
 
 
-def test_deallocation(sqlite_session_factory):
-    uow = unit_of_work.SqlAlchemyUnitOfWork(sqlite_session_factory)
-    messagebus.handle(commands.CreateBatch("b1", "sku1", 50, None), uow)
-    messagebus.handle(commands.CreateBatch("b2", "sku1", 50, today), uow)
-    messagebus.handle(commands.Allocate("o1", "sku1", 40), uow)
-    messagebus.handle(commands.ChangeBatchQuantity("b1", 10), uow)
+def test_deallocation(sqlite_bus):
+    sqlite_bus.handle(commands.CreateBatch("b1", "sku1", 50, None))
+    sqlite_bus.handle(commands.CreateBatch("b2", "sku1", 50, today))
+    sqlite_bus.handle(commands.Allocate("o1", "sku1", 40))
+    sqlite_bus.handle(commands.ChangeBatchQuantity("b1", 10))
 
-    assert views.allocations("o1", uow) == [
+    assert views.allocations("o1", sqlite_bus.uow) == [
         {"sku": "sku1", "batchref": "b2"},
     ]