test and Product change to emit event [model_emits_allocated_event]

hjwp · hjwp · commit 7a75df790e56 · 2021-02-24T20:08:31.000Z
diff --git a/src/allocation/domain/model.py b/src/allocation/domain/model.py
@@ -17,6 +17,14 @@ def allocate(self, line: OrderLine) -> str:
             batch = next(b for b in sorted(self.batches) if b.can_allocate(line))
             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
@@ -43,6 +43,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])
