Commit 253c069b authored by jordi collell's avatar jordi collell

if basic auth fails should write unauthorized to response

parent 64d20349
...@@ -31,6 +31,7 @@ func (a BasicAuth) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error ...@@ -31,6 +31,7 @@ func (a BasicAuth) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error
// Check credentials // Check credentials
if !ok || username != rule.Username || password != rule.Password { if !ok || username != rule.Username || password != rule.Password {
w.Header().Set("WWW-Authenticate", "Basic") w.Header().Set("WWW-Authenticate", "Basic")
w.WriteHeader(http.StatusUnauthorized)
return http.StatusUnauthorized, nil return http.StatusUnauthorized, nil
} }
......
...@@ -24,12 +24,14 @@ func TestBasicAuth(t *testing.T) { ...@@ -24,12 +24,14 @@ func TestBasicAuth(t *testing.T) {
result int result int
cred string cred string
}{ }{
{"/testing", http.StatusUnauthorized, "ttest:test"},
{"/testing", http.StatusOK, "test:ttest"}, {"/testing", http.StatusOK, "test:ttest"},
{"/testing", http.StatusUnauthorized, ""}, {"/testing", http.StatusUnauthorized, ""},
} }
//auth := "Basic " + base64.StdEncoding.EncodeToString([]byte("foo:bar"))
for i, test := range tests { for i, test := range tests {
...@@ -41,7 +43,14 @@ func TestBasicAuth(t *testing.T) { ...@@ -41,7 +43,14 @@ func TestBasicAuth(t *testing.T) {
req.Header.Set("Authorization", auth) req.Header.Set("Authorization", auth)
rec := httptest.NewRecorder() rec := httptest.NewRecorder()
rw.ServeHTTP(rec, req) result, err := rw.ServeHTTP(rec, req)
if err != nil {
t.Fatalf("Test %d: Could not ServeHTTP %v", i, err)
}
if result != test.result {
t.Errorf("Test %d: Expected Header '%d' but was '%d'",
i, test.result, result)
}
if rec.Code != test.result { if rec.Code != test.result {
t.Errorf("Test %d: Expected Header '%d' but was '%d'", t.Errorf("Test %d: Expected Header '%d' but was '%d'",
...@@ -54,5 +63,5 @@ func TestBasicAuth(t *testing.T) { ...@@ -54,5 +63,5 @@ func TestBasicAuth(t *testing.T) {
func contentHandler(w http.ResponseWriter, r *http.Request) (int, error) { func contentHandler(w http.ResponseWriter, r *http.Request) (int, error) {
fmt.Fprintf(w, r.URL.String()) fmt.Fprintf(w, r.URL.String())
return 0, nil return http.StatusOK, nil
} }
\ No newline at end of file
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment