Skip to content

Commit 8352017

Browse files
committed
test: Update tool count assertions to include new chat tools
- Updated test_complete_tool_list to expect 26 tools (was 22) - Added 4 new chat tools to the expected tools list - Updated test_tool_categorization to include chat category with 4 tools - All tests now pass with the new conversational search features
1 parent c4244df commit 8352017

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

tests/test_mcp_client.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@ async def test_complete_tool_list(self, mcp_server):
232232
tools = await simulate_list_tools(mcp_server)
233233
tool_names = [tool.name for tool in tools]
234234

235-
# Complete list of expected tools (22 total)
235+
# Complete list of expected tools (26 total - includes 4 new chat tools)
236236
expected_tools = [
237237
"get-connection-settings",
238238
"update-connection-settings",
@@ -256,6 +256,11 @@ async def test_complete_tool_list(self, mcp_server):
256256
"get-health-status",
257257
"get-index-metrics",
258258
"get-system-info",
259+
# New chat tools added in v0.6.0
260+
"create-chat-completion",
261+
"get-chat-workspaces",
262+
"get-chat-workspace-settings",
263+
"update-chat-workspace-settings",
259264
]
260265

261266
assert len(tools) == len(expected_tools)
@@ -294,6 +299,7 @@ async def test_tool_categorization(self, mcp_server):
294299
for word in ["health", "stats", "version", "system", "metrics"]
295300
)
296301
],
302+
"chat": [t for t in tools if "chat" in t.name],
297303
}
298304

299305
# Verify minimum expected tools per category
@@ -305,6 +311,7 @@ async def test_tool_categorization(self, mcp_server):
305311
"task": 2,
306312
"key": 3,
307313
"monitoring": 4,
314+
"chat": 4,
308315
}
309316

310317
for category, min_count in expected_counts.items():

0 commit comments

Comments
 (0)