mirror of
https://github.com/openfaas/faas.git
synced 2025-06-15 19:56:47 +00:00
fac3345668
This commit replaces occurences of http method strings with the corresponding consts from the http package. *Note* UPDATE is not strictly speaking a valid method and as such isn't part of the http package (should be a PUT or PATCH?) Signed-off-by: John McCabe <john@johnmccabe.net>
37 lines
860 B
Go
37 lines
860 B
Go
package tests
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/openfaas/faas/gateway/handlers"
|
|
)
|
|
|
|
type customHandler struct {
|
|
}
|
|
|
|
func (h customHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
}
|
|
|
|
func Test_HeadersAdded(t *testing.T) {
|
|
rr := httptest.NewRecorder()
|
|
handler := customHandler{}
|
|
host := "store.openfaas.com"
|
|
|
|
decorated := handlers.DecorateWithCORS(handler, host)
|
|
request, _ := http.NewRequest(http.MethodGet, "/", nil)
|
|
decorated.ServeHTTP(rr, request)
|
|
|
|
actual := rr.Header().Get("Access-Control-Allow-Origin")
|
|
if actual != host {
|
|
t.Errorf("Access-Control-Allow-Origin: want: %s got: %s", host, actual)
|
|
}
|
|
|
|
actualMethods := rr.Header().Get("Access-Control-Allow-Methods")
|
|
if actualMethods != http.MethodGet {
|
|
t.Errorf("Access-Control-Allow-Methods: want: %s got: %s", http.MethodGet, actualMethods)
|
|
}
|
|
|
|
}
|