403 lines
15 KiB
Python
403 lines
15 KiB
Python
"""Tests for myagents CLI."""
|
|
|
|
import json
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from click.testing import CliRunner
|
|
|
|
import myagents.launcher
|
|
from myagents.cli import cli
|
|
|
|
|
|
class TestMyagentsHelp:
|
|
"""Tests for top-level myagents command."""
|
|
|
|
def test_help_shows_agent_subcommands(self) -> None:
|
|
"""--help should list claude, kimi and codex subcommands."""
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["--help"])
|
|
assert result.exit_code == 0
|
|
assert "claude" in result.output
|
|
assert "kimi" in result.output
|
|
assert "codex" in result.output
|
|
assert "update" in result.output
|
|
assert "upgrade" in result.output
|
|
|
|
def test_version_shows_version(self) -> None:
|
|
"""--version should show package version."""
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["--version"])
|
|
assert result.exit_code == 0
|
|
assert "version" in result.output.lower()
|
|
|
|
def test_bare_invocation_shows_help(self) -> None:
|
|
"""Running myagents without subcommands should show help, not launch agent."""
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, [])
|
|
assert result.exit_code == 0
|
|
assert "Usage:" in result.output
|
|
|
|
|
|
class TestClaudeSubcommand:
|
|
"""Tests for ``myagents claude``."""
|
|
|
|
def test_help_shows_options(self) -> None:
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["claude", "--help"])
|
|
assert result.exit_code == 0
|
|
assert "--cwd" in result.output
|
|
assert "--list" in result.output
|
|
assert "--dangerously-skip-permissions" not in result.output
|
|
|
|
def test_runs_claude(self) -> None:
|
|
runner = CliRunner()
|
|
with (
|
|
patch("myagents.launcher.shutil.which", return_value="/usr/bin/claude"),
|
|
patch("myagents.launcher.subprocess.run") as mock_run,
|
|
):
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
result = runner.invoke(cli, ["claude"])
|
|
assert result.exit_code == 0
|
|
mock_run.assert_called_once()
|
|
assert mock_run.call_args[0][0] == [
|
|
"/usr/bin/claude",
|
|
"--dangerously-skip-permissions",
|
|
]
|
|
|
|
def test_missing_binary_error(self) -> None:
|
|
runner = CliRunner()
|
|
with patch("myagents.launcher.shutil.which", return_value=None):
|
|
result = runner.invoke(cli, ["claude"])
|
|
assert result.exit_code == 127
|
|
assert "not found" in result.output.lower()
|
|
|
|
def test_cwd_option_passed(self, tmp_path: Path) -> None:
|
|
runner = CliRunner()
|
|
test_dir = tmp_path / "test_cwd"
|
|
test_dir.mkdir()
|
|
|
|
with (
|
|
patch("myagents.launcher.shutil.which", return_value="/usr/bin/claude"),
|
|
patch("myagents.launcher.subprocess.run") as mock_run,
|
|
):
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
result = runner.invoke(cli, ["claude", "--cwd", str(test_dir)])
|
|
assert result.exit_code == 0
|
|
assert mock_run.call_args.kwargs.get("cwd") == str(test_dir.resolve())
|
|
|
|
def test_cwd_invalid_directory(self, tmp_path: Path) -> None:
|
|
runner = CliRunner()
|
|
bad_dir = tmp_path / "does_not_exist"
|
|
result = runner.invoke(cli, ["claude", "--cwd", str(bad_dir)])
|
|
assert result.exit_code == 1
|
|
assert "not a directory" in result.output.lower()
|
|
|
|
|
|
class TestClaudePassthrough:
|
|
"""Unknown leading flags forward to claude instead of erroring."""
|
|
|
|
def _invoke(self, args: list[str], tmp_path: Path) -> list[str]:
|
|
runner = CliRunner()
|
|
with (
|
|
patch("myagents.launcher.shutil.which", return_value="/usr/bin/claude"),
|
|
patch("myagents.launcher.subprocess.run") as mock_run,
|
|
):
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
result = runner.invoke(
|
|
cli, ["claude", "--cwd", str(tmp_path), *args]
|
|
)
|
|
assert result.exit_code == 0, result.output
|
|
mock_run.assert_called_once()
|
|
return mock_run.call_args[0][0]
|
|
|
|
def test_resume_flag_passes_through(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["--resume"], tmp_path)
|
|
assert cmd == [
|
|
"/usr/bin/claude",
|
|
"--dangerously-skip-permissions",
|
|
"--resume",
|
|
]
|
|
|
|
def test_resume_with_session_id(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["-r", "abc123"], tmp_path)
|
|
assert cmd[-2:] == ["-r", "abc123"]
|
|
|
|
def test_continue_flag_passes_through(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["--continue"], tmp_path)
|
|
assert cmd[-1] == "--continue"
|
|
|
|
|
|
class TestClaudeListSessions:
|
|
"""``myagents claude --list`` reports resumable sessions."""
|
|
|
|
def test_list_empty_directory(self, tmp_path: Path) -> None:
|
|
runner = CliRunner()
|
|
with patch("myagents.launcher.subprocess.run") as mock_run:
|
|
result = runner.invoke(cli, ["claude", "--cwd", str(tmp_path), "--list"])
|
|
assert result.exit_code == 0
|
|
assert "no sessions" in result.output.lower()
|
|
mock_run.assert_not_called()
|
|
|
|
|
|
class TestKimiSubcommand:
|
|
"""Tests for ``myagents kimi``."""
|
|
|
|
def test_help_shows_options(self) -> None:
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["kimi", "--help"])
|
|
assert result.exit_code == 0
|
|
assert "--cwd" in result.output
|
|
assert "--list" in result.output
|
|
assert "--dangerously-skip-permissions" not in result.output
|
|
|
|
def test_runs_kimi(self) -> None:
|
|
runner = CliRunner()
|
|
with (
|
|
patch("myagents.launcher.shutil.which", return_value="/usr/bin/kimi"),
|
|
patch("myagents.launcher.subprocess.run") as mock_run,
|
|
):
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
result = runner.invoke(cli, ["kimi"])
|
|
assert result.exit_code == 0
|
|
mock_run.assert_called_once()
|
|
assert mock_run.call_args[0][0] == ["/usr/bin/kimi"]
|
|
|
|
def test_missing_binary_error(self) -> None:
|
|
runner = CliRunner()
|
|
with patch("myagents.launcher.shutil.which", return_value=None):
|
|
result = runner.invoke(cli, ["kimi"])
|
|
assert result.exit_code == 127
|
|
assert "not found" in result.output.lower()
|
|
|
|
def test_cwd_option_passed(self, tmp_path: Path) -> None:
|
|
runner = CliRunner()
|
|
test_dir = tmp_path / "test_cwd"
|
|
test_dir.mkdir()
|
|
|
|
with (
|
|
patch("myagents.launcher.shutil.which", return_value="/usr/bin/kimi"),
|
|
patch("myagents.launcher.subprocess.run") as mock_run,
|
|
):
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
result = runner.invoke(cli, ["kimi", "--cwd", str(test_dir)])
|
|
assert result.exit_code == 0
|
|
assert mock_run.call_args.kwargs.get("cwd") == str(test_dir.resolve())
|
|
|
|
|
|
class TestKimiPassthrough:
|
|
"""Unknown leading flags forward to kimi instead of erroring."""
|
|
|
|
def _invoke(self, args: list[str], tmp_path: Path) -> list[str]:
|
|
runner = CliRunner()
|
|
with (
|
|
patch("myagents.launcher.shutil.which", return_value="/usr/bin/kimi"),
|
|
patch("myagents.launcher.subprocess.run") as mock_run,
|
|
):
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
result = runner.invoke(cli, ["kimi", "--cwd", str(tmp_path), *args])
|
|
assert result.exit_code == 0, result.output
|
|
mock_run.assert_called_once()
|
|
return mock_run.call_args[0][0]
|
|
|
|
def test_resume_flag_passes_through(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["--resume"], tmp_path)
|
|
assert cmd == ["/usr/bin/kimi", "--resume"]
|
|
|
|
def test_resume_with_session_id(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["-r", "abc123"], tmp_path)
|
|
assert cmd[-2:] == ["-r", "abc123"]
|
|
|
|
def test_continue_flag_passes_through(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["--continue"], tmp_path)
|
|
assert cmd[-1] == "--continue"
|
|
|
|
|
|
class TestUpdateSubcommand:
|
|
"""Tests for update/upgrade subcommands registration."""
|
|
|
|
def test_update_subcommand_exists(self) -> None:
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["update", "--help"])
|
|
assert result.exit_code == 0
|
|
assert (
|
|
"update" in result.output.lower()
|
|
or "reinstall" in result.output.lower()
|
|
)
|
|
|
|
def test_upgrade_alias_exists(self) -> None:
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["upgrade", "--help"])
|
|
assert result.exit_code == 0
|
|
|
|
|
|
class TestCodexSubcommand:
|
|
"""Tests for ``myagents codex``."""
|
|
|
|
def test_help_shows_options(self) -> None:
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["codex", "--help"])
|
|
assert result.exit_code == 0
|
|
assert "--cwd" in result.output
|
|
assert "--list" in result.output
|
|
|
|
def test_runs_codex(self) -> None:
|
|
runner = CliRunner()
|
|
with (
|
|
patch("myagents.launcher.shutil.which", return_value="/usr/bin/codex"),
|
|
patch("myagents.launcher.subprocess.run") as mock_run,
|
|
):
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
result = runner.invoke(cli, ["codex"])
|
|
assert result.exit_code == 0
|
|
mock_run.assert_called_once()
|
|
assert mock_run.call_args[0][0] == ["/usr/bin/codex", "--dangerously-bypass-approvals-and-sandbox"]
|
|
|
|
def test_missing_binary_error(self) -> None:
|
|
runner = CliRunner()
|
|
with patch("myagents.launcher.shutil.which", return_value=None):
|
|
result = runner.invoke(cli, ["codex"])
|
|
assert result.exit_code == 127
|
|
assert "not found" in result.output.lower()
|
|
|
|
def test_cwd_option_passed(self, tmp_path: Path) -> None:
|
|
runner = CliRunner()
|
|
test_dir = tmp_path / "test_cwd"
|
|
test_dir.mkdir()
|
|
|
|
with (
|
|
patch("myagents.launcher.shutil.which", return_value="/usr/bin/codex"),
|
|
patch("myagents.launcher.subprocess.run") as mock_run,
|
|
):
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
result = runner.invoke(cli, ["codex", "--cwd", str(test_dir)])
|
|
assert result.exit_code == 0
|
|
assert mock_run.call_args.kwargs.get("cwd") == str(test_dir.resolve())
|
|
|
|
|
|
class TestCodexPassthrough:
|
|
"""Codex resume subcommand forwards to codex instead of erroring."""
|
|
|
|
def _invoke(self, args: list[str], tmp_path: Path) -> list[str]:
|
|
runner = CliRunner()
|
|
with (
|
|
patch("myagents.launcher.shutil.which", return_value="/usr/bin/codex"),
|
|
patch("myagents.launcher.subprocess.run") as mock_run,
|
|
):
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
result = runner.invoke(
|
|
cli, ["codex", "--cwd", str(tmp_path), *args]
|
|
)
|
|
assert result.exit_code == 0, result.output
|
|
mock_run.assert_called_once()
|
|
return mock_run.call_args[0][0]
|
|
|
|
def test_resume_subcommand_passes_through(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["resume", "abc123"], tmp_path)
|
|
assert cmd == ["/usr/bin/codex", "--dangerously-bypass-approvals-and-sandbox", "resume", "abc123"]
|
|
|
|
def test_continue_subcommand_passes_through(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["continue"], tmp_path)
|
|
assert cmd == ["/usr/bin/codex", "--dangerously-bypass-approvals-and-sandbox", "continue"]
|
|
|
|
|
|
class TestCodexResumeOptions:
|
|
"""Map myagents-style resume flags to codex resume subcommand."""
|
|
|
|
def _invoke(self, args: list[str], tmp_path: Path) -> list[str]:
|
|
runner = CliRunner()
|
|
with (
|
|
patch("myagents.launcher.shutil.which", return_value="/usr/bin/codex"),
|
|
patch("myagents.launcher.subprocess.run") as mock_run,
|
|
):
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
result = runner.invoke(
|
|
cli, ["codex", "--cwd", str(tmp_path), *args]
|
|
)
|
|
assert result.exit_code == 0, result.output
|
|
mock_run.assert_called_once()
|
|
return mock_run.call_args[0][0]
|
|
|
|
def test_r_flag_maps_to_resume(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["-r", "abc123"], tmp_path)
|
|
assert cmd == ["/usr/bin/codex", "--dangerously-bypass-approvals-and-sandbox", "resume", "abc123"]
|
|
|
|
def test_resume_flag_maps_to_resume(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["--resume", "abc123"], tmp_path)
|
|
assert cmd == ["/usr/bin/codex", "--dangerously-bypass-approvals-and-sandbox", "resume", "abc123"]
|
|
|
|
def test_continue_flag_maps_to_continue(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["--continue"], tmp_path)
|
|
assert cmd == ["/usr/bin/codex", "--dangerously-bypass-approvals-and-sandbox", "continue"]
|
|
|
|
def test_last_flag_maps_to_resume_last(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["--last"], tmp_path)
|
|
assert cmd == ["/usr/bin/codex", "--dangerously-bypass-approvals-and-sandbox", "resume", "--last"]
|
|
|
|
def test_all_flag_maps_to_resume_all(self, tmp_path: Path) -> None:
|
|
cmd = self._invoke(["--all"], tmp_path)
|
|
assert cmd == ["/usr/bin/codex", "--dangerously-bypass-approvals-and-sandbox", "resume", "--all"]
|
|
|
|
|
|
class TestCodexListSessions:
|
|
"""``myagents codex --list`` reports resumable sessions."""
|
|
|
|
def test_list_empty_directory(self, tmp_path: Path) -> None:
|
|
runner = CliRunner()
|
|
with patch("myagents.launcher.subprocess.run") as mock_run:
|
|
result = runner.invoke(cli, ["codex", "--cwd", str(tmp_path), "--list"])
|
|
assert result.exit_code == 0
|
|
assert "no sessions" in result.output.lower()
|
|
mock_run.assert_not_called()
|
|
|
|
def test_list_shows_session_id_and_snippet(self, tmp_path: Path) -> None:
|
|
runner = CliRunner()
|
|
sessions_root = tmp_path / "sessions"
|
|
sessions_root.mkdir(parents=True)
|
|
session_file = (
|
|
sessions_root
|
|
/ "2026"
|
|
/ "01"
|
|
/ "02"
|
|
/ "rollout-2026-01-02T12-00-00-019aeb67-0620-71c1-8cbb-756bc8845c6e.jsonl"
|
|
)
|
|
session_file.parent.mkdir(parents=True)
|
|
session_file.write_text(
|
|
json.dumps(
|
|
{
|
|
"type": "session_meta",
|
|
"timestamp": "2026-01-02T12:00:00Z",
|
|
"payload": {
|
|
"id": "019aeb67-0620-71c1-8cbb-756bc8845c6e",
|
|
"cwd": str(tmp_path),
|
|
},
|
|
}
|
|
)
|
|
+ "\n"
|
|
+ json.dumps(
|
|
{
|
|
"type": "user_message",
|
|
"payload": {
|
|
"type": "user_message",
|
|
"message": "hello codex",
|
|
},
|
|
}
|
|
)
|
|
+ "\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
with patch(
|
|
"myagents.launcher._BACKENDS",
|
|
{
|
|
"codex": {
|
|
**myagents.launcher._BACKENDS["codex"],
|
|
"sessions_root": lambda: sessions_root,
|
|
}
|
|
},
|
|
):
|
|
result = runner.invoke(cli, ["codex", "--cwd", str(tmp_path), "--list"])
|
|
assert result.exit_code == 0, result.output
|
|
assert "019aeb67-0620-71c1-8cbb-756bc8845c6e" in result.output
|
|
assert "hello codex" in result.output
|