mirror of
https://github.com/interviewstreet/go-jira.git
synced 2025-05-15 22:06:45 +02:00
Adding basic auth and accompanying tests
This commit is contained in:
parent
e04b45321e
commit
e3130864a1
34
jira.go
34
jira.go
@ -281,3 +281,37 @@ func (r *Response) populatePageValues(v interface{}) {
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// BasicAuthTransport is an http.RoundTripper that authenticates all requests
|
||||
// using HTTP Basic Authentication with the provided username and password.
|
||||
type BasicAuthTransport struct {
|
||||
Username string
|
||||
Password string
|
||||
|
||||
// Transport is the underlying HTTP transport to use when making requests.
|
||||
// It will default to http.DefaultTransport if nil.
|
||||
Transport http.RoundTripper
|
||||
}
|
||||
|
||||
// RoundTrip implements the RoundTripper interface. We just add the
|
||||
// basic auth and return the RoundTripper for this transport type.
|
||||
func (t *BasicAuthTransport) RoundTrip(req *http.Request) (*http.Response, error) {
|
||||
req.SetBasicAuth(t.Username, t.Password)
|
||||
return t.transport().RoundTrip(req)
|
||||
}
|
||||
|
||||
// Client returns an *http.Client that makes requests that are authenticated
|
||||
// using HTTP Basic Authentication. This is a nice little bit of sugar
|
||||
// so we can just get the client instead of creating the client in the calling code.
|
||||
// If it's necessary to send more information on client init, the calling code can
|
||||
// always skip this and set the transport itself.
|
||||
func (t *BasicAuthTransport) Client() *http.Client {
|
||||
return &http.Client{Transport: t}
|
||||
}
|
||||
|
||||
func (t *BasicAuthTransport) transport() http.RoundTripper {
|
||||
if t.Transport != nil {
|
||||
return t.Transport
|
||||
}
|
||||
return http.DefaultTransport
|
||||
}
|
||||
|
44
jira_test.go
44
jira_test.go
@ -451,3 +451,47 @@ func TestClient_Do_PagingInfoEmptyByDefault(t *testing.T) {
|
||||
t.Errorf("StartAt not equal to 0")
|
||||
}
|
||||
}
|
||||
|
||||
func TestBasicAuthTransport(t *testing.T) {
|
||||
setup()
|
||||
defer teardown()
|
||||
|
||||
username, password := "username", "password"
|
||||
|
||||
testMux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
u, p, ok := r.BasicAuth()
|
||||
if !ok {
|
||||
t.Errorf("request does not contain basic auth credentials")
|
||||
}
|
||||
if u != username {
|
||||
t.Errorf("request contained basic auth username %q, want %q", u, username)
|
||||
}
|
||||
if p != password {
|
||||
t.Errorf("request contained basic auth password %q, want %q", p, password)
|
||||
}
|
||||
})
|
||||
|
||||
tp := &BasicAuthTransport{
|
||||
Username: username,
|
||||
Password: password,
|
||||
}
|
||||
basicAuthClient, _ := NewClient(tp.Client(), "/")
|
||||
req, _ := basicAuthClient.NewRequest("GET", ".", nil)
|
||||
basicAuthClient.Do(req, nil)
|
||||
}
|
||||
|
||||
func TestBasicAuthTransport_transport(t *testing.T) {
|
||||
// default transport
|
||||
tp := &BasicAuthTransport{}
|
||||
if tp.transport() != http.DefaultTransport {
|
||||
t.Errorf("Expected http.DefaultTransport to be used.")
|
||||
}
|
||||
|
||||
// custom transport
|
||||
tp = &BasicAuthTransport{
|
||||
Transport: &http.Transport{},
|
||||
}
|
||||
if tp.transport() == http.DefaultTransport {
|
||||
t.Errorf("Expected custom transport to be used.")
|
||||
}
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user