curo1305
51029d4a2d
- test_config.py: GeneralConfig defaults, plugin_settings round-trip - test_config_field.py: ConfigField dataclass, BasePlugin.config_fields() no-op, plugin subclass override - test_config_tui.py: _get/_set_nested, _fid/_pfid helpers, GENERAL_FIELDS validity, ConfigApp general tab rendering, save handler, plugins table, plugin tab visibility, q key exit — using Textual run_test() + Pilot - test_cli.py: auto-setup wizard on first run, skip wizard when config exists, /config in _STATIC_COMMANDS Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
140 lines
4.6 KiB
Python
140 lines
4.6 KiB
Python
import pytest
|
|
from click.testing import CliRunner
|
|
|
|
from pyra.cli import main
|
|
|
|
|
|
def test_memory_write_creates_file(tmp_pyra_home):
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["memory", "write", "user/note.md", "Hello world"])
|
|
assert result.exit_code == 0
|
|
assert (tmp_pyra_home / "memory" / "user" / "note.md").exists()
|
|
assert "Hello world" in (tmp_pyra_home / "memory" / "user" / "note.md").read_text()
|
|
|
|
|
|
def test_memory_write_updates_db(tmp_pyra_home):
|
|
runner = CliRunner()
|
|
runner.invoke(main, ["memory", "write", "user/note.md", "DB test content"])
|
|
from pyra.memory import database
|
|
rows = database.list_all()
|
|
assert any(r["path"] == "user/note.md" for r in rows)
|
|
|
|
|
|
def test_memory_append_adds_content(tmp_pyra_home):
|
|
runner = CliRunner()
|
|
runner.invoke(main, ["memory", "write", "user/note.md", "First line"])
|
|
runner.invoke(main, ["memory", "append", "user/note.md", "Second line"])
|
|
content = (tmp_pyra_home / "memory" / "user" / "note.md").read_text()
|
|
assert "First line" in content
|
|
assert "Second line" in content
|
|
|
|
|
|
def test_memory_read_existing(tmp_pyra_home):
|
|
from pyra.memory.writer import write_memory
|
|
write_memory("user/note.md", "Readable content")
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["memory", "read", "user/note.md"])
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def test_memory_read_missing_exits_cleanly(tmp_pyra_home):
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["memory", "read", "user/does_not_exist.md"])
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def test_memory_read_blocked_path_exits_cleanly(tmp_pyra_home):
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["memory", "read", "../../../../vault/secrets/api_keys.json"])
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def test_memory_list_empty(tmp_pyra_home):
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["memory", "list"])
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def test_memory_list_populated(tmp_pyra_home):
|
|
from pyra.memory.writer import write_memory
|
|
write_memory("user/profile.md", "# Profile")
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["memory", "list"])
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def test_plugin_list_no_config(tmp_pyra_home):
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["plugin", "list"])
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def _make_config():
|
|
from pyra.config.schema import PyraConfig, ProviderConfig
|
|
return PyraConfig(ai=ProviderConfig(provider_id="lmstudio", model="gemma"))
|
|
|
|
|
|
def test_plugin_enable_not_installed(tmp_pyra_home):
|
|
from pyra.config.manager import save_config
|
|
save_config(_make_config())
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["plugin", "enable", "nonexistent"])
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def test_plugin_enable_and_disable(tmp_pyra_home):
|
|
from pyra.config.manager import load_config, save_config
|
|
|
|
save_config(_make_config())
|
|
|
|
plugin_dir = tmp_pyra_home / "plugins" / "myplugin"
|
|
plugin_dir.mkdir(parents=True, exist_ok=True)
|
|
(plugin_dir / "manifest.json").write_text('{"name": "myplugin", "version": "1.0"}')
|
|
|
|
runner = CliRunner()
|
|
|
|
runner.invoke(main, ["plugin", "enable", "myplugin"])
|
|
cfg = load_config()
|
|
assert "myplugin" in cfg.plugins.enabled
|
|
|
|
runner.invoke(main, ["plugin", "disable", "myplugin"])
|
|
cfg = load_config()
|
|
assert "myplugin" not in cfg.plugins.enabled
|
|
|
|
|
|
def test_daemon_commands_exit_cleanly(tmp_pyra_home):
|
|
runner = CliRunner()
|
|
for cmd in ["start", "stop", "status", "restart"]:
|
|
result = runner.invoke(main, ["daemon", cmd])
|
|
assert result.exit_code == 0, f"daemon {cmd} exited with {result.exit_code}"
|
|
|
|
|
|
def test_main_calls_setup_wizard_when_no_config(tmp_pyra_home, monkeypatch):
|
|
setup_calls = []
|
|
monkeypatch.setattr("pyra.setup.wizard.run_setup", lambda: setup_calls.append(1))
|
|
monkeypatch.setattr("pyra.chat.session.start_chat", lambda: None)
|
|
|
|
runner = CliRunner()
|
|
runner.invoke(main, [])
|
|
|
|
assert len(setup_calls) == 1, "run_setup should be called once when no config exists"
|
|
|
|
|
|
def test_main_skips_setup_when_config_exists(tmp_pyra_home, monkeypatch):
|
|
from pyra.config.manager import save_config
|
|
save_config(_make_config())
|
|
|
|
setup_calls = []
|
|
monkeypatch.setattr("pyra.setup.wizard.run_setup", lambda: setup_calls.append(1))
|
|
monkeypatch.setattr("pyra.chat.session.start_chat", lambda: None)
|
|
|
|
runner = CliRunner()
|
|
runner.invoke(main, [])
|
|
|
|
assert len(setup_calls) == 0, "run_setup should NOT be called when config already exists"
|
|
|
|
|
|
def test_config_slash_command_registered():
|
|
from pyra.chat.session import _STATIC_COMMANDS
|
|
assert "/config" in _STATIC_COMMANDS
|