forked from Mirrors/oauth2
86 lines
2.4 KiB
Go
86 lines
2.4 KiB
Go
|
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) {
|
||
|
|
DefaultTransport = http.DefaultTransport
|
||
|
|
|
||
|
|
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) {
|
||
|
|
conf := newTestConf()
|
||
|
|
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) {
|
||
|
|
conf := newTestConf()
|
||
|
|
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) {
|
||
|
|
DefaultTransport = http.DefaultTransport
|
||
|
|
|
||
|
|
fetcher := newTestConf().(tokenFetcher)
|
||
|
|
_, err := fetcher.fetchToken(&Token{})
|
||
|
|
if err == nil {
|
||
|
|
t.Fatalf("Fetch should return an error if no refresh token is set")
|
||
|
|
}
|
||
|
|
}
|