integration test for our view

hjwp · hjwp · commit bd3501df3c15 · 2020-03-05T19:57:00.000Z
diff --git a/tests/integration/test_views.py b/tests/integration/test_views.py
@@ -0,0 +1,23 @@
+from datetime import date
+from allocation import views
+from allocation.domain import commands
+from allocation.service_layer import messagebus, 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)
+    # 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)
+
+    assert views.allocations('order1', uow) == [
+        {'sku': 'sku1', 'batchref': 'sku1batch'},
+        {'sku': 'sku2', 'batchref': 'sku2batch'},
+    ]
