func (s *AuthSuite) TestRequestBadPassword(c *C) {
a := &AuthMiddleware{Username: "******", Password: "******"}
h := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
io.WriteString(w, "treasure")
})
auth, err := a.NewHandler(h)
c.Assert(err, IsNil)
srv := httptest.NewServer(auth)
defer srv.Close()
// bad pass
re, _, err := testutils.Get(srv.URL, testutils.BasicAuth(a.Username, "open please"))
c.Assert(err, IsNil)
c.Assert(re.StatusCode, Equals, http.StatusForbidden)
// missing header
re, _, err = testutils.Get(srv.URL)
c.Assert(err, IsNil)
c.Assert(re.StatusCode, Equals, http.StatusForbidden)
// malformed header
re, _, err = testutils.Get(srv.URL, testutils.Header("Authorization", "blablabla="))
c.Assert(err, IsNil)
c.Assert(re.StatusCode, Equals, http.StatusForbidden)
}
func (s *AuthSuite) TestRequestSuccess(c *C) {
a := &AuthMiddleware{Username: "******", Password: "******"}
h := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
io.WriteString(w, "treasure")
})
auth, err := a.NewHandler(h)
c.Assert(err, IsNil)
srv := httptest.NewServer(auth)
defer srv.Close()
_, body, err := testutils.Get(srv.URL, testutils.BasicAuth(a.Username, a.Password))
c.Assert(err, IsNil)
c.Assert(string(body), Equals, "treasure")
}