test_docker.py

#!/usr/bin/env python3

import subprocess
import time
import requests
import sys

DOCKER_COMPOSE_FILE = "compose.yaml"
BEACON_PORT = 8088
SMTP4DEV_PORT = 5080

def run_compose_cmd(*args):
    """Helper to run a docker compose command."""
    cmd = ["docker", "compose", "-f", DOCKER_COMPOSE_FILE] + list(args)
    print(f"> Running: {' '.join(cmd)}")
    subprocess.check_call(cmd)

# todo: refactor
def main():
    try:
        print("Starting Beacon test.")
        print("---------------------")
        run_compose_cmd("down")
        run_compose_cmd("up", "-d", "--build")

        print("Test API...")
        print("---------------------")

        for _ in range(10):
            try:
                resp = requests.get(f"http://localhost:{BEACON_PORT}")
                resp.raise_for_status()
                print("Beacon Up")
                break
            except requests.RequestException as ex:
                print("Waiting for Beacon to be ready...", ex)
                time.sleep(1)
        else:
            print("Beacon did not become ready in time.")
            sys.exit(1)

        # POST request
        print("Sending POST to /services/sly-fox/beat")
        post_url = f"http://localhost:{BEACON_PORT}/services/sly-fox/beat"
        resp = requests.post(post_url)
        resp.raise_for_status()
        print("POST succeeded!")

        # GET request
        print("Sending GET to /services/sly-fox/status")
        get_url = f"http://localhost:{BEACON_PORT}/services/sly-fox/status"
        resp = requests.get(get_url)
        resp.raise_for_status()
        print("GET succeeded! Response:")
        print(resp.text)

        print("Test Sending report...")
        print("---------------------")

        # Wait for smtp4dev to be up
        smtp_health_url = f"http://localhost:{SMTP4DEV_PORT}/api/Version"
        for _ in range(10):
            try:
                resp = requests.get(smtp_health_url)
                resp.raise_for_status()
                print("SMTP Up")
                break
            except requests.RequestException:
                print("Waiting for smtp4dev to be ready...")
                time.sleep(1)
        else:
            print("smtp4dev did not become ready in time.")
            sys.exit(1)

        # Count emails before
        email_count_before = get_email_count()
        print(f"Email count before: {email_count_before}")

        # Trigger the report in the container
        # "docker compose exec beacon /app/beacon report --config /app/beacon.yaml"
        run_compose_cmd("exec", "beacon", "/app/beacon", "report", "--config", "/app/beacon.yaml")

        # Count emails after
        email_count_after = get_email_count()
        print(f"Email count after: {email_count_after}")

        if email_count_before == email_count_after:
            print("Testing: FAIL - email not sent")
            print("-------------")
            sys.exit(1)
        else:
            print("Testing: SUCCESS")
            print("----------------")
            sys.exit(0)

    except subprocess.CalledProcessError as e:
        print(f"Command failed: {e}")
        sys.exit(1)
    except requests.RequestException as e:
        print(f"HTTP request failed: {e}")
        sys.exit(1)


def get_email_count():
    """
    Fetches the list of messages from smtp4dev and counts
    how many contain 'you@example.fake' in the message data.
    """
    smtp_messages_url = f"http://localhost:{SMTP4DEV_PORT}/api/Messages"
    resp = requests.get(smtp_messages_url)
    resp.raise_for_status()

    messages_json = resp.json()
    results = messages_json["results"]

    return len(results)


if __name__ == "__main__":
    main()