wehub-resource-sync
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
import pytest
|
|
|
|
from services.registry import get_registered_tools, mcp_for_unity_tool
|
|
import services.registry.tool_registry as tool_registry_module
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def restore_tool_registry_state():
|
|
original_registry = list(tool_registry_module._tool_registry)
|
|
try:
|
|
yield
|
|
finally:
|
|
tool_registry_module._tool_registry[:] = original_registry
|
|
|
|
|
|
def test_tool_registry_defaults_unity_target_to_tool_name():
|
|
@mcp_for_unity_tool()
|
|
def _default_target_tool():
|
|
return None
|
|
|
|
registered_tools = get_registered_tools()
|
|
tool_info = next(item for item in registered_tools if item["name"] == "_default_target_tool")
|
|
assert tool_info["unity_target"] == "_default_target_tool"
|
|
|
|
|
|
def test_tool_registry_supports_server_only_and_alias_targets():
|
|
@mcp_for_unity_tool(unity_target=None)
|
|
def _server_only_tool():
|
|
return None
|
|
|
|
@mcp_for_unity_tool(unity_target="manage_script")
|
|
def _manage_script_alias_tool():
|
|
return None
|
|
|
|
registered_tools = get_registered_tools()
|
|
server_only = next(item for item in registered_tools if item["name"] == "_server_only_tool")
|
|
alias_tool = next(item for item in registered_tools if item["name"] == "_manage_script_alias_tool")
|
|
|
|
assert server_only["unity_target"] is None
|
|
assert alias_tool["unity_target"] == "manage_script"
|
|
|
|
|
|
def test_tool_registry_does_not_leak_unity_target_into_tool_kwargs():
|
|
@mcp_for_unity_tool(unity_target="manage_script", annotations={"title": "x"})
|
|
def _non_leaking_target_tool():
|
|
return None
|
|
|
|
registered_tools = get_registered_tools()
|
|
tool_info = next(item for item in registered_tools if item["name"] == "_non_leaking_target_tool")
|
|
assert tool_info["unity_target"] == "manage_script"
|
|
assert "unity_target" not in tool_info["kwargs"]
|
|
assert tool_info["kwargs"]["annotations"] == {"title": "x"}
|
|
|
|
|
|
def test_tool_registry_rejects_invalid_unity_target_values():
|
|
with pytest.raises(ValueError, match="Invalid unity_target"):
|
|
@mcp_for_unity_tool(unity_target="")
|
|
def _invalid_empty_target_tool():
|
|
return None
|
|
|
|
with pytest.raises(ValueError, match="Invalid unity_target"):
|
|
@mcp_for_unity_tool(unity_target=123) # type: ignore[arg-type]
|
|
def _invalid_non_string_target_tool():
|
|
return None
|