integration test for emails

hjwp · hjwp · commit 819840784d25 · 2020-01-03T23:35:41.000Z
diff --git a/tests/integration/test_email.py b/tests/integration/test_email.py
@@ -0,0 +1,44 @@
+#pylint: disable=redefined-outer-name
+import uuid
+import pytest
+import requests
+from allocation import config
+from allocation.domain import commands
+from allocation.adapters import notifications
+from allocation.service_layer import messagebus, unit_of_work
+
+email_config = config.get_email_host_and_port()
+
+@pytest.fixture
+def bus(sqlite_session_factory):
+    uow = unit_of_work.SqlAlchemyUnitOfWork(sqlite_session_factory)
+    bus = messagebus.MessageBus(
+        uow=uow,
+        notifications=notifications.EmailNotifications(
+            smtp_host=email_config['host'],
+            port=email_config['port'],
+        ),
+        publish=lambda *_, **__: None
+    )
+    uow.bus = bus
+    return bus
+
+
+def random_sku():
+    return uuid.uuid4().hex[:6]
+
+
+def test_out_of_stock_email(bus):
+    sku = random_sku()
+    bus.handle(commands.CreateBatch('batch1', sku, 9, None))
+    bus.handle(commands.Allocate('order1', sku, 10))
+    messages = requests.get(
+        f'http://{email_config["host"]}:{email_config["http_port"]}/api/v2/messages'
+    ).json()
+    message = next(
+        m for m in messages['items']
+        if sku in str(m)
+    )
+    assert message['Raw']['From'] == 'allocations@example.com'
+    assert message['Raw']['To'] == ['stock@made.com']
+    assert f'Out of stock for {sku}' in message['Raw']['Data']
