132 lines
3.6 KiB
Python
132 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
from fastapi.testclient import TestClient
|
|
from main import create_app
|
|
from models import AgentResponse, LevelInfo
|
|
|
|
|
|
def write_config(path) -> None:
|
|
path.write_text(
|
|
"""
|
|
[llm]
|
|
model = "test-model"
|
|
api_key = "test-key"
|
|
|
|
[server]
|
|
host = "127.0.0.1"
|
|
port = 8000
|
|
reload = false
|
|
|
|
[game]
|
|
hard_mode_rotation_interval = 5
|
|
level2_output_blocked_response_text = "blocked-2-out"
|
|
level3_input_blocked_response_text = "blocked-3-in"
|
|
level3_output_blocked_response_text = "blocked-3-out"
|
|
level4_input_blocked_response_text = "blocked-4-in"
|
|
level4_output_blocked_response_text = "blocked-4-out"
|
|
level5_input_blocked_response_text = "blocked-5-in"
|
|
level5_output_blocked_response_text = "blocked-5-out"
|
|
level5_exchange_blocked_response_text = "blocked-5-pair"
|
|
password_words = [
|
|
"w01", "w02", "w03", "w04", "w05",
|
|
"w06", "w07", "w08", "w09", "w10",
|
|
"w11", "w12", "w13", "w14", "w15",
|
|
"w16", "w17", "w18", "w19", "w20",
|
|
]
|
|
""".strip(),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
|
|
def test_create_app_exposes_level_and_query_routes(monkeypatch, tmp_path) -> None:
|
|
config_path = tmp_path / "config.toml"
|
|
write_config(config_path)
|
|
|
|
class FakeService:
|
|
def __init__(self, settings) -> None:
|
|
self.settings = settings
|
|
|
|
def list_levels(self) -> list[LevelInfo]:
|
|
return [
|
|
LevelInfo(
|
|
id=1,
|
|
title="Level 1",
|
|
description="Simple agent",
|
|
)
|
|
]
|
|
|
|
def run_level(
|
|
self, level_id, session_id, user_text, hard_mode
|
|
) -> AgentResponse:
|
|
return AgentResponse(
|
|
session_id=session_id,
|
|
response_text=f"echo:{user_text}:{hard_mode}",
|
|
success=False,
|
|
session_rotated=False,
|
|
level_id=level_id,
|
|
)
|
|
|
|
monkeypatch.setattr("main.AgentService", FakeService)
|
|
app = create_app(config_path)
|
|
client = TestClient(app)
|
|
|
|
levels_response = client.get("/api/v1/levels")
|
|
query_response = client.post(
|
|
"/api/v1/levels/query/1/",
|
|
json={
|
|
"session_id": "00000000-0000-0000-0000-000000000111",
|
|
"text": "hello",
|
|
"hard_mode": True,
|
|
},
|
|
)
|
|
|
|
assert levels_response.status_code == 200
|
|
assert levels_response.json() == [
|
|
{
|
|
"id": 1,
|
|
"title": "Level 1",
|
|
"description": "Simple agent",
|
|
}
|
|
]
|
|
assert query_response.status_code == 200
|
|
assert query_response.json() == {
|
|
"session_id": "00000000-0000-0000-0000-000000000111",
|
|
"response_text": "echo:hello:True",
|
|
"success": False,
|
|
"session_rotated": False,
|
|
"level_id": 1,
|
|
}
|
|
|
|
|
|
def test_query_route_returns_404_for_unknown_level(monkeypatch, tmp_path) -> None:
|
|
config_path = tmp_path / "config.toml"
|
|
write_config(config_path)
|
|
|
|
class FakeService:
|
|
def __init__(self, settings) -> None:
|
|
self.settings = settings
|
|
|
|
def list_levels(self) -> list[LevelInfo]:
|
|
return []
|
|
|
|
def run_level(
|
|
self, level_id, session_id, user_text, hard_mode
|
|
) -> AgentResponse:
|
|
raise ValueError(f"Unknown level_id={level_id}")
|
|
|
|
monkeypatch.setattr("main.AgentService", FakeService)
|
|
app = create_app(config_path)
|
|
client = TestClient(app)
|
|
|
|
response = client.post(
|
|
"/api/v1/levels/query/99/",
|
|
json={
|
|
"session_id": "00000000-0000-0000-0000-000000000222",
|
|
"text": "hello",
|
|
"hard_mode": False,
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 404
|
|
assert response.json() == {"detail": "Unknown level_id=99"}
|