101 lines
2.9 KiB
Python
101 lines
2.9 KiB
Python
import pytest
|
|
import subprocess
|
|
import time
|
|
import requests
|
|
import os
|
|
import sys
|
|
from playwright.sync_api import Page, expect
|
|
|
|
# Define the host and port for the application
|
|
HOST = "127.0.0.1"
|
|
PORT = 5000
|
|
BASE_URL = f"http://{HOST}:{PORT}"
|
|
STDOUT_FILE = "app_stdout.log"
|
|
STDERR_FILE = "app_stderr.log"
|
|
|
|
@pytest.fixture(scope="module")
|
|
def run_app():
|
|
"""
|
|
Fixture to run the Flask application in a test environment.
|
|
"""
|
|
# Set the environment variable for the subprocess
|
|
env = os.environ.copy()
|
|
env["PUPILOMETER_ENV"] = "test"
|
|
|
|
command = [sys.executable, "-u", "app.py"]
|
|
|
|
with open(STDOUT_FILE, "w") as stdout_f, open(STDERR_FILE, "w") as stderr_f:
|
|
process = subprocess.Popen(
|
|
command,
|
|
cwd="src/controllerSoftware",
|
|
stdout=stdout_f,
|
|
stderr=stderr_f,
|
|
text=True,
|
|
env=env
|
|
)
|
|
|
|
# Wait for the app to start
|
|
start_time = time.time()
|
|
while True:
|
|
if os.path.exists(STDERR_FILE):
|
|
with open(STDERR_FILE, "r") as f:
|
|
if "* Running on http" in f.read():
|
|
break
|
|
if time.time() - start_time > 15:
|
|
raise TimeoutError("Flask app failed to start in time.")
|
|
time.sleep(0.5)
|
|
|
|
yield process
|
|
|
|
process.terminate()
|
|
process.wait()
|
|
|
|
if os.path.exists(STDOUT_FILE):
|
|
os.remove(STDOUT_FILE)
|
|
if os.path.exists(STDERR_FILE):
|
|
os.remove(STDERR_FILE)
|
|
|
|
|
|
def test_program_output(run_app):
|
|
"""
|
|
Tests that the mock backend is initialized.
|
|
"""
|
|
with open(STDOUT_FILE, "r") as f:
|
|
stdout = f.read()
|
|
|
|
assert "Initializing Mock backend for testing..." in stdout
|
|
assert "MockBackend initialized." in stdout
|
|
|
|
|
|
def test_curl_output(run_app):
|
|
"""
|
|
Tests the API endpoints using requests (similar to curl).
|
|
"""
|
|
# Test the /ble_status endpoint
|
|
response_ble = requests.get(f"{BASE_URL}/ble_status")
|
|
assert response_ble.status_code == 200
|
|
assert response_ble.json() == {"connected": True} # In DEBUG_MODE
|
|
|
|
# Test the /vision/pupil_data endpoint
|
|
response_vision = requests.get(f"{BASE_URL}/vision/pupil_data")
|
|
assert response_vision.status_code == 200
|
|
assert "data" in response_vision.json()
|
|
assert "success" in response_vision.json()
|
|
|
|
def test_playwright_checks(page: Page, run_app):
|
|
"""
|
|
Performs basic and visual checks using Playwright.
|
|
"""
|
|
page.goto(BASE_URL)
|
|
|
|
# Basic output check: Title and heading
|
|
expect(page).to_have_title("Lamp Matrix Control")
|
|
heading = page.locator("h1")
|
|
expect(heading).to_have_text("Lamp Matrix Control")
|
|
|
|
# Visual check: Screenshot
|
|
os.makedirs("screenshots", exist_ok=True)
|
|
screenshot_path = "screenshots/homepage.png"
|
|
page.screenshot(path=screenshot_path)
|
|
assert os.path.exists(screenshot_path)
|