| // 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" |
| "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 TestTokenSourceGrantTypeOverride(t *testing.T) { |
| wantGrantType := "password" |
| var gotGrantType string |
| ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
| body, err := ioutil.ReadAll(r.Body) |
| if err != nil { |
| t.Errorf("ioutil.ReadAll(r.Body) == %v, %v, want _, <nil>", body, err) |
| } |
| if err := r.Body.Close(); err != nil { |
| t.Errorf("r.Body.Close() == %v, want <nil>", err) |
| } |
| values, err := url.ParseQuery(string(body)) |
| if err != nil { |
| t.Errorf("url.ParseQuery(%q) == %v, %v, want _, <nil>", body, values, err) |
| } |
| gotGrantType = values.Get("grant_type") |
| w.Header().Set("Content-Type", "application/x-www-form-urlencoded") |
| w.Write([]byte("access_token=90d64460d14870c08c81352a05dedd3465940a7c&token_type=bearer")) |
| })) |
| config := &Config{ |
| ClientID: "CLIENT_ID", |
| ClientSecret: "CLIENT_SECRET", |
| Scopes: []string{"scope"}, |
| TokenURL: ts.URL + "/token", |
| EndpointParams: url.Values{ |
| "grant_type": {wantGrantType}, |
| }, |
| } |
| token, err := config.TokenSource(context.Background()).Token() |
| if err != nil { |
| t.Errorf("config.TokenSource(_).Token() == %v, %v, want !<nil>, <nil>", token, err) |
| } |
| if gotGrantType != wantGrantType { |
| t.Errorf("grant_type == %q, want %q", gotGrantType, wantGrantType) |
| } |
| } |
| |
| 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: %q", r.URL) |
| } |
| headerContentType := r.Header.Get("Content-Type") |
| if got, want := headerContentType, "application/x-www-form-urlencoded"; got != want { |
| t.Errorf("Content-Type = %q; want %q", got, want) |
| } |
| body, _ := ioutil.ReadAll(r.Body) |
| const want = "audience=audience1&grant_type=client_credentials&scope=scope1+scope2" |
| if string(body) != want { |
| t.Errorf("Unexpected refresh token payload.\n got: %s\nwant: %s\n", body, want) |
| } |
| w.Header().Set("Content-Type", "application/json") |
| io.WriteString(w, `{"access_token": "foo", "refresh_token": "bar"}`) |
| })) |
| defer ts.Close() |
| conf := newConf(ts.URL) |
| c := conf.Client(context.Background()) |
| c.Get(ts.URL + "/somethingelse") |
| } |