红尘
d73ff9b0fb
Deploy Site / deploy-vercel (push) Has been cancelled
Deploy Site / deploy-docs (push) Has been cancelled
Docker / shell lint / Lint Dockerfile (hadolint) (push) Has been cancelled
Docker / shell lint / Lint docker/ shell scripts (shellcheck) (push) Has been cancelled
Docker Build and Publish / build-amd64 (push) Has been cancelled
Docker Build and Publish / build-arm64 (push) Has been cancelled
Lint (ruff + ty) / ruff + ty diff (push) Has been cancelled
Lint (ruff + ty) / ruff enforcement (blocking) (push) Has been cancelled
Lint (ruff + ty) / Windows footguns (blocking) (push) Has been cancelled
Nix Lockfile Fix / auto-fix-main (push) Has been cancelled
Nix Lockfile Fix / fix (push) Has been cancelled
Nix / nix (macos-latest) (push) Has been cancelled
Nix / nix (ubuntu-latest) (push) Has been cancelled
OSV-Scanner / Scan lockfiles (push) Has been cancelled
Build Skills Index / build-index (push) Has been cancelled
Tests / test (1) (push) Has been cancelled
Tests / test (2) (push) Has been cancelled
Tests / test (3) (push) Has been cancelled
Tests / test (4) (push) Has been cancelled
Tests / test (5) (push) Has been cancelled
Tests / test (6) (push) Has been cancelled
Tests / e2e (push) Has been cancelled
uv.lock check / uv lock --check (push) Has been cancelled
Docker Build and Publish / merge (push) Has been cancelled
Build Skills Index / trigger-deploy (push) Has been cancelled
Tests / save-durations (push) Has been cancelled
63 lines
2.4 KiB
Python
63 lines
2.4 KiB
Python
"""Tests for feishu_doc_tool and feishu_drive_tool — registration and schema validation."""
|
|
|
|
import importlib
|
|
import unittest
|
|
|
|
from tools.registry import registry
|
|
|
|
# Trigger tool discovery so feishu tools get registered
|
|
importlib.import_module("tools.feishu_doc_tool")
|
|
importlib.import_module("tools.feishu_drive_tool")
|
|
|
|
|
|
class TestFeishuToolRegistration(unittest.TestCase):
|
|
"""Verify feishu tools are registered and have valid schemas."""
|
|
|
|
EXPECTED_TOOLS = {
|
|
"feishu_doc_read": "feishu_doc",
|
|
"feishu_drive_list_comments": "feishu_drive",
|
|
"feishu_drive_list_comment_replies": "feishu_drive",
|
|
"feishu_drive_reply_comment": "feishu_drive",
|
|
"feishu_drive_add_comment": "feishu_drive",
|
|
}
|
|
|
|
def test_all_tools_registered(self):
|
|
for tool_name, toolset in self.EXPECTED_TOOLS.items():
|
|
entry = registry.get_entry(tool_name)
|
|
self.assertIsNotNone(entry, f"{tool_name} not registered")
|
|
self.assertEqual(entry.toolset, toolset)
|
|
|
|
def test_schemas_have_required_fields(self):
|
|
for tool_name in self.EXPECTED_TOOLS:
|
|
entry = registry.get_entry(tool_name)
|
|
schema = entry.schema
|
|
self.assertIn("name", schema)
|
|
self.assertEqual(schema["name"], tool_name)
|
|
self.assertIn("description", schema)
|
|
self.assertIn("parameters", schema)
|
|
self.assertIn("type", schema["parameters"])
|
|
self.assertEqual(schema["parameters"]["type"], "object")
|
|
|
|
def test_handlers_are_callable(self):
|
|
for tool_name in self.EXPECTED_TOOLS:
|
|
entry = registry.get_entry(tool_name)
|
|
self.assertTrue(callable(entry.handler))
|
|
|
|
def test_doc_read_schema_params(self):
|
|
entry = registry.get_entry("feishu_doc_read")
|
|
props = entry.schema["parameters"].get("properties", {})
|
|
self.assertIn("doc_token", props)
|
|
|
|
def test_drive_tools_require_file_token(self):
|
|
for tool_name in self.EXPECTED_TOOLS:
|
|
if tool_name == "feishu_doc_read":
|
|
continue
|
|
entry = registry.get_entry(tool_name)
|
|
props = entry.schema["parameters"].get("properties", {})
|
|
self.assertIn("file_token", props, f"{tool_name} missing file_token param")
|
|
self.assertIn("file_type", props, f"{tool_name} missing file_type param")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|