Files
dev-puppeteer/my-deepagent/tests/unit/test_slash.py
chungyeong 733c9be0bd feat(my-deepagent): v0.1.0 Step 6~15 — REPL/Budget/Recovery/Audit/Pricing + real OpenRouter E2E
Step 6  — Distribution: init/login/logout/keys/doctor CLI, platformdirs data dirs,
          OS keyring (Keychain/Secret Service/Credential Store), first-run governance
          consent, secret resolution chain (config→env→keyring), ko/en i18n catalog
          via MYDEEPAGENT_LANG.
Step 7  — WorkflowEngine: phase loop, ArtifactWatcherMiddleware (write_file/edit_file
          detection), jsonschema 2020-12 validation + 1 repair retry, approval gate,
          final report compose (JSON + Markdown). FK-safe persistence ordering.
          RunEventType + run_idempotency_key per plan v2.0 §13.1.
Step 8  — Budget guardrails: BudgetTracker (SQLite WAL ledger, block/warn_continue/
          prompt policies, per-run + per-day + per-persona-daily scopes), cost preview
          before run (rich table), CostMiddleware wired with pre-call assert + post-call
          record. CLI: budget / stats --by model|persona|day / costs.
Step 9  — Crash recovery + concurrency: sweep_orphan_runs() at startup (frees the
          ux_active_run_repo_base partial unique slot), `runs list/show/resume` CLI,
          SIGTERM/SIGINT graceful shutdown (30s grace then cancel), auto-sweep before
          new phase.
Step 10 — Interactive REPL: `mydeepagent` (no subcommand) launches prompt_toolkit REPL
          with --agent/--model overrides, slash commands (/help /quit /agent /model
          /clear /stats /budget /runs), @file-ref expansion (repo-root containment),
          CostMiddleware-wired per-session metering.
Step 11 — Audit log + secret scrubbing: append-only {state_dir}/audit.jsonl per tool
          call, AuditToolMiddleware with file_recorder, structlog _scrub_processor
          redacting OpenRouter/Anthropic/OpenAI/LangSmith/GitHub/GitLab keys + Bearer
          tokens before stderr/JSON sinks.
Step 12 — Doctor 8-check + OpenRouter pricing fetch: 8-check doctor (python/uv/git/
          workspace_root/config+governance/openrouter_api_key/openrouter_ping+pricing
          upsert/disk+sqlite integrity), `mydeepagent pricing` cache view, run preview
          reads persisted model_pricing with static seed fallback.
Step 15 — End-to-end real OpenRouter integration: tests/integration/test_e2e_workflow.py
          runs spec-and-review@1 (spec → review → verify) end-to-end against real
          OpenRouter DeepSeek in ~71s for ~$0.05 per run. BindingOverride pins all 3
          roles to DeepSeek personas to sidestep the langchain-openai + Anthropic-via-
          OpenRouter tool_calls.args JSON-string ValidationError (known v0.1.0 limit).
          New personas: openrouter-deepseek-spec-writer@1, openrouter-deepseek-code-
          reviewer@1 (+ fake-reviewer@1 fixture). _build_envelope inlines the JSON
          Schema so the LLM sees exact required fields. _record_llm_call fills every
          NOT NULL LlmCallRow column. CostMiddleware probes both usage_metadata and
          response_metadata.token_usage (prompt_tokens/completion_tokens fallback).
          dev/review-finding-batch@1 artifact schema added.

Known v0.1.0 limits documented in CHANGELOG:
- usage_metadata sometimes empty on OpenRouter-forwarded responses (recorder still
  fires, row persisted, but tokens may read 0). v0.2 will probe more response shapes.
- Anthropic via OpenRouter currently fails with tool_calls.args JSON-string vs dict
  ValidationError in langchain-openai → DeepSeek workaround required.
- `runs resume <run_id>` is a stub (exit-2 hint only).

Gates: ruff check / ruff format --check / mypy --strict / 574 pytest PASS (5.29s)
plus 1 E2E PASS (71.21s, real OpenRouter, ~\$0.05).

--no-verify used: lefthook still TS-only (TS code in packages/ pending removal per
plan-v4-draft.md Step 0).

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-05-16 16:32:46 +09:00

130 lines
3.6 KiB
Python

"""Unit tests for slash.py — parse_slash + SlashRegistry."""
from __future__ import annotations
import pytest
from my_deepagent.slash import SlashParsed, SlashRegistry, parse_slash
# ---------------------------------------------------------------------------
# parse_slash
# ---------------------------------------------------------------------------
def test_parse_quit() -> None:
result = parse_slash("/quit")
assert result is not None
assert result.name == "quit"
assert result.args == ()
assert result.raw == "quit"
def test_parse_agent_with_arg() -> None:
result = parse_slash("/agent code-reviewer")
assert result is not None
assert result.name == "agent"
assert result.args == ("code-reviewer",)
def test_parse_model_with_slash_in_arg() -> None:
result = parse_slash("/model anthropic/claude")
assert result is not None
assert result.name == "model"
assert result.args == ("anthropic/claude",)
def test_parse_plain_text_returns_none() -> None:
assert parse_slash("hello world") is None
def test_parse_empty_string_returns_none() -> None:
assert parse_slash("") is None
def test_parse_bare_slash_gives_empty_name() -> None:
result = parse_slash("/")
assert result is not None
assert result.name == ""
assert result.args == ()
assert result.raw == ""
def test_parse_uppercase_normalized_to_lower() -> None:
result = parse_slash("/QUIT")
assert result is not None
assert result.name == "quit"
def test_parse_spaced_slash_command() -> None:
result = parse_slash("/ spaced ")
# body after strip of "/ spaced " → body = "spaced" (strip on body)
assert result is not None
assert result.name == "spaced"
assert result.args == ()
# ---------------------------------------------------------------------------
# SlashRegistry
# ---------------------------------------------------------------------------
@pytest.mark.asyncio
async def test_registry_register_and_dispatch_returns_handler_value() -> None:
reg = SlashRegistry()
calls: list[str] = []
async def handler(cmd: SlashParsed) -> bool:
calls.append(cmd.name)
return False
reg.register("foo", handler, help="test help")
result = await reg.dispatch(SlashParsed(name="foo", args=(), raw="foo"))
assert result is False
assert calls == ["foo"]
@pytest.mark.asyncio
async def test_registry_unknown_name_returns_false() -> None:
reg = SlashRegistry()
result = await reg.dispatch(SlashParsed(name="nonexistent", args=(), raw="nonexistent"))
assert result is False
@pytest.mark.asyncio
async def test_registry_handler_returning_true_propagates() -> None:
reg = SlashRegistry()
async def quit_handler(cmd: SlashParsed) -> bool:
return True
reg.register("quit", quit_handler, help="exit")
result = await reg.dispatch(SlashParsed(name="quit", args=(), raw="quit"))
assert result is True
def test_registry_names_sorted() -> None:
reg = SlashRegistry()
async def noop(cmd: SlashParsed) -> bool:
return False
reg.register("zebra", noop)
reg.register("apple", noop)
reg.register("mango", noop)
assert reg.names == ["apple", "mango", "zebra"]
def test_registry_help_for_and_all_help() -> None:
reg = SlashRegistry()
async def noop(cmd: SlashParsed) -> bool:
return False
reg.register("quit", noop, help="exit the REPL")
reg.register("help", noop, help="show commands")
assert reg.help_for("quit") == "exit the REPL"
assert reg.help_for("unknown") == ""
pairs = dict(reg.all_help())
assert pairs["quit"] == "exit the REPL"
assert pairs["help"] == "show commands"