Function TestSessionPrompterPromptActions

internal/ghmcp/oauth_test.go:143–195

(t *testing.T)

}

func TestSessionPrompterPromptActions(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name        string
		action      string
		wantDecline bool
	}{
		{name: "accept", action: "accept", wantDecline: false},
		{name: "decline", action: "decline", wantDecline: true},
		{name: "cancel", action: "cancel", wantDecline: true},
	}

	caps := &mcp.ClientCapabilities{Elicitation: &mcp.ElicitationCapabilities{
		URL:  &mcp.URLElicitationCapabilities{},
		Form: &mcp.FormElicitationCapabilities{},
	}}

	for _, tc := range tests {
		// URL and form modes share the accept/decline mapping; cover both.
		for _, mode := range []string{"url", "form"} {
			t.Run(tc.name+"/"+mode, func(t *testing.T) {
				t.Parallel()

				handler := func(_ context.Context, _ *mcp.ElicitRequest) (*mcp.ElicitResult, error) {
					return &mcp.ElicitResult{Action: tc.action}, nil
				}

				got := runProbe(t, caps, handler, func(ctx context.Context, p *sessionPrompter) string {
					var err error
					if mode == "url" {
						err = p.PromptURL(ctx, oauth.Prompt{Message: "msg", URL: "https://example.com/auth"})
					} else {
						err = p.PromptForm(ctx, oauth.Prompt{Message: "msg"})
					}
					if err == nil {
						return "ok"
					}
					if err == oauth.ErrPromptDeclined {
						return "declined"
					}
					return "error: " + err.Error()
				})

				if tc.wantDecline {
					assert.Equal(t, "declined", got)
				} else {
					assert.Equal(t, "ok", got)
				}
			})
		}
	}
}

// TestSessionPrompterTransportError verifies that a prompt which fails to be
// delivered (the client errors instead of returning an action) is reported as