mirror of https://github.com/jlelse/GoBlog
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package maprouter
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMapRouter(t *testing.T) {
|
|
router := &MapRouter{
|
|
DefaultHandler: http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
|
_, _ = rw.Write([]byte("Default"))
|
|
}),
|
|
Handlers: map[string]http.Handler{
|
|
"a.example.org": http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
|
_, _ = rw.Write([]byte("a"))
|
|
}),
|
|
"b.example.org": http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
|
_, _ = rw.Write([]byte("b"))
|
|
}),
|
|
},
|
|
}
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "http://a.example.org", nil)
|
|
rec := httptest.NewRecorder()
|
|
router.ServeHTTP(rec, req)
|
|
assert.Equal(t, "a", rec.Body.String())
|
|
|
|
req = httptest.NewRequest(http.MethodGet, "http://b.example.org", nil)
|
|
rec = httptest.NewRecorder()
|
|
router.ServeHTTP(rec, req)
|
|
assert.Equal(t, "b", rec.Body.String())
|
|
|
|
req = httptest.NewRequest(http.MethodGet, "http://c.example.org", nil)
|
|
rec = httptest.NewRecorder()
|
|
router.ServeHTTP(rec, req)
|
|
assert.Equal(t, "Default", rec.Body.String())
|
|
|
|
router.KeyFunc = func(r *http.Request) string {
|
|
return "a.example.org"
|
|
}
|
|
|
|
req = httptest.NewRequest(http.MethodGet, "http://c.example.org", nil)
|
|
rec = httptest.NewRecorder()
|
|
router.ServeHTTP(rec, req)
|
|
assert.Equal(t, "a", rec.Body.String())
|
|
}
|