Abdelrahman Abdallah
81 lines
3.8 KiB
Python
81 lines
3.8 KiB
Python
from __future__ import annotations
|
|
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from src.agent_runtime import LocalCodingAgent
|
|
from src.agent_slash_commands import looks_like_command, parse_slash_command
|
|
from src.agent_types import AgentRuntimeConfig, ModelConfig
|
|
|
|
|
|
class AgentSlashCommandTests(unittest.TestCase):
|
|
def test_parse_slash_command(self) -> None:
|
|
parsed = parse_slash_command('/context extra args')
|
|
assert parsed is not None
|
|
self.assertEqual(parsed.command_name, 'context')
|
|
self.assertEqual(parsed.args, 'extra args')
|
|
self.assertFalse(parsed.is_mcp)
|
|
|
|
def test_looks_like_command(self) -> None:
|
|
self.assertTrue(looks_like_command('context'))
|
|
self.assertFalse(looks_like_command('foo/bar'))
|
|
|
|
def test_model_command_updates_agent_model(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
agent = LocalCodingAgent(
|
|
model_config=ModelConfig(model='Qwen/Qwen3-Coder-30B-A3B-Instruct'),
|
|
runtime_config=AgentRuntimeConfig(cwd=Path(tmp_dir)),
|
|
)
|
|
result = agent.run('/model local/test-model')
|
|
self.assertIn('Set model to local/test-model', result.final_output)
|
|
self.assertEqual(agent.model_config.model, 'local/test-model')
|
|
|
|
def test_unknown_command_returns_local_error(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
agent = LocalCodingAgent(
|
|
model_config=ModelConfig(model='Qwen/Qwen3-Coder-30B-A3B-Instruct'),
|
|
runtime_config=AgentRuntimeConfig(cwd=Path(tmp_dir)),
|
|
)
|
|
result = agent.run('/unknown-command')
|
|
self.assertEqual(result.final_output, 'Unknown skill: unknown-command')
|
|
|
|
def test_context_command_renders_usage_report(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
workspace = Path(tmp_dir)
|
|
(workspace / 'CLAUDE.md').write_text('repo instructions\n', encoding='utf-8')
|
|
agent = LocalCodingAgent(
|
|
model_config=ModelConfig(model='Qwen/Qwen3-Coder-30B-A3B-Instruct'),
|
|
runtime_config=AgentRuntimeConfig(cwd=workspace),
|
|
)
|
|
result = agent.run('/context')
|
|
self.assertIn('## Context Usage', result.final_output)
|
|
self.assertIn('### Estimated usage by category', result.final_output)
|
|
self.assertIn('### Memory Files', result.final_output)
|
|
|
|
def test_tools_and_status_commands_render_local_reports(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
agent = LocalCodingAgent(
|
|
model_config=ModelConfig(model='Qwen/Qwen3-Coder-30B-A3B-Instruct'),
|
|
runtime_config=AgentRuntimeConfig(cwd=Path(tmp_dir)),
|
|
)
|
|
tools_result = agent.run('/tools')
|
|
status_result = agent.run('/status')
|
|
self.assertIn('# Tools', tools_result.final_output)
|
|
self.assertIn('`read_file`', tools_result.final_output)
|
|
self.assertIn('# Status', status_result.final_output)
|
|
self.assertIn('Last run: none', status_result.final_output)
|
|
|
|
def test_clear_command_clears_saved_runtime_state(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
agent = LocalCodingAgent(
|
|
model_config=ModelConfig(model='Qwen/Qwen3-Coder-30B-A3B-Instruct'),
|
|
runtime_config=AgentRuntimeConfig(cwd=Path(tmp_dir)),
|
|
)
|
|
agent.last_session = agent.build_session('hello')
|
|
agent.last_run_result = object() # type: ignore[assignment]
|
|
result = agent.run('/clear')
|
|
self.assertIn('Cleared ephemeral Python agent state', result.final_output)
|
|
self.assertIsNone(agent.last_session)
|
|
self.assertIsNone(agent.last_run_result)
|