mirror of
https://github.com/openfaas/faas.git
synced 2025-06-18 20:16:37 +00:00
af0ccc9a45
- Added Copyright in handlers where missing - Renamed Project to Author(s) where needed Signed-off-by: Alex Ellis (VMware) <alexellis2@gmail.com>
69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
// Copyright (c) OpenFaaS Author(s). All rights reserved.
|
|
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
|
|
|
package handlers
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/openfaas/faas/gateway/types"
|
|
)
|
|
|
|
func Test_AuthWithValidPassword_Gives200(t *testing.T) {
|
|
|
|
handler := func(w http.ResponseWriter, r *http.Request) {
|
|
io.WriteString(w, "<html><body>Hello World!</body></html>")
|
|
}
|
|
w := httptest.NewRecorder()
|
|
|
|
wantUser := "admin"
|
|
wantPassword := "password"
|
|
r := httptest.NewRequest(http.MethodGet, "http://localhost:8080", nil)
|
|
r.SetBasicAuth(wantUser, wantPassword)
|
|
wantCredentials := &types.BasicAuthCredentials{
|
|
User: wantUser,
|
|
Password: wantPassword,
|
|
}
|
|
|
|
decorated := DecorateWithBasicAuth(handler, wantCredentials)
|
|
decorated.ServeHTTP(w, r)
|
|
|
|
wantCode := http.StatusOK
|
|
|
|
if w.Code != wantCode {
|
|
t.Errorf("status code, want: %d, got: %d", wantCode, w.Code)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func Test_AuthWithInvalidPassword_Gives403(t *testing.T) {
|
|
|
|
handler := func(w http.ResponseWriter, r *http.Request) {
|
|
io.WriteString(w, "<html><body>Hello World!</body></html>")
|
|
}
|
|
|
|
w := httptest.NewRecorder()
|
|
|
|
wantUser := "admin"
|
|
wantPassword := "test"
|
|
r := httptest.NewRequest(http.MethodGet, "http://localhost:8080", nil)
|
|
r.SetBasicAuth(wantUser, wantPassword)
|
|
|
|
wantCredentials := &types.BasicAuthCredentials{
|
|
User: wantUser,
|
|
Password: "",
|
|
}
|
|
|
|
decorated := DecorateWithBasicAuth(handler, wantCredentials)
|
|
decorated.ServeHTTP(w, r)
|
|
|
|
wantCode := http.StatusUnauthorized
|
|
if w.Code != wantCode {
|
|
t.Errorf("status code, want: %d, got: %d", wantCode, w.Code)
|
|
t.Fail()
|
|
}
|
|
}
|