Skip to content

Commit 6aea971

Browse files
committed
test(#3996): avoid shadowing session import
1 parent f8941b9 commit 6aea971

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

pkg/mcp/server_test.go

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -286,17 +286,17 @@ func TestStreamableHTTPHandler_StatelessNegotiates20260728(t *testing.T) {
286286

287287
rec := &recordingRoundTripper{}
288288
client := mcp.NewClient(&mcp.Implementation{Name: "test-client", Version: "0.0.1"}, nil)
289-
session, err := client.Connect(t.Context(), &mcp.StreamableClientTransport{
289+
clientSession, err := client.Connect(t.Context(), &mcp.StreamableClientTransport{
290290
Endpoint: httpSrv.URL,
291291
HTTPClient: &http.Client{Transport: rec},
292292
}, nil)
293293
require.NoError(t, err)
294-
defer session.Close()
294+
defer clientSession.Close()
295295

296-
assert.Equal(t, "2026-07-28", session.InitializeResult().ProtocolVersion,
296+
assert.Equal(t, "2026-07-28", clientSession.InitializeResult().ProtocolVersion,
297297
"stateless handler must negotiate the 2026-07-28 revision with a v1.7 client")
298298

299-
toolsRes, err := session.ListTools(t.Context(), nil)
299+
toolsRes, err := clientSession.ListTools(t.Context(), nil)
300300
require.NoError(t, err)
301301
require.Len(t, toolsRes.Tools, 1)
302302
assert.Equal(t, "root", toolsRes.Tools[0].Name)
@@ -469,12 +469,12 @@ func TestStreamableHTTPHandler_ConcurrentStatelessClients(t *testing.T) {
469469
for range 8 {
470470
g.Go(func() error {
471471
client := mcp.NewClient(&mcp.Implementation{Name: "concurrent-client", Version: "0.0.1"}, nil)
472-
session, err := client.Connect(ctx, &mcp.StreamableClientTransport{Endpoint: httpSrv.URL}, nil)
472+
clientSession, err := client.Connect(ctx, &mcp.StreamableClientTransport{Endpoint: httpSrv.URL}, nil)
473473
if err != nil {
474474
return err
475475
}
476-
defer session.Close()
477-
res, err := session.ListTools(ctx, nil)
476+
defer clientSession.Close()
477+
res, err := clientSession.ListTools(ctx, nil)
478478
if err != nil {
479479
return err
480480
}

0 commit comments

Comments
 (0)