mirror of https://github.com/jlelse/GoBlog
Add two middleware tests
This commit is contained in:
parent
4a4086dcf1
commit
53e90075f5
|
@ -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)
|
||||
|
||||
}
|
Loading…
Reference in New Issue