98 lines
3.1 KiB
Go
98 lines
3.1 KiB
Go
|
// Copyright 2014 The Go Authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package clientcredentials
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"net/url"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func newConf(serverURL string) *Config {
|
||
|
|
return &Config{
|
||
|
|
ClientID: "CLIENT_ID",
|
||
|
|
ClientSecret: "CLIENT_SECRET",
|
||
|
|
Scopes: []string{"scope1", "scope2"},
|
||
|
|
TokenURL: serverURL + "/token",
|
||
|
|
EndpointParams: url.Values{"audience": {"audience1"}},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type mockTransport struct {
|
||
|
|
rt func(req *http.Request) (resp *http.Response, err error)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *mockTransport) RoundTrip(req *http.Request) (resp *http.Response, err error) {
|
||
|
|
return t.rt(req)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTokenRequest(t *testing.T) {
|
||
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
if r.URL.String() != "/token" {
|
||
|
|
t.Errorf("authenticate client request URL = %q; want %q", r.URL, "/token")
|
||
|
|
}
|
||
|
|
headerAuth := r.Header.Get("Authorization")
|
||
|
|
if headerAuth != "Basic Q0xJRU5UX0lEOkNMSUVOVF9TRUNSRVQ=" {
|
||
|
|
t.Errorf("Unexpected authorization header, %v is found.", headerAuth)
|
||
|
|
}
|
||
|
|
if got, want := r.Header.Get("Content-Type"), "application/x-www-form-urlencoded"; got != want {
|
||
|
|
t.Errorf("Content-Type header = %q; want %q", got, want)
|
||
|
|
}
|
||
|
|
body, err := ioutil.ReadAll(r.Body)
|
||
|
|
if err != nil {
|
||
|
|
r.Body.Close()
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("failed reading request body: %s.", err)
|
||
|
|
}
|
||
|
|
if string(body) != "audience=audience1&grant_type=client_credentials&scope=scope1+scope2" {
|
||
|
|
t.Errorf("payload = %q; want %q", string(body), "grant_type=client_credentials&scope=scope1+scope2")
|
||
|
|
}
|
||
|
|
w.Header().Set("Content-Type", "application/x-www-form-urlencoded")
|
||
|
|
w.Write([]byte("access_token=90d64460d14870c08c81352a05dedd3465940a7c&token_type=bearer"))
|
||
|
|
}))
|
||
|
|
defer ts.Close()
|
||
|
|
conf := newConf(ts.URL)
|
||
|
|
tok, err := conf.Token(context.Background())
|
||
|
|
if err != nil {
|
||
|
|
t.Error(err)
|
||
|
|
}
|
||
|
|
if !tok.Valid() {
|
||
|
|
t.Fatalf("token invalid. got: %#v", tok)
|
||
|
|
}
|
||
|
|
if tok.AccessToken != "90d64460d14870c08c81352a05dedd3465940a7c" {
|
||
|
|
t.Errorf("Access token = %q; want %q", tok.AccessToken, "90d64460d14870c08c81352a05dedd3465940a7c")
|
||
|
|
}
|
||
|
|
if tok.TokenType != "bearer" {
|
||
|
|
t.Errorf("token type = %q; want %q", tok.TokenType, "bearer")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTokenRefreshRequest(t *testing.T) {
|
||
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
if r.URL.String() == "/somethingelse" {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if r.URL.String() != "/token" {
|
||
|
|
t.Errorf("Unexpected token refresh request URL, %v is found.", r.URL)
|
||
|
|
}
|
||
|
|
headerContentType := r.Header.Get("Content-Type")
|
||
|
|
if headerContentType != "application/x-www-form-urlencoded" {
|
||
|
|
t.Errorf("Unexpected Content-Type header, %v is found.", headerContentType)
|
||
|
|
}
|
||
|
|
body, _ := ioutil.ReadAll(r.Body)
|
||
|
|
if string(body) != "audience=audience1&grant_type=client_credentials&scope=scope1+scope2" {
|
||
|
|
t.Errorf("Unexpected refresh token payload, %v is found.", string(body))
|
||
|
|
}
|
||
|
|
}))
|
||
|
|
defer ts.Close()
|
||
|
|
conf := newConf(ts.URL)
|
||
|
|
c := conf.Client(context.Background())
|
||
|
|
c.Get(ts.URL + "/somethingelse")
|
||
|
|
}
|