feat: add Python integration tests for HTTP endpoints
This commit is contained in:
@@ -0,0 +1,31 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""Run integration tests for pipeline-runner."""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
project_dir = os.path.dirname(os.path.abspath(__file__))
|
||||||
|
tests_dir = os.path.join(project_dir, "tests")
|
||||||
|
|
||||||
|
# Build in FAKE mode first
|
||||||
|
build_script = os.path.join(project_dir, "build.sh")
|
||||||
|
print("==> Building in FAKE mode...")
|
||||||
|
result = subprocess.run([build_script, "--fake"], cwd=project_dir)
|
||||||
|
if result.returncode != 0:
|
||||||
|
print("Build failed", file=sys.stderr)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
# Run pytest
|
||||||
|
print("==> Running integration tests...")
|
||||||
|
result = subprocess.run(
|
||||||
|
[sys.executable, "-m", "pytest", tests_dir, "-v"],
|
||||||
|
cwd=project_dir,
|
||||||
|
)
|
||||||
|
sys.exit(result.returncode)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
@@ -0,0 +1,102 @@
|
|||||||
|
"""Pytest fixture for managing the pipeline-runner server lifecycle during tests."""
|
||||||
|
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import shutil
|
||||||
|
import socket
|
||||||
|
import subprocess
|
||||||
|
import tempfile
|
||||||
|
import time
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
|
def get_free_port():
|
||||||
|
"""Get a free port for the test server."""
|
||||||
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||||
|
s.bind(("", 0))
|
||||||
|
port = s.getsockname()[1]
|
||||||
|
s.close()
|
||||||
|
return port
|
||||||
|
|
||||||
|
|
||||||
|
class PipelineServer:
|
||||||
|
"""Manages a pipeline-runner server instance for tests."""
|
||||||
|
|
||||||
|
def __init__(self, port, work_dir, verbose=False):
|
||||||
|
self.port = port
|
||||||
|
self.work_dir = work_dir
|
||||||
|
self.verbose = verbose
|
||||||
|
|
||||||
|
test_dir = os.path.dirname(os.path.abspath(__file__))
|
||||||
|
project_root = os.path.dirname(test_dir)
|
||||||
|
self.bin_path = os.path.join(project_root, "build", "pipeline-runner")
|
||||||
|
|
||||||
|
config = {
|
||||||
|
"port": port,
|
||||||
|
"pipelines_dir": os.path.join(work_dir, "pipelines") + "/",
|
||||||
|
"logs_dir": os.path.join(work_dir, "logs") + "/",
|
||||||
|
}
|
||||||
|
self.config_path = os.path.join(work_dir, "pipeline-runner.conf")
|
||||||
|
with open(self.config_path, "w") as f:
|
||||||
|
json.dump(config, f)
|
||||||
|
|
||||||
|
self.process = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def base_url(self):
|
||||||
|
return f"http://localhost:{self.port}"
|
||||||
|
|
||||||
|
def start(self, timeout=30):
|
||||||
|
"""Start the server and wait until it responds."""
|
||||||
|
self.process = subprocess.Popen(
|
||||||
|
[self.bin_path, self.config_path],
|
||||||
|
cwd=self.work_dir,
|
||||||
|
stdout=subprocess.PIPE,
|
||||||
|
stderr=subprocess.PIPE,
|
||||||
|
)
|
||||||
|
|
||||||
|
deadline = time.time() + timeout
|
||||||
|
while time.time() < deadline:
|
||||||
|
try:
|
||||||
|
resp = __import__("requests").get(
|
||||||
|
f"{self.base_url}/api/pipelines", timeout=1
|
||||||
|
)
|
||||||
|
if resp.status_code == 200:
|
||||||
|
return True
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
time.sleep(0.3)
|
||||||
|
|
||||||
|
return False
|
||||||
|
|
||||||
|
def stop(self):
|
||||||
|
"""Stop the server."""
|
||||||
|
if self.process:
|
||||||
|
self.process.terminate()
|
||||||
|
try:
|
||||||
|
self.process.wait(timeout=10)
|
||||||
|
except subprocess.TimeoutExpired:
|
||||||
|
self.process.kill()
|
||||||
|
self.process.wait(timeout=5)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope="session")
|
||||||
|
def pipeline_server():
|
||||||
|
"""Session-scoped fixture that builds, starts, and stops the pipeline-runner server."""
|
||||||
|
project_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||||||
|
build_script = os.path.join(project_dir, "build.sh")
|
||||||
|
|
||||||
|
subprocess.check_call([build_script, "--fake"], cwd=project_dir)
|
||||||
|
|
||||||
|
work_dir = tempfile.mkdtemp(prefix="pipeline-test-")
|
||||||
|
port = get_free_port()
|
||||||
|
|
||||||
|
server = PipelineServer(port, work_dir)
|
||||||
|
if not server.start():
|
||||||
|
pytest.fail("Could not start pipeline-runner server")
|
||||||
|
|
||||||
|
yield server
|
||||||
|
|
||||||
|
server.stop()
|
||||||
|
shutil.rmtree(work_dir, ignore_errors=True)
|
||||||
@@ -0,0 +1,85 @@
|
|||||||
|
"""Integration tests for the full pipeline CRUD + execution flow."""
|
||||||
|
|
||||||
|
import time
|
||||||
|
import uuid
|
||||||
|
|
||||||
|
import requests
|
||||||
|
|
||||||
|
|
||||||
|
def _unique_id():
|
||||||
|
return str(uuid.uuid4())
|
||||||
|
|
||||||
|
|
||||||
|
def test_full_pipeline_lifecycle(pipeline_server):
|
||||||
|
"""Create, list, get, run, poll, log, delete a pipeline end-to-end."""
|
||||||
|
base = pipeline_server.base_url
|
||||||
|
pipeline_id = _unique_id()
|
||||||
|
|
||||||
|
# --- Create pipeline ---
|
||||||
|
pipeline = {
|
||||||
|
"id": pipeline_id,
|
||||||
|
"name": "integration-test-pipeline",
|
||||||
|
"working_dir": "/tmp",
|
||||||
|
"prompts": [
|
||||||
|
{"id": _unique_id(), "text": "echo step1", "title": "Step 1", "order": 0},
|
||||||
|
{"id": _unique_id(), "text": "echo step2", "title": "Step 2", "order": 1},
|
||||||
|
{"id": _unique_id(), "text": "echo step3", "title": "Step 3", "order": 2},
|
||||||
|
],
|
||||||
|
}
|
||||||
|
resp = requests.post(f"{base}/api/pipelines", json=pipeline)
|
||||||
|
assert resp.status_code == 200
|
||||||
|
created = resp.json()
|
||||||
|
assert created["id"] == pipeline_id
|
||||||
|
assert created["name"] == "integration-test-pipeline"
|
||||||
|
assert len(created["prompts"]) == 3
|
||||||
|
|
||||||
|
# --- List pipelines ---
|
||||||
|
resp = requests.get(f"{base}/api/pipelines")
|
||||||
|
assert resp.status_code == 200
|
||||||
|
pipelines = resp.json()
|
||||||
|
assert any(p["id"] == pipeline_id for p in pipelines)
|
||||||
|
|
||||||
|
# --- Get single pipeline ---
|
||||||
|
resp = requests.get(f"{base}/api/pipelines/{pipeline_id}")
|
||||||
|
assert resp.status_code == 200
|
||||||
|
fetched = resp.json()
|
||||||
|
assert fetched["id"] == pipeline_id
|
||||||
|
assert fetched["name"] == "integration-test-pipeline"
|
||||||
|
assert len(fetched["prompts"]) == 3
|
||||||
|
assert fetched["prompts"][0]["order"] == 0
|
||||||
|
|
||||||
|
# --- Start a run ---
|
||||||
|
resp = requests.post(f"{base}/api/runs", json={"pipeline_id": pipeline_id})
|
||||||
|
assert resp.status_code == 200
|
||||||
|
run_data = resp.json()
|
||||||
|
run_id = run_data["run_id"]
|
||||||
|
assert run_id
|
||||||
|
assert run_data["status"] == "running"
|
||||||
|
|
||||||
|
# --- Poll run status until completed ---
|
||||||
|
for _ in range(60):
|
||||||
|
resp = requests.get(f"{base}/api/runs/{run_id}/status")
|
||||||
|
assert resp.status_code == 200
|
||||||
|
status_data = resp.json()
|
||||||
|
if status_data["status"] in ("completed", "error"):
|
||||||
|
break
|
||||||
|
time.sleep(0.5)
|
||||||
|
else:
|
||||||
|
pytest.fail("Run did not complete within timeout")
|
||||||
|
|
||||||
|
assert status_data["status"] == "completed"
|
||||||
|
assert len(status_data["steps"]) == 3
|
||||||
|
|
||||||
|
# --- Get log ---
|
||||||
|
resp = requests.get(f"{base}/api/runs/{run_id}/log")
|
||||||
|
assert resp.status_code == 200
|
||||||
|
assert len(resp.text) > 0
|
||||||
|
|
||||||
|
# --- Delete pipeline ---
|
||||||
|
resp = requests.delete(f"{base}/api/pipelines/{pipeline_id}")
|
||||||
|
assert resp.status_code == 200
|
||||||
|
assert resp.json()["deleted"] == pipeline_id
|
||||||
|
|
||||||
|
# --- Verify deletion ---
|
||||||
|
resp = requests.get(f"{base}/api/pipelines/{pipeline_id}")
|
||||||
|
assert resp.status_code == 404
|
||||||
@@ -0,0 +1,26 @@
|
|||||||
|
"""Smoke tests for pipeline-runner REST API endpoints."""
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
import requests
|
||||||
|
|
||||||
|
|
||||||
|
test_get_endpoints = [
|
||||||
|
("api/pipelines", 200, list),
|
||||||
|
("api/runs/nonexistent-id/status", 404, dict),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize("endpoint,expected_status,expected_type", test_get_endpoints)
|
||||||
|
def test_get_endpoints(pipeline_server, endpoint, expected_status, expected_type):
|
||||||
|
"""Check HTTP status code and response content-type for GET endpoints."""
|
||||||
|
response = requests.get(f"{pipeline_server.base_url}/{endpoint}")
|
||||||
|
assert response.status_code == expected_status
|
||||||
|
assert "application/json" in response.headers.get("Content-Type", "")
|
||||||
|
data = response.json()
|
||||||
|
assert isinstance(data, expected_type)
|
||||||
|
|
||||||
|
|
||||||
|
def test_cors_headers(pipeline_server):
|
||||||
|
"""Check that CORS headers are present on responses."""
|
||||||
|
response = requests.get(f"{pipeline_server.base_url}/api/pipelines")
|
||||||
|
assert response.headers.get("Access-Control-Allow-Origin") == "*"
|
||||||
Reference in New Issue
Block a user