test.py

import subprocess
import atexit
import time
import webbrowser
import argparse
import os
import socket


def get_host_ip_address():
    """
    Copied from https://stackoverflow.com/a/28950776/11741232
    Returns: the local computer's IP address, which tries to not return 127.0.0.1 which is the localhost IP address.
    We want this address for the Docker to be able to send messages out to the local computer. Since the Docker
    container can't find its hosts IP address easily, we have to pass it in as an environment variable.
    """
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    s.settimeout(0)
    try:
        # doesn't even have to be reachable
        s.connect(("10.254.254.254", 1))
        ip = s.getsockname()[0]
    except Exception:
        ip = "127.0.0.1"
    finally:
        s.close()
    return ip


def run_and_wait(command, env=None):
    process = subprocess.Popen(command, shell=True, env=env)
    process.wait()


def post_script_tasks():
    run_and_wait("docker compose stop")


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Test entire system, locally or via the server"
    )
    parser.add_argument(
        "--local", action="store_true", help="Run in local mode, bypassing the server"
    )
    args = parser.parse_args()
    env = os.environ.copy()
    if args.local:
        print("Running with docker-compose-local.yml")
        docker = "docker compose -f docker-compose-local.yml up  --build -d"
        env["remote_server"] = "False"
        env["host_ip"] = get_host_ip_address()
        print(
            f"Using host ip address for docker to python communication: {env['host_ip']}"
        )
        gamergui = "python gamer-gui/main.py --test --local"
    else:
        docker = "docker compose up --build -d"
        env["remote_server"] = "True"
        gamergui = "python gamer-gui/main.py --test"

    icarus_url = "http://localhost:3000/"
    minnie_url = "http://localhost:3001/"

    atexit.register(post_script_tasks)

    run_and_wait(docker, env=env)
    time.sleep(6)
    webbrowser.open(icarus_url)
    webbrowser.open(minnie_url)
    time.sleep(3)
    subprocess.Popen(gamergui, shell=True, env=env)

    while True:
        time.sleep(1)