test: address validation feedback on go-sdk canary changes

Agent-Logs-Url: https://github.com/github/gh-aw-mcpg/sessions/4d4e8792-d37a-4028-8eaa-15752f44b0d7

Co-authored-by: lpcox <15877973+lpcox@users.noreply.github.com>

Author: Copilot

internal/mcp/connection_test.go

@@ -982,9 +982,9 @@ func TestPaginateAll(t *testing.T) {
 		callCount := 0
 		_, err := paginateAll("server1", "tools", func(cursor string) (paginatedPage[string], error) {
 			callCount++
-			nextCursor := cursor + "next"
-			if nextCursor == "" {
-				nextCursor = "next"
+			nextCursor := "next"
+			if cursor != "" {
+				nextCursor = cursor + "next"
 			}
 			return paginatedPage[string]{Items: []string{"x"}, NextCursor: nextCursor}, nil
 		})

internal/server/routed_test.go

@@ -718,7 +718,7 @@ func TestRegisterToolWithoutValidation(t *testing.T) {
 		},
 	}, func(ctx context.Context, req *sdk.CallToolRequest, state interface{}) (*sdk.CallToolResult, interface{}, error) {
 		strictHandlerCalled = true
-		return &sdk.CallToolResult{IsError: false}, nil, nil
+		return &sdk.CallToolResult{IsError: false}, state, nil
 	})
 
 	// Use in-memory transports to connect a client to the server and invoke the tool