add test

Signed-off-by: Huabing (Robin) Zhao <[email protected]>

Author: zhaohuabing

mcp/mcp_test.go

@@ -1725,3 +1725,30 @@ func TestPointerArgEquivalence(t *testing.T) {
 func ptr[T any](v T) *T {
 	return &v
 }
+
+func TestComplete(t *testing.T) {
+	serverOpts := &ServerOptions{
+		CompletionHandler: func(_ context.Context, request *CompleteRequest) (*CompleteResult, error) {
+			return &CompleteResult{
+				Completion: CompletionResultDetails{
+					Values: []string{"python", "pytorch", "pyside"},
+				},
+			}, nil
+		},
+	}
+	server := NewServer(testImpl, serverOpts)
+	cs, _ := basicClientServerConnection(t, nil, server, func(s *Server) {})
+	_, err := cs.Complete(context.Background(), &CompleteParams{
+		Argument: CompleteParamsArgument{
+			Name:  "language",
+			Value: "py",
+		},
+		Ref: &CompleteReference{
+			Type: "ref/prompt",
+			Name: "code_review",
+		},
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+}