mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-27 21:05:34 +00:00
Prathap PandGitHub
c2756964a7
* fix(channels): wire channel→agent handler and fix Telegram send pipeline * format code * add supported tests
62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
"""Tests for the ``jarvis add`` CLI command."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from pathlib import Path
|
|
from unittest import mock
|
|
|
|
from click.testing import CliRunner
|
|
|
|
from openjarvis.cli.add_cmd import _MCP_TEMPLATES, add
|
|
|
|
|
|
class TestAddCmd:
|
|
def test_add_help(self) -> None:
|
|
result = CliRunner().invoke(add, ["--help"])
|
|
assert result.exit_code == 0
|
|
assert "MCP server" in result.output
|
|
|
|
def test_add_unknown_server(self) -> None:
|
|
result = CliRunner().invoke(add, ["unknown_server"])
|
|
assert result.exit_code != 0
|
|
assert "Unknown MCP server" in result.output
|
|
# Should list known servers
|
|
assert "github" in result.output
|
|
assert "filesystem" in result.output
|
|
|
|
def test_add_known_server(self, tmp_path: Path) -> None:
|
|
mcp_dir = tmp_path / "mcp"
|
|
with mock.patch("openjarvis.cli.add_cmd._MCP_CONFIG_DIR", mcp_dir):
|
|
result = CliRunner().invoke(add, ["filesystem"])
|
|
assert result.exit_code == 0
|
|
assert "Added MCP server: filesystem" in result.output
|
|
|
|
config_file = mcp_dir / "filesystem.json"
|
|
assert config_file.exists()
|
|
data = json.loads(config_file.read_text())
|
|
assert data["command"] == "npx"
|
|
assert "@modelcontextprotocol/server-filesystem" in data["args"]
|
|
|
|
def test_add_with_key(self, tmp_path: Path) -> None:
|
|
mcp_dir = tmp_path / "mcp"
|
|
with mock.patch("openjarvis.cli.add_cmd._MCP_CONFIG_DIR", mcp_dir):
|
|
result = CliRunner().invoke(
|
|
add,
|
|
["github", "--key", "test_token"],
|
|
)
|
|
assert result.exit_code == 0
|
|
|
|
config_file = mcp_dir / "github.json"
|
|
assert config_file.exists()
|
|
data = json.loads(config_file.read_text())
|
|
assert "env" in data
|
|
assert data["env"]["GITHUB_PERSONAL_ACCESS_TOKEN"] == "test_token"
|
|
|
|
def test_mcp_templates_complete(self) -> None:
|
|
required_fields = {"command", "args", "env_key", "description"}
|
|
for name, tmpl in _MCP_TEMPLATES.items():
|
|
assert required_fields.issubset(tmpl.keys()), (
|
|
f"Template '{name}' missing fields: {required_fields - tmpl.keys()}"
|
|
)
|