tests/test_api_reset.py

from fastapi.testclient import TestClient

from server import api
from server.constants import TaskName
from server.models import Observation, SystemMetrics


class _FakeEnv:
    def __init__(self) -> None:
        self.reset_calls: list[TaskName] = []

    def start(self) -> None:
        return None

    def close(self) -> None:
        return None

    def reset(self, task_name: TaskName) -> Observation:
        self.reset_calls.append(task_name)
        return Observation(
            command_output="ready",
            metrics=SystemMetrics(
                gateway_success_rate=0.0,
                gateway_p99_latency_ms=0.0,
                queue_depth=0,
                worker_restart_count=0,
                consumer_stall_count=0,
            ),
            process_status={"gateway": "running"},
        )


def test_reset_defaults_to_cascading_timeout_when_task_missing(monkeypatch) -> None:
    holder: dict[str, _FakeEnv] = {}

    def fake_env_factory() -> _FakeEnv:
        env = _FakeEnv()
        holder["env"] = env
        return env

    monkeypatch.setattr(api, "DistributedDebugEnv", fake_env_factory)

    with TestClient(api.app) as client:
        response = client.post("/reset", json={})

    assert response.status_code == 200
    assert holder["env"].reset_calls == [TaskName.CASCADING_TIMEOUT]


def test_reset_rejects_unknown_explicit_task(monkeypatch) -> None:
    holder: dict[str, _FakeEnv] = {}

    def fake_env_factory() -> _FakeEnv:
        env = _FakeEnv()
        holder["env"] = env
        return env

    monkeypatch.setattr(api, "DistributedDebugEnv", fake_env_factory)

    with TestClient(api.app) as client:
        response = client.post("/reset", params={"task_name": "not-a-task"}, json={})

    assert response.status_code == 400
    assert response.json()["detail"] == "Unknown task: not-a-task"
    assert holder["env"].reset_calls == []