protokeyandClaude Opus 5.5
1a4bd8d672
- Postgres в compose; пакет db (SQLAlchemy async): clients, windows, requests. - Все эндпоинты, кроме /health, — только с ключом в X-API-Key, иначе 401; без выпущенных ключей доступа нет. В базе — только sha256 ключа. - Задача и окно сессии привязаны к клиенту: чужой ключ их не видит. - Воркер пишет журнал: окно (открыто, последнее обращение, закрыто) и задачу (в очереди, начало, конец, итог, данные без секретов). Сбой записи не роняет задачу. - python -m db.cli create|list|revoke|stats; make key-create/key-list/key-revoke/stats. - smoke и loadtest с ключом. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
86 lines
3.3 KiB
Python
86 lines
3.3 KiB
Python
"""Проверка запущенного стенда (make up): api, очередь, воркер с Chrome, удержание окна.
|
|
|
|
make smoke API_KEY=rbx_... # http://localhost:8000
|
|
make smoke API_KEY=rbx_... API_URL=https://app.robotex.sokolab.xyz
|
|
|
|
Ключ: make key-create name=smoke.
|
|
"""
|
|
import sys
|
|
import time
|
|
|
|
import httpx
|
|
|
|
URL = "https://example.com"
|
|
READ_TITLE = [{"type": "inner_html", "selector": "title"}]
|
|
|
|
|
|
def check(cond: bool, what: str) -> None:
|
|
print((" ok " if cond else " FAIL ") + what)
|
|
if not cond:
|
|
sys.exit(1)
|
|
|
|
|
|
def wait_task(client: httpx.Client, task_id: str, timeout: float = 180) -> dict:
|
|
deadline = time.monotonic() + timeout
|
|
while time.monotonic() < deadline:
|
|
task = client.get(f"/v2/tasks/{task_id}").json()
|
|
if task["status"] not in ("queued", "running"):
|
|
return task
|
|
time.sleep(0.5)
|
|
raise TimeoutError(f"task {task_id} not finished in {timeout}s")
|
|
|
|
|
|
def windows(client: httpx.Client) -> int:
|
|
return sum(w["windows"] for w in client.get("/health").json()["workers"])
|
|
|
|
|
|
def main(base_url: str, api_key: str) -> None:
|
|
headers = {"X-API-Key": api_key}
|
|
client = httpx.Client(base_url=base_url, timeout=300, headers=headers)
|
|
|
|
print("auth")
|
|
resp = httpx.post(f"{base_url}/v1/solve", json={"url": URL, "actions": []}, timeout=30)
|
|
check(resp.status_code == 401, f"no key -> {resp.status_code}")
|
|
|
|
print("health")
|
|
health = client.get("/health").json()
|
|
check(len(health["workers"]) > 0, f"workers: {[w['worker_id'] for w in health['workers']]}, free slots {health['free_slots']}")
|
|
|
|
print("v1 /v1/solve")
|
|
started = time.monotonic()
|
|
data = client.post("/v1/solve", json={"url": URL, "actions": READ_TITLE}).json()
|
|
check(data["status"] == "ok", f"status {data['status']} {data.get('error', '')} ({time.monotonic() - started:.1f}s)")
|
|
check(data["actions"][0]["result"] == "Example Domain", "title = Example Domain")
|
|
|
|
print("v2 window held between tasks")
|
|
v2 = httpx.Client(base_url=base_url, timeout=60, headers=headers) # своя сессия
|
|
started = time.monotonic()
|
|
created = v2.post("/v2/tasks", json={"url": URL, "actions": READ_TITLE, "browser_idle_timeout": 2})
|
|
check(created.status_code == 202, f"POST /v2/tasks -> {created.status_code}")
|
|
task = wait_task(v2, created.json()["task_id"])
|
|
check(task["status"] == "done" and task["result"]["status"] == "ok", f"first task done ({time.monotonic() - started:.1f}s)")
|
|
check(windows(client) >= 1, "window stays open")
|
|
|
|
started = time.monotonic()
|
|
created = v2.post("/v2/tasks", json={"actions": READ_TITLE, "browser_idle_timeout": 2})
|
|
task = wait_task(v2, created.json()["task_id"])
|
|
check(
|
|
task["status"] == "done" and task["result"]["actions"][0]["result"] == "Example Domain",
|
|
f"second task without url ran in the open window ({time.monotonic() - started:.1f}s)",
|
|
)
|
|
|
|
check(client.get(f"/v2/tasks/{created.json()['task_id']}").status_code == 404, "task hidden from other sessions")
|
|
|
|
check(v2.delete("/v2/sessions/current").json() == {"closed": True}, "DELETE /v2/sessions/current")
|
|
deadline = time.monotonic() + 15
|
|
while windows(client) and time.monotonic() < deadline:
|
|
time.sleep(0.5)
|
|
check(windows(client) == 0, "window closed, slot released")
|
|
print("all good")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) < 3 or not sys.argv[2]:
|
|
sys.exit("usage: smoke.py <api url> <api key> (make smoke API_KEY=...)")
|
|
main(sys.argv[1], sys.argv[2])
|