integration test for our view

hjwp · hjwp · commit 0866c72d5522 · 2021-02-24T20:08:31.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"},
+    ]
