Adjust unit tests now we're no longer raising out of stock exception

hjwp · hjwp · commit db5bcd60082b · 2019-07-18T22:23:22.000+01:00
diff --git a/tests/unit/test_product.py b/tests/unit/test_product.py
@@ -1,6 +1,5 @@
 from datetime import date, timedelta
-import pytest
-from allocation import events, exceptions
+from allocation import events
 from allocation.model import Product, OrderLine, Batch
 
 today = date.today()
@@ -42,23 +41,14 @@ def test_returns_allocated_batch_ref():
     assert allocation == in_stock_batch.reference
 
 
-def test_raises_out_of_stock_exception_if_cannot_allocate():
-    batch = Batch('batch1', 'HEAVY-SPOON', 100, eta=today)
-    different_sku_line = OrderLine('oref', 'SMALL-FORK', 10)
-    product = Product(sku="HIGHBROW-POSTER", batches=[batch])
-
-    with pytest.raises(exceptions.OutOfStock, match='SMALL-FORK'):
-        product.allocate(different_sku_line)
-
-
 def test_records_out_of_stock_event_if_cannot_allocate():
-    sku1_batch = Batch('batch1', 'sku1', 100, eta=today)
-    sku2_line = OrderLine('oref', 'sku2', 10)
-    product = Product(sku='sku1', batches=[sku1_batch])
+    batch = Batch("batch1", "HEAVY-SPOON", 100, eta=today)
+    different_sku_line = OrderLine("oref", "SMALL-FORK", 10)
+    product = Product(sku="HEAVY-SPOON", batches=[batch])
 
-    with pytest.raises(exceptions.OutOfStock):
-        product.allocate(sku2_line)
-    assert product.events[-1] == events.OutOfStock(sku='sku2')
+    allocation = product.allocate(different_sku_line)
+    assert product.events[-1] == events.OutOfStock(sku="SMALL-FORK")
+    assert allocation is None
 
 
 def test_increments_version_number():
diff --git a/tests/unit/test_services.py b/tests/unit/test_services.py
@@ -70,8 +70,7 @@ def test_sends_email_on_out_of_stock_error():
     services.add_batch("b1", "POPULAR-CURTAINS", 9, None, uow)
 
     with mock.patch("allocation.email.send_mail") as mock_send_mail:
-        with pytest.raises(exceptions.OutOfStock):
-            services.allocate("o1", "POPULAR-CURTAINS", 10, uow)
+        services.allocate("o1", "POPULAR-CURTAINS", 10, uow)
         assert mock_send_mail.call_args == mock.call(
             "stock@made.com",
             f"Out of stock for POPULAR-CURTAINS",
