Skip to content

Commit 1d033d8

Browse files
anxkhncopybara-github
authored andcommitted
test: assert stored tool_filter in McpToolset init test
Merge #6327 PiperOrigin-RevId: 948456169
1 parent 4448117 commit 1d033d8

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

tests/unittests/tools/mcp_tool/test_mcp_toolset.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -196,9 +196,8 @@ def test_init_with_tool_filter_list(self):
196196
connection_params=self.mock_stdio_params, tool_filter=tool_filter
197197
)
198198

199-
# The tool filter is stored in the parent BaseToolset class
200-
# We can verify it by checking the filtering behavior in get_tools
201-
assert toolset._is_tool_selected is not None
199+
# The tool filter is stored on the parent BaseToolset class.
200+
assert toolset.tool_filter == tool_filter
202201

203202
def test_init_with_auth(self):
204203
"""Test initialization with authentication."""

0 commit comments

Comments
 (0)