Skip to content

Commit 0fcf476

Browse files
committed
test: fix server tests for renamed tools
- Update test_list_tools to expect 4 tools instead of 2 - Update test_call_tool_exception to use pia_search_content instead of pia_search - All 20 tests now passing locally Fixes GitHub Actions test failures
1 parent 2992339 commit 0fcf476

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

tests/test_server.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,14 @@ async def test_list_tools():
1111
"""Test that tools are properly listed."""
1212
tools = await list_tools()
1313

14-
assert len(tools) == 2
14+
assert len(tools) == 4
1515
tool_names = [tool.name for tool in tools]
1616

1717
expected_tools = [
18-
"pia_search",
19-
"pia_search_facets",
18+
"pia_search_content",
19+
"pia_search_content_facets",
20+
"pia_search_titles",
21+
"pia_search_titles_facets",
2022
]
2123

2224
for expected_tool in expected_tools:
@@ -45,7 +47,7 @@ async def test_call_tool_exception():
4547
mock_client_instance.post.side_effect = Exception("Test error")
4648
mock_client.return_value.__aenter__.return_value = mock_client_instance
4749

48-
result = await call_tool("pia_search", {"query": "test"})
50+
result = await call_tool("pia_search_content", {"query": "test"})
4951

5052
assert len(result) == 1
5153
assert result[0].type == "text"

0 commit comments

Comments
 (0)