mirror of
https://github.com/Tencent/WeKnora.git
synced 2026-09-01 14:53:07 +08:00
wizardchen
a76c6ef24e
Rename the distribution to tencent-weknora-mcp, point metadata at Tencent/WeKnora, bump to 1.1.1 for the transport/session fixes, and tighten publish to main/tags only via Trusted Publishing.
93 lines
2.8 KiB
Python
93 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
WeKnora MCP Server 模组测试脚本
|
||
|
||
测试模组的各种启动方式和功能。unittest discover 会收集本文件中的 TestCase；
|
||
也可直接运行: python test_module.py
|
||
"""
|
||
|
||
import os
|
||
import subprocess
|
||
import sys
|
||
import unittest
|
||
from pathlib import Path
|
||
|
||
MCP_SERVER_DIR = Path(__file__).resolve().parent
|
||
|
||
REQUIRED_FILES = [
|
||
"__init__.py",
|
||
"main.py",
|
||
"run_server.py",
|
||
"weknora_mcp_server.py",
|
||
"requirements.txt",
|
||
"setup.py",
|
||
"pyproject.toml",
|
||
"README.md",
|
||
"INSTALL.md",
|
||
"LICENSE",
|
||
"MANIFEST.in",
|
||
]
|
||
|
||
|
||
class ModuleIntegrationTest(unittest.TestCase):
|
||
def test_imports(self):
|
||
import mcp # noqa: F401
|
||
import requests # noqa: F401
|
||
import weknora_mcp_server # noqa: F401
|
||
from weknora_mcp_server import WeKnoraClient, run # noqa: F401
|
||
import main # noqa: F401
|
||
|
||
def test_environment_optional_vars(self):
|
||
os.getenv("WEKNORA_BASE_URL")
|
||
os.getenv("WEKNORA_API_KEY")
|
||
|
||
def test_client_creation(self):
|
||
from weknora_mcp_server import WeKnoraClient
|
||
|
||
base_url = os.getenv("WEKNORA_BASE_URL", "http://localhost:8080/api/v1")
|
||
api_key = os.getenv("WEKNORA_API_KEY", "test_key")
|
||
client = WeKnoraClient(base_url, api_key)
|
||
self.assertEqual(client.base_url, base_url)
|
||
self.assertEqual(client.api_key, api_key)
|
||
|
||
def test_required_files_exist(self):
|
||
missing = [name for name in REQUIRED_FILES if not (MCP_SERVER_DIR / name).exists()]
|
||
self.assertEqual(missing, [], f"Missing files: {missing}")
|
||
|
||
def test_main_help(self):
|
||
result = subprocess.run(
|
||
[sys.executable, "main.py", "--help"],
|
||
capture_output=True,
|
||
text=True,
|
||
timeout=10,
|
||
cwd=MCP_SERVER_DIR,
|
||
)
|
||
self.assertEqual(result.returncode, 0, result.stderr)
|
||
|
||
def test_main_check_only(self):
|
||
result = subprocess.run(
|
||
[sys.executable, "main.py", "--check-only"],
|
||
capture_output=True,
|
||
text=True,
|
||
timeout=10,
|
||
cwd=MCP_SERVER_DIR,
|
||
)
|
||
self.assertEqual(result.returncode, 0, result.stderr)
|
||
|
||
def test_wiki_tools(self):
|
||
import weknora_mcp_server
|
||
|
||
client = weknora_mcp_server.WeKnoraClient("http://localhost:8080/api/v1", "test")
|
||
for method in ["wiki_search", "wiki_read_page", "wiki_index_view"]:
|
||
self.assertTrue(hasattr(client, method), f"WeKnoraClient missing: {method}")
|
||
self.assertTrue(callable(getattr(client, method)), f"{method} not callable")
|
||
|
||
def test_pyproject_metadata(self):
|
||
text = (MCP_SERVER_DIR / "pyproject.toml").read_text(encoding="utf-8")
|
||
self.assertIn("tencent-weknora-mcp", text)
|
||
self.assertIn("weknora-mcp-server", text)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main(verbosity=2)
|