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") }