130 lines
4.1 KiB
Python
130 lines
4.1 KiB
Python
"""Tests for the /plan gateway slash command."""
|
|
|
|
from datetime import datetime
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from agent.skill_commands import scan_skill_commands
|
|
from gateway.config import GatewayConfig, Platform, PlatformConfig
|
|
from gateway.platforms.base import MessageEvent
|
|
from gateway.session import SessionEntry, SessionSource
|
|
|
|
|
|
def _make_runner():
|
|
from gateway.run import GatewayRunner
|
|
|
|
runner = object.__new__(GatewayRunner)
|
|
runner.config = GatewayConfig(
|
|
platforms={Platform.TELEGRAM: PlatformConfig(enabled=True, token="***")}
|
|
)
|
|
runner.adapters = {}
|
|
runner._voice_mode = {}
|
|
runner.hooks = SimpleNamespace(emit=AsyncMock(), loaded_hooks=False)
|
|
runner.session_store = MagicMock()
|
|
runner.session_store.get_or_create_session.return_value = SessionEntry(
|
|
session_key="agent:main:telegram:dm:c1:u1",
|
|
session_id="sess-1",
|
|
created_at=datetime.now(),
|
|
updated_at=datetime.now(),
|
|
platform=Platform.TELEGRAM,
|
|
chat_type="dm",
|
|
)
|
|
runner.session_store.load_transcript.return_value = []
|
|
runner.session_store.has_any_sessions.return_value = True
|
|
runner.session_store.append_to_transcript = MagicMock()
|
|
runner.session_store.rewrite_transcript = MagicMock()
|
|
runner._running_agents = {}
|
|
runner._pending_messages = {}
|
|
runner._pending_approvals = {}
|
|
runner._session_db = None
|
|
runner._reasoning_config = None
|
|
runner._provider_routing = {}
|
|
runner._fallback_model = None
|
|
runner._show_reasoning = False
|
|
runner._is_user_authorized = lambda _source: True
|
|
runner._set_session_env = lambda _context: None
|
|
runner._run_agent = AsyncMock(
|
|
return_value={
|
|
"final_response": "planned",
|
|
"messages": [],
|
|
"tools": [],
|
|
"history_offset": 0,
|
|
"last_prompt_tokens": 0,
|
|
}
|
|
)
|
|
return runner
|
|
|
|
|
|
def _make_event(text="/plan"):
|
|
return MessageEvent(
|
|
text=text,
|
|
source=SessionSource(
|
|
platform=Platform.TELEGRAM,
|
|
user_id="u1",
|
|
chat_id="c1",
|
|
user_name="tester",
|
|
chat_type="dm",
|
|
),
|
|
message_id="m1",
|
|
)
|
|
|
|
|
|
def _make_plan_skill(skills_dir):
|
|
skill_dir = skills_dir / "plan"
|
|
skill_dir.mkdir(parents=True, exist_ok=True)
|
|
(skill_dir / "SKILL.md").write_text(
|
|
"""---
|
|
name: plan
|
|
description: Plan mode skill.
|
|
---
|
|
|
|
# Plan
|
|
|
|
Use the current conversation context when no explicit instruction is provided.
|
|
Save plans under the active workspace's .hermes/plans directory.
|
|
"""
|
|
)
|
|
|
|
|
|
class TestGatewayPlanCommand:
|
|
@pytest.mark.asyncio
|
|
async def test_plan_command_loads_skill_and_runs_agent(self, monkeypatch, tmp_path):
|
|
import gateway.run as gateway_run
|
|
|
|
runner = _make_runner()
|
|
event = _make_event("/plan Add OAuth login")
|
|
|
|
monkeypatch.setattr(gateway_run, "_resolve_runtime_agent_kwargs", lambda: {"api_key": "***"})
|
|
monkeypatch.setattr(
|
|
"agent.model_metadata.get_model_context_length",
|
|
lambda *_args, **_kwargs: 100_000,
|
|
)
|
|
|
|
with patch("tools.skills_tool.SKILLS_DIR", tmp_path):
|
|
_make_plan_skill(tmp_path)
|
|
scan_skill_commands()
|
|
result = await runner._handle_message(event)
|
|
|
|
assert result == "planned"
|
|
forwarded = runner._run_agent.call_args.kwargs["message"]
|
|
assert "Plan mode skill" in forwarded
|
|
assert "Add OAuth login" in forwarded
|
|
assert ".hermes/plans" in forwarded
|
|
assert str(tmp_path / "plans") not in forwarded
|
|
assert "active workspace/backend cwd" in forwarded
|
|
assert "Runtime note:" in forwarded
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_plan_command_appears_in_help_output_via_skill_listing(self, tmp_path):
|
|
runner = _make_runner()
|
|
event = _make_event("/help")
|
|
|
|
with patch("tools.skills_tool.SKILLS_DIR", tmp_path):
|
|
_make_plan_skill(tmp_path)
|
|
scan_skill_commands()
|
|
result = await runner._handle_help_command(event)
|
|
|
|
assert "/plan" in result
|