From 8e5316782687b505a9eaa2d35ce28a787477bb01 Mon Sep 17 00:00:00 2001
From: Mike Edmunds <medmunds@gmail.com>
Date: Wed, 7 Feb 2024 12:45:13 -0800
Subject: [PATCH] Add integration tests

---
 tests/test_mailpace_integration.py | 113 +++++++++++++++++++++++++++++
 1 file changed, 113 insertions(+)
 create mode 100644 tests/test_mailpace_integration.py

diff --git a/tests/test_mailpace_integration.py b/tests/test_mailpace_integration.py
new file mode 100644
index 0000000..125541e
--- /dev/null
+++ b/tests/test_mailpace_integration.py
@@ -0,0 +1,113 @@
+import os
+import unittest
+from email.headerregistry import Address
+
+from django.test import SimpleTestCase, override_settings, tag
+
+from anymail.exceptions import AnymailAPIError
+from anymail.message import AnymailMessage
+
+from .utils import AnymailTestMixin, sample_image_path
+
+ANYMAIL_TEST_MAILPACE_SERVER_TOKEN = os.getenv("ANYMAIL_TEST_MAILPACE_SERVER_TOKEN")
+ANYMAIL_TEST_MAILPACE_DOMAIN = os.getenv("ANYMAIL_TEST_MAILPACE_DOMAIN")
+
+
+@tag("mailpace", "live")
+@unittest.skipUnless(
+    ANYMAIL_TEST_MAILPACE_SERVER_TOKEN and ANYMAIL_TEST_MAILPACE_DOMAIN,
+    "Set ANYMAIL_TEST_MAILPACE_SERVER_TOKEN and ANYMAIL_TEST_MAILPACE_DOMAIN"
+    " environment variables to run MailPace integration tests",
+)
+@override_settings(
+    ANYMAIL_MAILPACE_SERVER_TOKEN=ANYMAIL_TEST_MAILPACE_SERVER_TOKEN,
+    EMAIL_BACKEND="anymail.backends.mailpace.EmailBackend",
+)
+class MailPaceBackendIntegrationTests(AnymailTestMixin, SimpleTestCase):
+    """
+    MailPace API integration tests
+
+    These tests run against the **live** MailPace API, using the
+    environment variable `ANYMAIL_TEST_MAILPACE_SERVER_TOKEN` as the API key,
+    and `ANYMAIL_TEST_MAILPACE_DOMAIN` to construct sender addresses.
+    If those variables are not set, these tests won't run.
+    """
+
+    def setUp(self):
+        super().setUp()
+        self.from_email = str(
+            Address(username="from", domain=ANYMAIL_TEST_MAILPACE_DOMAIN)
+        )
+        self.message = AnymailMessage(
+            "Anymail MailPace integration test",
+            "Text content",
+            self.from_email,
+            ["test+to1@anymail.dev"],
+        )
+        self.message.attach_alternative("<p>HTML content</p>", "text/html")
+
+    def test_simple_send(self):
+        # Example of getting the MailPace send status and message id from the message
+        sent_count = self.message.send()
+        self.assertEqual(sent_count, 1)
+
+        anymail_status = self.message.anymail_status
+        sent_status = anymail_status.recipients["test+to1@anymail.dev"].status
+        message_id = anymail_status.recipients["test+to1@anymail.dev"].message_id
+
+        self.assertEqual(sent_status, "queued")
+        self.assertGreater(message_id, 0)  # integer MailPace reference ID
+        # set of all recipient statuses:
+        self.assertEqual(anymail_status.status, {sent_status})
+        self.assertEqual(anymail_status.message_id, message_id)
+
+    def test_all_options(self):
+        message = AnymailMessage(
+            subject="Anymail MailPace all-options integration test",
+            body="This is the text body",
+            from_email=str(
+                Address(
+                    display_name="Test From, with comma",
+                    username="sender",
+                    domain=ANYMAIL_TEST_MAILPACE_DOMAIN,
+                )
+            ),
+            to=[
+                "test+to1@anymail.dev",
+                '"Recipient 2, with comma" <test+to2@anymail.dev>',
+            ],
+            cc=["test+cc1@anymail.dev", "Copy 2 <test+cc2@anymail.dev>"],
+            bcc=["test+bcc1@anymail.dev", "Blind Copy 2 <test+bcc2@anymail.dev>"],
+            reply_to=["reply1@example.com", "Reply 2 <reply2@example.com>"],
+            headers={"List-Unsubscribe": "<https://example.com/unsub?id=123>"},
+            tags=["tag 1", "tag 2"],
+        )
+        message.attach("attachment1.txt", "Here is some\ntext for you", "text/plain")
+        message.attach("attachment2.csv", "ID,Name\n1,Amy Lina", "text/csv")
+        cid = message.attach_inline_image_file(sample_image_path())
+        message.attach_alternative(
+            "<p><b>HTML:</b> with <a href='http://example.com'>link</a>"
+            "and image: <img src='cid:%s'></div>" % cid,
+            "text/html",
+        )
+
+        message.send()
+        self.assertEqual(message.anymail_status.status, {"queued"})
+        self.assertEqual(
+            message.anymail_status.recipients["test+to1@anymail.dev"].status, "queued"
+        )
+        self.assertEqual(
+            message.anymail_status.recipients["test+to2@anymail.dev"].status, "queued"
+        )
+
+    def test_invalid_from(self):
+        self.message.from_email = "webmaster@localhost"  # Django's default From
+        with self.assertRaisesMessage(
+            AnymailAPIError, "does not match domain in From field (localhost)"
+        ):
+            self.message.send()
+
+    @override_settings(ANYMAIL_MAILPACE_SERVER_TOKEN="Hey, that's not a server token!")
+    def test_invalid_server_token(self):
+        with self.assertRaisesMessage(AnymailAPIError, "Invalid API Token"):
+            self.message.send()