File size: 3,145 Bytes
638c806 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | from __future__ import annotations
import pytest
from conftest import import_any, install_fake_adapter, maybe_import, text_blob, unload_promptboost_modules
API_MODULES = (
"promptboost.api",
"promptboost.app",
"promptboost.main",
"promptboost.server",
)
def _fake_boost(*_args, **_kwargs):
return {
"run_id": "test-run",
"boosted_prompt": "API boosted prompt without paid API calls.",
"target_harness": _kwargs.get("target_harness") or _kwargs.get("harness") or "codex",
"target_model": _kwargs.get("target_model") or _kwargs.get("model") or "test-model",
}
def _patch_api_boost(monkeypatch):
for module_name in (
"promptboost",
"promptboost.core",
"promptboost.boost",
"promptboost.api",
"promptboost.app",
"promptboost.main",
"promptboost.server",
):
module = maybe_import(module_name)
if module is not None:
monkeypatch.setattr(module, "boost", _fake_boost, raising=False)
def _load_app(monkeypatch):
api_module = import_any(*API_MODULES)
_patch_api_boost(monkeypatch)
create_app = getattr(api_module, "create_app", None)
if callable(create_app):
return create_app()
app = getattr(api_module, "app", None)
if app is None:
pytest.fail(
"FastAPI module must expose either app or create_app() for route tests."
)
return app
def _request_first_existing(client, method: str, paths: list[str], **kwargs):
for path in paths:
response = getattr(client, method)(path, **kwargs)
if response.status_code != 404:
return path, response
pytest.fail(f"None of the expected API routes exist: {', '.join(paths)}")
def test_api_health_route(promptboost_db_path, monkeypatch):
testclient = pytest.importorskip("fastapi.testclient")
unload_promptboost_modules()
install_fake_adapter(monkeypatch, "API boosted prompt without paid API calls.")
app = _load_app(monkeypatch)
client = testclient.TestClient(app)
path, response = _request_first_existing(
client,
"get",
["/health", "/healthz", "/api/health", "/"],
)
assert response.status_code == 200, path
assert any(token in text_blob(response.json()) for token in ("ok", "healthy", "promptboost"))
def test_api_boost_route_returns_boosted_prompt(promptboost_db_path, monkeypatch):
testclient = pytest.importorskip("fastapi.testclient")
unload_promptboost_modules()
install_fake_adapter(monkeypatch, "API boosted prompt without paid API calls.")
app = _load_app(monkeypatch)
client = testclient.TestClient(app)
_path, response = _request_first_existing(
client,
"post",
["/boost", "/api/boost", "/v1/boost"],
json={
"raw_prompt": "Improve this API prompt without calling paid services.",
"target_harness": "codex",
"target_model": "test-model",
},
)
assert response.status_code in (200, 201)
assert "api boosted prompt without paid api calls" in text_blob(response.json())
|