fix view tests to use bootstrap. [bootstrap_view_tests]

hjwp · hjwp · commit f954daec3959 · 2020-03-05T19:59:43.000Z
diff --git a/tests/integration/test_views.py b/tests/integration/test_views.py
@@ -1,35 +1,47 @@
+# 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'},
     ]
