@@ -12,49 +12,49 @@ func TestStreamableHttpSessionImplementsSessionWithClientInfo(t *testing.T) {
1212 resourceStore := newSessionResourcesStore ()
1313 templatesStore := newSessionResourceTemplatesStore ()
1414 logStore := newSessionLogLevelsStore ()
15-
15+
1616 // Create a streamable HTTP session
1717 session := newStreamableHttpSession ("test-session" , toolStore , resourceStore , templatesStore , logStore )
18-
18+
1919 // Verify it implements SessionWithClientInfo
2020 var clientSession ClientSession = session
2121 clientInfoSession , ok := clientSession .(SessionWithClientInfo )
2222 if ! ok {
2323 t .Fatal ("streamableHttpSession should implement SessionWithClientInfo" )
2424 }
25-
25+
2626 // Test GetClientInfo with no data set (should return empty)
2727 clientInfo := clientInfoSession .GetClientInfo ()
2828 if clientInfo .Name != "" || clientInfo .Version != "" {
2929 t .Errorf ("expected empty client info, got %+v" , clientInfo )
3030 }
31-
31+
3232 // Test SetClientInfo and GetClientInfo
3333 expectedClientInfo := mcp.Implementation {
3434 Name : "test-client" ,
3535 Version : "1.0.0" ,
3636 }
3737 clientInfoSession .SetClientInfo (expectedClientInfo )
38-
38+
3939 actualClientInfo := clientInfoSession .GetClientInfo ()
4040 if actualClientInfo .Name != expectedClientInfo .Name || actualClientInfo .Version != expectedClientInfo .Version {
4141 t .Errorf ("expected client info %+v, got %+v" , expectedClientInfo , actualClientInfo )
4242 }
43-
43+
4444 // Test GetClientCapabilities with no data set (should return empty)
4545 capabilities := clientInfoSession .GetClientCapabilities ()
4646 if capabilities .Sampling != nil || capabilities .Roots != nil {
4747 t .Errorf ("expected empty client capabilities, got %+v" , capabilities )
4848 }
49-
49+
5050 // Test SetClientCapabilities and GetClientCapabilities
5151 expectedCapabilities := mcp.ClientCapabilities {
5252 Sampling : & struct {}{},
5353 }
5454 clientInfoSession .SetClientCapabilities (expectedCapabilities )
55-
55+
5656 actualCapabilities := clientInfoSession .GetClientCapabilities ()
5757 if actualCapabilities .Sampling == nil {
5858 t .Errorf ("expected sampling capability to be set" )
5959 }
60- }
60+ }
0 commit comments