Start of an integration test harness

Author: jamesridgway

.github/workflows/ci.yml

@@ -8,7 +8,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.10.15, 3.11.10, 3.12.6]
+        python-version: [3.12.6]
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python ${{ matrix.python-version }}
@@ -21,3 +21,5 @@ jobs:
       run: ./run-pylint.sh
     - name: Tests
       run: ./run-tests.sh
+    - name: Integration Tests
+      run: ./run-integration-tests.sh

attachment_downloader/version_info.py

@@ -13,6 +13,17 @@ class Version:
     """
     Version information.
     """
+
+    @staticmethod
+    def commit_hash_short():
+        """
+        Returns the short commit hash.
+        """
+        with subprocess.Popen(["git", "rev-parse", "--short", "HEAD"],
+                              stdout=subprocess.PIPE,
+                              stderr=None) as process:
+            return process.communicate()[0].decode('ascii').strip()
+
     @staticmethod
     def generate():
         """

requirements_tests.txt

@@ -0,0 +1 @@
+requests==2.32.3

run-integration-tests.sh

@@ -0,0 +1,2 @@
+#!/usr/bin/env bash
+./venv/bin/pytest -s tests_integration

setup.sh

@@ -4,3 +4,4 @@ python3 -m pip install --user virtualenv
 python3 -m venv venv
 ./venv/bin/pip install --upgrade pip
 ./venv/bin/pip install -r requirements.txt
+./venv/bin/pip install -r requirements_tests.txt

tests_integration/__init__.py

@@ -0,0 +1,43 @@
+import os
+import tempfile
+
+import requests
+
+from attachment_downloader.version_info import Version
+
+
+def send_email_with_attachment(api_key, domain, sender, recipient, subject, text, attachment_path):
+    url = f"https://api.eu.mailgun.net/v3/{domain}/messages"
+
+    with open(attachment_path, 'rb') as attachment_file:
+        response = requests.post(
+            url,
+            auth=("api", api_key),
+            files=[("attachment", (attachment_path, attachment_file.read()))],
+            data={
+                "from": sender,
+                "to": recipient,
+                "subject": subject,
+                "text": text,
+            }
+        )
+    return response.status_code == 200
+
+class TestSendAndDownloadAttachment:
+    def test_send_and_download_attachment(self):
+        mailgun_domain = os.getenv('MAILGUN_DOMAIN')
+        mailgun_api_key = os.getenv('MAILGUN_API_KEY')
+        recipient = os.getenv('AD_INT_TEST_RECIPIENT')
+        version_info = Version.commit_hash_short()
+
+        with tempfile.NamedTemporaryFile(mode='w+t',prefix=version_info,suffix='.txt') as temp_attachment:
+            temp_attachment.write(f"This is an example attachment from attachment-downloader {version_info}")
+            temp_attachment.flush()
+
+            send_email_with_attachment(mailgun_api_key,
+                                       mailgun_domain,
+                                       f'attachment-downloader@{mailgun_domain}',
+                                       recipient,
+                                       f"Integration Test - {version_info}",
+                                       'This email is part of an automated integration test.',
+                                       temp_attachment.name)