forked from Mirrors/oauth2
96 lines
2.7 KiB
Go
96 lines
2.7 KiB
Go
// Copyright 2014 The oauth2 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 oauth2
|
|
|
|
import (
|
|
"errors"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"testing"
|
|
)
|
|
|
|
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 newTestConf() *Config {
|
|
conf, _ := NewConfig(&Options{
|
|
ClientID: "CLIENT_ID",
|
|
ClientSecret: "CLIENT_SECRET",
|
|
RedirectURL: "REDIRECT_URL",
|
|
Scopes: []string{
|
|
"scope1",
|
|
"scope2",
|
|
},
|
|
AccessType: "offline",
|
|
ApprovalPrompt: "force",
|
|
}, "auth-url", "token-url")
|
|
return conf
|
|
}
|
|
|
|
func TestAuthCodeURL(t *testing.T) {
|
|
conf := newTestConf()
|
|
url, err := conf.AuthCodeURL("foo")
|
|
if err != nil {
|
|
t.Fatalf("Expected to generate an auth URL, failed with %v.", err)
|
|
}
|
|
if url != "auth-url?access_type=offline&approval_prompt=force&client_id=CLIENT_ID&redirect_uri=REDIRECT_URL&response_type=code&scope=scope1+scope2&state=foo" {
|
|
t.Fatalf("Generated auth URL is not the expected. Found %v.", url)
|
|
}
|
|
}
|
|
|
|
func TestExchangePayload(t *testing.T) {
|
|
oldDefaultTransport := http.DefaultTransport
|
|
defer func() {
|
|
http.DefaultTransport = oldDefaultTransport
|
|
}()
|
|
|
|
conf := newTestConf()
|
|
http.DefaultTransport = &mockTransport{
|
|
rt: func(req *http.Request) (resp *http.Response, err error) {
|
|
headerContentType := req.Header.Get("Content-Type")
|
|
if headerContentType != "application/x-www-form-urlencoded" {
|
|
t.Fatalf("Content-Type header is expected to be application/x-www-form-urlencoded, %v found.", headerContentType)
|
|
}
|
|
body, _ := ioutil.ReadAll(req.Body)
|
|
if string(body) != "client_id=CLIENT_ID&client_secret=CLIENT_SECRET&code=exchange-code&grant_type=authorization_code&redirect_uri=REDIRECT_URL&scope=scope1+scope2" {
|
|
t.Fatalf("Exchange payload is found to be %v", string(body))
|
|
}
|
|
return nil, errors.New("no response")
|
|
},
|
|
}
|
|
conf.exchange("exchange-code")
|
|
}
|
|
|
|
func TestExchangingTransport(t *testing.T) {
|
|
oldDefaultTransport := http.DefaultTransport
|
|
defer func() {
|
|
http.DefaultTransport = oldDefaultTransport
|
|
}()
|
|
|
|
conf := newTestConf()
|
|
http.DefaultTransport = &mockTransport{
|
|
rt: func(req *http.Request) (resp *http.Response, err error) {
|
|
if req.URL.RequestURI() != "token-url" {
|
|
t.Fatalf("NewTransportWithCode should have exchanged the code, but it didn't.")
|
|
}
|
|
return nil, errors.New("no response")
|
|
},
|
|
}
|
|
conf.NewTransportWithCode("exchange-code")
|
|
}
|
|
|
|
func TestFetchWithNoRedirect(t *testing.T) {
|
|
fetcher := newTestConf()
|
|
_, err := fetcher.FetchToken(&Token{})
|
|
if err == nil {
|
|
t.Fatalf("Fetch should return an error if no refresh token is set")
|
|
}
|
|
}
|