Skip to content

Commit 06e356f

Browse files
committed
fix broken tests due to new tool being added
1 parent 0575f58 commit 06e356f

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/shared/test_streamable_http.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -882,7 +882,7 @@ async def test_streamablehttp_client_tool_invocation(initialized_client_session:
882882
"""Test client tool invocation."""
883883
# First list tools
884884
tools = await initialized_client_session.list_tools()
885-
assert len(tools.tools) == 6
885+
assert len(tools.tools) == 7
886886
assert tools.tools[0].name == "test_tool"
887887

888888
# Call the tool
@@ -919,7 +919,7 @@ async def test_streamablehttp_client_session_persistence(basic_server: None, bas
919919

920920
# Make multiple requests to verify session persistence
921921
tools = await session.list_tools()
922-
assert len(tools.tools) == 6
922+
assert len(tools.tools) == 7
923923

924924
# Read a resource
925925
resource = await session.read_resource(uri=AnyUrl("foobar://test-persist"))
@@ -948,7 +948,7 @@ async def test_streamablehttp_client_json_response(json_response_server: None, j
948948

949949
# Check tool listing
950950
tools = await session.list_tools()
951-
assert len(tools.tools) == 6
951+
assert len(tools.tools) == 7
952952

953953
# Call a tool and verify JSON response handling
954954
result = await session.call_tool("test_tool", {})
@@ -1019,7 +1019,7 @@ async def test_streamablehttp_client_session_termination(basic_server: None, bas
10191019

10201020
# Make a request to confirm session is working
10211021
tools = await session.list_tools()
1022-
assert len(tools.tools) == 6
1022+
assert len(tools.tools) == 7
10231023

10241024
headers: dict[str, str] = {}
10251025
if captured_session_id:
@@ -1085,7 +1085,7 @@ async def mock_delete(self: httpx.AsyncClient, *args: Any, **kwargs: Any) -> htt
10851085

10861086
# Make a request to confirm session is working
10871087
tools = await session.list_tools()
1088-
assert len(tools.tools) == 6
1088+
assert len(tools.tools) == 7
10891089

10901090
headers: dict[str, str] = {}
10911091
if captured_session_id:

0 commit comments

Comments
 (0)