Mocky test for email [mocky_test_for_send_email]

hjwp · hjwp · commit a65d54409847 · 2019-07-18T22:12:32.000+01:00
diff --git a/src/allocation/email.py b/src/allocation/email.py
@@ -0,0 +1,2 @@
+def send_mail(*args):
+    print('SENDING EMAIL:', *args)
diff --git a/tests/unit/test_services.py b/tests/unit/test_services.py
@@ -1,3 +1,4 @@
+from unittest import mock
 import pytest
 from allocation import services, exceptions, repository, unit_of_work
 
@@ -31,7 +32,7 @@ def rollback(self):
 def test_add_batch_for_new_product():
     uow = FakeUnitOfWork()
     services.add_batch("b1", "CRUNCHY-ARMCHAIR", 100, None, uow)
-    assert uow.products.get('CRUNCHY-ARMCHAIR') is not None
+    assert uow.products.get("CRUNCHY-ARMCHAIR") is not None
     assert uow.committed
 
 
@@ -62,3 +63,16 @@ def test_allocate_commits():
     services.add_batch("b1", "OMINOUS-MIRROR", 100, None, uow)
     services.allocate("o1", "OMINOUS-MIRROR", 10, uow)
     assert uow.committed
+
+
+def test_sends_email_on_out_of_stock_error():
+    uow = FakeUnitOfWork()
+    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)
+        assert mock_send_mail.call_args == mock.call(
+            "stock@made.com",
+            f"Out of stock for POPULAR-CURTAINS",
+        )
