mirror of
https://github.com/openfaas/faas.git
synced 2025-06-09 00:36:46 +00:00
c48be47db9
Headers such as Realm / etc are important for auth proxies, so this change and unit test make sure they get copied back to the client. Signed-off-by: Alex Ellis <alexellis2@gmail.com>
30 lines
594 B
Go
30 lines
594 B
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func Test_makeLogger_CopiesResponseHeaders(t *testing.T) {
|
|
handler := http.HandlerFunc(makeLogger(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("X-Unit-Test", "true")
|
|
})))
|
|
|
|
s := httptest.NewServer(handler)
|
|
defer s.Close()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, s.URL, nil)
|
|
rr := httptest.NewRecorder()
|
|
|
|
handler.ServeHTTP(rr, req)
|
|
|
|
got := rr.Header().Get("X-Unit-Test")
|
|
want := "true"
|
|
if want != got {
|
|
t.Errorf("Header X-Unit-Test, want: %s, got %s", want, got)
|
|
}
|
|
|
|
}
|