mirror of https://github.com/grafana/grafana.git
46 lines
1.4 KiB
Go
46 lines
1.4 KiB
Go
|
package api
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/base64"
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
. "github.com/smartystreets/goconvey/convey"
|
||
|
|
"gopkg.in/macaron.v1"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestBasicAuthenticatedRequest(t *testing.T) {
|
||
|
|
expectedUser := "prometheus"
|
||
|
|
expectedPass := "password"
|
||
|
|
|
||
|
|
Convey("Given a valid set of basic auth credentials", t, func() {
|
||
|
|
httpReq, err := http.NewRequest("GET", "http://localhost:3000/metrics", nil)
|
||
|
|
So(err, ShouldBeNil)
|
||
|
|
req := macaron.Request{
|
||
|
|
Request: httpReq,
|
||
|
|
}
|
||
|
|
encodedCreds := encodeBasicAuthCredentials(expectedUser, expectedPass)
|
||
|
|
req.Header.Add("Authorization", fmt.Sprintf("Basic %s", encodedCreds))
|
||
|
|
authenticated := BasicAuthenticatedRequest(req, expectedUser, expectedPass)
|
||
|
|
So(authenticated, ShouldBeTrue)
|
||
|
|
})
|
||
|
|
|
||
|
|
Convey("Given an invalid set of basic auth credentials", t, func() {
|
||
|
|
httpReq, err := http.NewRequest("GET", "http://localhost:3000/metrics", nil)
|
||
|
|
So(err, ShouldBeNil)
|
||
|
|
req := macaron.Request{
|
||
|
|
Request: httpReq,
|
||
|
|
}
|
||
|
|
encodedCreds := encodeBasicAuthCredentials("invaliduser", "invalidpass")
|
||
|
|
req.Header.Add("Authorization", fmt.Sprintf("Basic %s", encodedCreds))
|
||
|
|
authenticated := BasicAuthenticatedRequest(req, expectedUser, expectedPass)
|
||
|
|
So(authenticated, ShouldBeFalse)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func encodeBasicAuthCredentials(user, pass string) string {
|
||
|
|
creds := fmt.Sprintf("%s:%s", user, pass)
|
||
|
|
return base64.StdEncoding.EncodeToString([]byte(creds))
|
||
|
|
}
|