File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -97,9 +97,13 @@ def tool_b() -> str:
9797 result_a = await server .call_tool ("do_work" , {}, tenant_id = "tenant-a" )
9898 result_b = await server .call_tool ("do_work" , {}, tenant_id = "tenant-b" )
9999
100- # Results are non-empty (structured output returns a tuple)
101- assert result_a is not None
102- assert result_b is not None
100+ # Structured output returns (unstructured_content, structured_content)
101+ assert isinstance (result_a , tuple )
102+ assert isinstance (result_b , tuple )
103+ assert result_a [0 ][0 ].text == "result-a" # type: ignore[union-attr]
104+ assert result_a [1 ] == {"result" : "result-a" }
105+ assert result_b [0 ][0 ].text == "result-b" # type: ignore[union-attr]
106+ assert result_b [1 ] == {"result" : "result-b" }
103107
104108
105109async def test_call_tool_wrong_tenant_raises ():
You can’t perform that action at this time.
0 commit comments