conductor(checkpoint): Checkpoint end of Phase 1: Infrastructure & Core Utilities
This commit is contained in:
73
tests/conftest.py
Normal file
73
tests/conftest.py
Normal file
@@ -0,0 +1,73 @@
|
||||
import pytest
|
||||
import subprocess
|
||||
import time
|
||||
import requests
|
||||
import os
|
||||
import signal
|
||||
|
||||
def kill_process_tree(pid):
|
||||
"""Robustly kills a process and all its children."""
|
||||
if pid is None:
|
||||
return
|
||||
try:
|
||||
print(f"[Fixture] Attempting to kill process tree for PID {pid}...")
|
||||
if os.name == 'nt':
|
||||
# /F is force, /T is tree (includes children)
|
||||
subprocess.run(["taskkill", "/F", "/T", "/PID", str(pid)],
|
||||
stdout=subprocess.DEVNULL,
|
||||
stderr=subprocess.DEVNULL,
|
||||
check=False)
|
||||
else:
|
||||
# On Unix, kill the process group
|
||||
os.killpg(os.getpgid(pid), signal.SIGKILL)
|
||||
print(f"[Fixture] Process tree {pid} killed.")
|
||||
except Exception as e:
|
||||
print(f"[Fixture] Error killing process tree {pid}: {e}")
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def live_gui():
|
||||
"""
|
||||
Session-scoped fixture that starts gui.py with --enable-test-hooks.
|
||||
Ensures the GUI is running before tests start and shuts it down after.
|
||||
"""
|
||||
print("\n[Fixture] Starting gui.py --enable-test-hooks...")
|
||||
|
||||
# Start gui.py as a subprocess.
|
||||
process = subprocess.Popen(
|
||||
["uv", "run", "python", "gui.py", "--enable-test-hooks"],
|
||||
stdout=subprocess.DEVNULL,
|
||||
stderr=subprocess.DEVNULL,
|
||||
text=True,
|
||||
creationflags=subprocess.CREATE_NEW_PROCESS_GROUP if os.name == 'nt' else 0
|
||||
)
|
||||
|
||||
# Wait for the hook server to be ready (Port 8999 per api_hooks.py)
|
||||
max_retries = 5
|
||||
ready = False
|
||||
print(f"[Fixture] Waiting up to {max_retries}s for Hook Server on port 8999...")
|
||||
|
||||
start_time = time.time()
|
||||
while time.time() - start_time < max_retries:
|
||||
try:
|
||||
# Using /status endpoint defined in HookHandler
|
||||
response = requests.get("http://127.0.0.1:8999/status", timeout=0.5)
|
||||
if response.status_code == 200:
|
||||
ready = True
|
||||
print(f"[Fixture] GUI Hook Server is ready after {round(time.time() - start_time, 2)}s.")
|
||||
break
|
||||
except (requests.exceptions.ConnectionError, requests.exceptions.Timeout):
|
||||
if process.poll() is not None:
|
||||
print("[Fixture] Process died unexpectedly during startup.")
|
||||
break
|
||||
time.sleep(0.5)
|
||||
|
||||
if not ready:
|
||||
print("[Fixture] TIMEOUT/FAILURE: Hook server failed to respond on port 8999 within 5s. Cleaning up...")
|
||||
kill_process_tree(process.pid)
|
||||
pytest.fail("Failed to start gui.py with test hooks within 5 seconds.")
|
||||
|
||||
try:
|
||||
yield process
|
||||
finally:
|
||||
print("\n[Fixture] Finally block triggered: Shutting down gui.py...")
|
||||
kill_process_tree(process.pid)
|
||||
28
tests/test_api_hook_client_enhanced.py
Normal file
28
tests/test_api_hook_client_enhanced.py
Normal file
@@ -0,0 +1,28 @@
|
||||
import pytest
|
||||
import time
|
||||
import sys
|
||||
import os
|
||||
import requests
|
||||
|
||||
# Ensure project root is in path
|
||||
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
||||
|
||||
from api_hook_client import ApiHookClient
|
||||
|
||||
def test_client_wait_for_server(live_gui):
|
||||
"""Verifies that the client can wait for the server to become ready."""
|
||||
client = ApiHookClient()
|
||||
assert client.wait_for_server(timeout=5) is True
|
||||
|
||||
def test_client_get_performance_retry(live_gui):
|
||||
"""Verifies that the client can retrieve metrics correctly."""
|
||||
client = ApiHookClient()
|
||||
perf = client.get_performance()
|
||||
assert "performance" in perf
|
||||
|
||||
def test_client_connection_error():
|
||||
"""Verifies that the client raises a Connection or Timeout error when the server is down."""
|
||||
# Use a port that is unlikely to be in use and not intercepted
|
||||
client = ApiHookClient(base_url="http://127.0.0.1:9998", max_retries=1, retry_delay=0.1)
|
||||
with pytest.raises((requests.exceptions.ConnectionError, requests.exceptions.Timeout)):
|
||||
client.get_project()
|
||||
21
tests/test_gui_fixture_green.py
Normal file
21
tests/test_gui_fixture_green.py
Normal file
@@ -0,0 +1,21 @@
|
||||
import pytest
|
||||
import requests
|
||||
|
||||
def test_gui_fixture_auto_starts(live_gui):
|
||||
"""
|
||||
Verifies that the live_gui fixture correctly starts the GUI
|
||||
and the hook server is reachable on port 8999.
|
||||
"""
|
||||
response = requests.get("http://127.0.0.1:8999/status")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["status"] == "ok"
|
||||
|
||||
def test_get_performance_metrics(live_gui):
|
||||
"""
|
||||
Verifies that we can retrieve performance metrics via the hook server.
|
||||
"""
|
||||
response = requests.get("http://127.0.0.1:8999/api/performance")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert "performance" in data
|
||||
13
tests/test_gui_fixture_red.py
Normal file
13
tests/test_gui_fixture_red.py
Normal file
@@ -0,0 +1,13 @@
|
||||
import pytest
|
||||
import subprocess
|
||||
import time
|
||||
import requests
|
||||
|
||||
def test_gui_fixture_auto_starts():
|
||||
# This test should fail if the fixture isn't working yet.
|
||||
# It attempts to reach the hook server without starting it manually.
|
||||
try:
|
||||
response = requests.post("http://localhost:5000/get_ui_performance", json={})
|
||||
assert response.status_code == 200
|
||||
except requests.exceptions.ConnectionError:
|
||||
pytest.fail("Hook server is not running. Fixture failed or is missing.")
|
||||
Reference in New Issue
Block a user