Add two middleware tests

This commit is contained in:
Jan-Lukas Else 2021-10-30 19:53:22 +02:00
parent 4a4086dcf1
commit 53e90075f5
1 changed files with 45 additions and 0 deletions

45
httpMiddlewares_test.go Normal file
View File

@ -0,0 +1,45 @@
package main
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/assert"
)
func Test_noIndexHeader(t *testing.T) {
h := noIndexHeader(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
// Do nothing
}))
req := httptest.NewRequest(http.MethodGet, "http://example.org", nil)
rec := httptest.NewRecorder()
h.ServeHTTP(rec, req)
assert.Equal(t, "noindex", rec.Result().Header.Get("X-Robots-Tag"))
}
func Test_fixHTTPHandler(t *testing.T) {
var got *http.Request
h := fixHTTPHandler(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
got = r
}))
rec := httptest.NewRecorder()
req := httptest.NewRequest(http.MethodGet, "http://example.org/übung", nil)
h.ServeHTTP(rec, req)
assert.Equal(t, "/übung", got.URL.Path)
assert.Equal(t, "", got.URL.RawPath)
req = httptest.NewRequest(http.MethodGet, "http://example.org/%C3%BCbung", nil)
h.ServeHTTP(rec, req)
assert.Equal(t, "/übung", got.URL.Path)
assert.Equal(t, "", got.URL.RawPath)
}