integration test for emails [chapter_13_dependency_injection_ends]

Author: hjwp

tests/integration/test_email.py

@@ -0,0 +1,37 @@
+# pylint: disable=redefined-outer-name
+import pytest
+import requests
+from sqlalchemy.orm import clear_mappers
+from allocation import bootstrap, config
+from allocation.domain import commands
+from allocation.adapters import notifications
+from allocation.service_layer import unit_of_work
+from ..random_refs import random_sku
+
+
+@pytest.fixture
+def bus(sqlite_session_factory):
+    bus = bootstrap.bootstrap(
+        start_orm=True,
+        uow=unit_of_work.SqlAlchemyUnitOfWork(sqlite_session_factory),
+        notifications=notifications.EmailNotifications(),
+        publish=lambda *args: None,
+    )
+    yield bus
+    clear_mappers()
+
+
+def get_email_from_mailhog(sku):
+    host, port = map(config.get_email_host_and_port().get, ["host", "http_port"])
+    all_emails = requests.get(f"http://{host}:{port}/api/v2/messages").json()
+    return next(m for m in all_emails["items"] if sku in str(m))
+
+
+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))
+    email = get_email_from_mailhog(sku)
+    assert email["Raw"]["From"] == "[email protected]"
+    assert email["Raw"]["To"] == ["[email protected]"]
+    assert f"Out of stock for {sku}" in email["Raw"]["Data"]