tests/test_webapp.py

import io
import json
import zipfile

from conftest import complete
from ros_fixtures import INTERFACES, as_file_blocks, python_package

from api.ros_check import load_interfaces


def test_app_page_and_assets_are_served(run_env):
    client = run_env[0]
    page = client.get("/")
    assert page.status_code == 200 and "RosDiff" in page.text and "/static/app/app.js" in page.text
    assert client.get("/static/app/app.js").status_code == 200
    assert client.get("/static/app/app.css").status_code == 200


def test_history_generation_and_media(run_env):
    client, keys_, backend, services, gen_id = run_env
    h = {"X-API-Key": keys_["alice"]}
    hist = client.get("/me/history", headers=h).json()
    assert hist["tester"] == "alice" and hist["items"][0]["id"] == gen_id
    assert hist["items"][0]["kind"] == "simulate" and hist["items"][0]["title"].startswith("G1 near a slope")
    assert client.get("/me/history", headers={"X-API-Key": keys_["bob"]}).json()["items"] == []

    run_id = client.post("/simulate/run", json={"generation_id": gen_id}, headers=h).json()["run_id"]
    complete(backend, "job-1", run_id)
    job = client.get(f"/jobs/{run_id}", headers=h).json()
    assert job["media"]["video"].endswith(f"runs/{run_id}/video.mp4") and "video_webm" in job["media"]

    g = client.get(f"/generations/{gen_id}", headers=h).json()
    assert g["kind"] == "simulate" and g["valid"] and "<mujoco" in g["output"]
    assert [j["id"] for j in g["jobs"]] == [run_id] and "/report/" in g["jobs"][0]["report_url"]
    assert client.get(f"/generations/{gen_id}", headers={"X-API-Key": keys_["bob"]}).status_code == 404


def test_code_zip(run_env):
    client, keys_, backend, services, _ = run_env
    services.settings.data_dir.mkdir(parents=True, exist_ok=True)
    services.settings.interfaces_file.write_text(json.dumps(INTERFACES))
    load_interfaces.cache_clear()
    services.llm_factory().responses[:] = [as_file_blocks(python_package())]
    h = {"X-API-Key": keys_["alice"]}
    gen = client.post("/code/generate", json={"request": "Stop on a close LaserScan reading"}, headers=h).json()
    r = client.get(f"/code/{gen['id']}/zip", headers=h)
    assert r.status_code == 200 and r.headers["content-type"] == "application/zip"
    names = zipfile.ZipFile(io.BytesIO(r.content)).namelist()
    assert "safety_stop/package.xml" in names
    assert client.get(f"/code/{gen['id']}/zip", headers={"X-API-Key": keys_["bob"]}).status_code == 404
    g = client.get(f"/generations/{gen['id']}", headers=h).json()
    assert g["kind"] == "code" and "safety_stop/package.xml" in g["output"]