mirror of
https://github.com/0xWheatyz/handler.git
synced 2026-08-30 17:26:25 +00:00
Claude
eba0e19ec9
Implements the Phase 1 MVP from the README: a stateless control layer + HTTP API over a centralized database, with hook-enforced test/push gates. - DB layer: SQLAlchemy Core, one schema rendering both Postgres (BIGSERIAL / TIMESTAMPTZ / JSONB) and SQLite (INTEGER PK / TEXT / JSON) via portable types; native ON CONFLICT DO UPDATE checkmark upsert on both dialects. - Alembic dual-dialect migrations (render_as_batch for SQLite); tests run a real `alembic upgrade head`. - FastAPI: projects/agents/checkmark/log/answer/resume + shared log/context routes, single global bearer token, higher-trust token gating shared-context writes, project isolation on every route. - Hooks (`python -m handler.hooks <event>`): Stop test gate (block on red), PreToolUse AskUserQuestion defer + `git push` gate (tests then throwaway build), Notification generic webhook (no-op without WEBHOOK_URL). Identity via env injected at spawn; verify is the mock seam. - Control CLI: spawn/list/attach/kill, hard `.mise.toml [tasks.test]` gate, generated per-agent settings.json, identity + DATABASE_URL injected via tmux; tmux is the mock seam. - 45 tests (SQLite), ruff clean. Live claude/tmux/mise spawning deferred behind the mocked seams. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01W5ZuS5pV1NS6eKsRZHXonY
108 lines
3.0 KiB
Python
108 lines
3.0 KiB
Python
"""Shared fixtures. Everything runs on a fresh SQLite file per test, materialized via
|
|
a *real* ``alembic upgrade head`` — so the migration path itself is under test, not
|
|
just ``create_all``. No live claude/tmux/mise is ever touched: the three seams
|
|
(``control.tmux``, ``hooks.verify``, ``control.spawn.resume``) are faked.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from alembic import command
|
|
from alembic.config import Config
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
def _reset_caches() -> None:
|
|
from handler import config
|
|
from handler.db import engine
|
|
|
|
config.get_settings.cache_clear()
|
|
engine.get_engine.cache_clear()
|
|
|
|
|
|
@pytest.fixture
|
|
def env(tmp_path, monkeypatch):
|
|
"""Point every entrypoint at a fresh SQLite db + a known token, migrated."""
|
|
db_path = tmp_path / "handler.db"
|
|
url = f"sqlite:///{db_path}"
|
|
monkeypatch.setenv("DATABASE_URL", url)
|
|
monkeypatch.setenv("AUTH_TOKEN", "test-token")
|
|
monkeypatch.setenv("SHARED_CONTEXT_WRITE_TOKEN", "shared-token")
|
|
monkeypatch.setenv("PROJECTS_ROOT", str(tmp_path / "projects"))
|
|
monkeypatch.delenv("WEBHOOK_URL", raising=False)
|
|
_reset_caches()
|
|
|
|
cfg = Config(str(REPO_ROOT / "alembic.ini"))
|
|
cfg.set_main_option("script_location", str(REPO_ROOT / "src" / "handler" / "migrations"))
|
|
command.upgrade(cfg, "head")
|
|
|
|
yield {"url": url, "token": "test-token", "shared_token": "shared-token", "tmp": tmp_path}
|
|
|
|
_reset_caches()
|
|
|
|
|
|
@pytest.fixture
|
|
def engine(env):
|
|
from handler.db.engine import get_engine
|
|
|
|
return get_engine()
|
|
|
|
|
|
@pytest.fixture
|
|
def conn(engine):
|
|
with engine.begin() as c:
|
|
yield c
|
|
|
|
|
|
@pytest.fixture
|
|
def client(env):
|
|
from fastapi.testclient import TestClient
|
|
|
|
from handler.api.app import create_app
|
|
|
|
return TestClient(create_app())
|
|
|
|
|
|
@pytest.fixture
|
|
def auth(env):
|
|
return {"Authorization": f"Bearer {env['token']}"}
|
|
|
|
|
|
@pytest.fixture
|
|
def fake_tmux(monkeypatch):
|
|
"""Record tmux calls instead of spawning; report sessions as live by default."""
|
|
calls: dict[str, list] = {"new_session": [], "kill_session": [], "send_keys": []}
|
|
live: set[str] = set()
|
|
|
|
from handler.control import tmux
|
|
|
|
def new_session(name, cwd, command, env):
|
|
calls["new_session"].append(
|
|
{"name": name, "cwd": cwd, "command": command, "env": env}
|
|
)
|
|
live.add(name)
|
|
|
|
def has_session(name):
|
|
return name in live
|
|
|
|
def kill_session(name):
|
|
calls["kill_session"].append(name)
|
|
live.discard(name)
|
|
|
|
def send_keys(name, keys):
|
|
calls["send_keys"].append({"name": name, "keys": keys})
|
|
|
|
def list_sessions():
|
|
return list(live)
|
|
|
|
monkeypatch.setattr(tmux, "new_session", new_session)
|
|
monkeypatch.setattr(tmux, "has_session", has_session)
|
|
monkeypatch.setattr(tmux, "kill_session", kill_session)
|
|
monkeypatch.setattr(tmux, "send_keys", send_keys)
|
|
monkeypatch.setattr(tmux, "list_sessions", list_sessions)
|
|
|
|
return {"calls": calls, "live": live}
|