mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-28 05:12:26 +00:00
04c28ef7ed
* feat: add CLI commands: config, registry, tool * fix: address review feedback on CLI config/registry/tool commands - Replace non-existent `create_config_template` with `generate_default_toml` to fix ImportError when config file is missing - Deduplicate registry maps into shared `_load_registry_map()` helper - Remove dead `_get_registry_class()` function and its tests - Route JSON output to stdout for pipeability (`jarvis config show json | jq`) Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com> --------- Co-authored-by: Jon Saad-Falcon <41205309+jonsaadfalcon@users.noreply.github.com> Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
155 lines
6.7 KiB
Python
155 lines
6.7 KiB
Python
"""Tests for the ``jarvis registry`` CLI commands."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from click.testing import CliRunner
|
|
|
|
from openjarvis.cli import cli
|
|
from openjarvis.core.registry import (
|
|
ToolRegistry,
|
|
)
|
|
|
|
|
|
class TestRegistryCmd:
|
|
"""Test cases for the jarvis registry CLI group."""
|
|
|
|
def test_registry_group_help(self) -> None:
|
|
"""Test that the registry group help displays correctly."""
|
|
result = CliRunner().invoke(cli, ["registry", "--help"])
|
|
assert result.exit_code == 0
|
|
assert "registry" in result.output.lower()
|
|
|
|
def test_registry_list_help(self) -> None:
|
|
"""Test that the registry list help displays correctly."""
|
|
result = CliRunner().invoke(cli, ["registry", "list", "--help"])
|
|
assert result.exit_code == 0
|
|
|
|
def test_registry_show_help(self) -> None:
|
|
"""Test that the registry show help displays correctly."""
|
|
result = CliRunner().invoke(cli, ["registry", "show", "--help"])
|
|
assert result.exit_code == 0
|
|
|
|
def test_registry_list_shows_all_registries(self) -> None:
|
|
"""Test that registry list displays all available registries."""
|
|
result = CliRunner().invoke(cli, ["registry", "list"])
|
|
assert result.exit_code == 0
|
|
# Should show registry-related content
|
|
output = result.output.lower()
|
|
assert "registry" in output
|
|
|
|
def test_registry_show_unknown_registry(self) -> None:
|
|
"""Test that showing an unknown registry shows an error."""
|
|
result = CliRunner().invoke(cli, ["registry", "show", "unknown_registry_xyz"])
|
|
assert result.exit_code == 0 # CLI still exits 0, just shows error message
|
|
assert (
|
|
"unknown" in result.output.lower() or "not found" in result.output.lower()
|
|
)
|
|
|
|
def test_registry_show_tool_registry(self) -> None:
|
|
"""Test that showing the tool registry displays entries."""
|
|
# Trigger tool registration
|
|
import openjarvis.tools # noqa: F401
|
|
|
|
result = CliRunner().invoke(cli, ["registry", "show", "tool"])
|
|
assert result.exit_code == 0
|
|
# Should show tool-related content
|
|
assert "tool" in result.output.lower() or "key" in result.output.lower()
|
|
|
|
def test_registry_show_tool_registry_verbose(self) -> None:
|
|
"""Test that showing the tool registry with verbose flag shows details."""
|
|
# Trigger tool registration
|
|
import openjarvis.tools # noqa: F401
|
|
|
|
result = CliRunner().invoke(cli, ["registry", "show", "tool", "-v"])
|
|
assert result.exit_code == 0
|
|
|
|
def test_registry_show_empty_registry(self) -> None:
|
|
"""Test registry show behavior with an empty registry."""
|
|
result = CliRunner().invoke(cli, ["registry", "show", "agent"])
|
|
assert result.exit_code == 0
|
|
# Should handle empty registries gracefully
|
|
assert "agent" in result.output.lower() or "empty" in result.output.lower()
|
|
|
|
def test_registry_show_accepts_aliases(self) -> None:
|
|
"""Test that registry show accepts various aliases."""
|
|
# Trigger tool registration
|
|
import openjarvis.tools # noqa: F401
|
|
|
|
# Test with 'tools' alias
|
|
result = CliRunner().invoke(cli, ["registry", "show", "tools"])
|
|
assert result.exit_code == 0
|
|
|
|
def test_registry_keys_match_actual_registries(self) -> None:
|
|
"""Test that the list command shows all expected registry names."""
|
|
result = CliRunner().invoke(cli, ["registry", "list"])
|
|
assert result.exit_code == 0
|
|
output = result.output
|
|
|
|
# Check that key registries are mentioned
|
|
assert "ToolRegistry" in output or "tool" in output.lower()
|
|
assert "EngineRegistry" in output or "engine" in output.lower()
|
|
assert "MemoryRegistry" in output or "memory" in output.lower()
|
|
assert "ChannelRegistry" in output or "channel" in output.lower()
|
|
|
|
def test_registry_show_nonexistent_key(self) -> None:
|
|
"""Test that showing a nonexistent key in a registry is handled."""
|
|
result = CliRunner().invoke(cli, ["registry", "show", "nonexistent"])
|
|
assert result.exit_code == 0
|
|
assert (
|
|
"unknown" in result.output.lower() or "not found" in result.output.lower()
|
|
)
|
|
|
|
def test_registry_list_handles_import_error(self) -> None:
|
|
"""Test that registry list handles import errors gracefully."""
|
|
# This tests the exception handling path in list_registries
|
|
result = CliRunner().invoke(cli, ["registry", "list"])
|
|
assert result.exit_code == 0
|
|
# Should still complete even if some registries have issues
|
|
|
|
def test_registry_show_handles_exception(self) -> None:
|
|
"""Test that registry show handles exceptions gracefully."""
|
|
# Patch the registry to raise an exception during keys() call
|
|
from unittest.mock import patch
|
|
|
|
with patch(
|
|
"openjarvis.core.registry.ToolRegistry.keys",
|
|
side_effect=Exception("Test error"),
|
|
):
|
|
result = CliRunner().invoke(cli, ["registry", "show", "tool"])
|
|
assert result.exit_code == 0
|
|
assert "error" in result.output.lower()
|
|
|
|
def test_registry_list_handles_registry_error(self) -> None:
|
|
"""Test that registry list handles errors for specific registries."""
|
|
from unittest.mock import patch
|
|
|
|
# Make one of the registry classes raise an error during keys()
|
|
with patch.object(ToolRegistry, "keys", side_effect=Exception("Import error")):
|
|
result = CliRunner().invoke(cli, ["registry", "list"])
|
|
assert result.exit_code == 0
|
|
# Should still complete and show other registries
|
|
|
|
def test_registry_show_handles_error_during_import(self) -> None:
|
|
"""Test that registry show handles exceptions during import."""
|
|
from unittest.mock import patch
|
|
|
|
# Patch the entire import to raise an exception
|
|
with patch.dict("sys.modules", {"openjarvis.core.registry": None}):
|
|
# This tests the outer exception handler
|
|
result = CliRunner().invoke(cli, ["registry", "show", "tool"])
|
|
assert result.exit_code == 0
|
|
|
|
def test_registry_show_handles_error_during_iteration(self) -> None:
|
|
"""Test that registry show handles exceptions during entry iteration."""
|
|
from unittest.mock import patch
|
|
|
|
# Patch keys to return a fake entry so iteration happens,
|
|
# then patch get to raise
|
|
with patch.object(ToolRegistry, "keys", return_value=["fake_tool"]):
|
|
with patch.object(
|
|
ToolRegistry, "get", side_effect=Exception("Iteration error")
|
|
):
|
|
result = CliRunner().invoke(cli, ["registry", "show", "tool"])
|
|
assert result.exit_code == 0
|
|
assert "error" in result.output.lower()
|