e2cf95ca22Add auth.RequireBearerToken and associated types. This piece of middleware authenticates clients using OAuth 2.0, as specified in the MCP spec. For #237. Usage: ``` st := mcp.NewStreamableServerTransport(...) http.Handle(path, auth.RequireBearerToken(verifier, nil)(st.ServeHTTP)) ```
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
// Copyright 2025 The Go MCP SDK Authors. All rights reserved.
|
|
// Use of this source code is governed by an MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package auth
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestVerify(t *testing.T) {
|
|
ctx := context.Background()
|
|
verifier := func(_ context.Context, token string) (*TokenInfo, error) {
|
|
switch token {
|
|
case "valid":
|
|
return &TokenInfo{Expiration: time.Now().Add(time.Hour)}, nil
|
|
case "invalid":
|
|
return nil, ErrInvalidToken
|
|
case "noexp":
|
|
return &TokenInfo{}, nil
|
|
case "expired":
|
|
return &TokenInfo{Expiration: time.Now().Add(-time.Hour)}, nil
|
|
default:
|
|
return nil, errors.New("unknown")
|
|
}
|
|
}
|
|
|
|
for _, tt := range []struct {
|
|
name string
|
|
opts *RequireBearerTokenOptions
|
|
header string
|
|
wantMsg string
|
|
wantCode int
|
|
}{
|
|
{
|
|
"valid", nil, "Bearer valid",
|
|
"", 0,
|
|
},
|
|
{
|
|
"bad header", nil, "Barer valid",
|
|
"no bearer token", 401,
|
|
},
|
|
{
|
|
"invalid", nil, "bearer invalid",
|
|
"invalid token", 401,
|
|
},
|
|
{
|
|
"no expiration", nil, "Bearer noexp",
|
|
"token missing expiration", 401,
|
|
},
|
|
{
|
|
"expired", nil, "Bearer expired",
|
|
"token expired", 401,
|
|
},
|
|
{
|
|
"missing scope", &RequireBearerTokenOptions{Scopes: []string{"s1"}}, "Bearer valid",
|
|
"insufficient scope", 403,
|
|
},
|
|
} {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
_, gotMsg, gotCode := verify(ctx, verifier, tt.opts, tt.header)
|
|
if gotMsg != tt.wantMsg || gotCode != tt.wantCode {
|
|
t.Errorf("got (%q, %d), want (%q, %d)", gotMsg, gotCode, tt.wantMsg, tt.wantCode)
|
|
}
|
|
})
|
|
}
|
|
}
|