test and Product change to emit event [model_emits_allocated_event]

hjwp · hjwp · commit dc2042958b05 · 2019-11-20T21:12:36.000Z
diff --git a/src/allocation/model.py b/src/allocation/model.py
@@ -20,6 +20,10 @@ def allocate(self, line: OrderLine) -> str:
             )
             batch.allocate(line)
             self.version_number += 1
+            self.events.append(events.Allocated(
+                orderid=line.orderid, sku=line.sku, qty=line.qty,
+                batchref=batch.reference,
+            ))
             return batch.reference
         except StopIteration:
             self.events.append(events.OutOfStock(line.sku))
diff --git a/tests/unit/test_product.py b/tests/unit/test_product.py
@@ -41,6 +41,17 @@ def test_returns_allocated_batch_ref():
     assert allocation == in_stock_batch.reference
 
 
+def test_outputs_allocated_event():
+    batch = Batch("batchref", "RETRO-LAMPSHADE", 100, eta=None)
+    line = OrderLine("oref", "RETRO-LAMPSHADE", 10)
+    product = Product(sku="RETRO-LAMPSHADE", batches=[batch])
+    product.allocate(line)
+    expected = events.Allocated(
+        orderid="oref", sku="RETRO-LAMPSHADE", qty=10, batchref=batch.reference
+    )
+    assert product.events[-1] == expected
+
+
 def test_records_out_of_stock_event_if_cannot_allocate():
     batch = Batch('batch1', 'SMALL-FORK', 10, eta=today)
     product = Product(sku="SMALL-FORK", batches=[batch])
