maciej/yarr
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

router base

Browse source
This commit is contained in:
Nazar Kanaev 2021-03-17 15:54:05 +00:00
parent c8bc511e04
commit 9bf7f45354
3 changed files with 59 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

64
src/router/router_test.go
View file

@ -8,7 +8,7 @@ import (
func TestRouter(t *testing.T) {
middlecalled := false
router := NewRouter()
router := NewRouter("")
router.Use(func(c *Context) {
middlecalled = true
c.Next()
@ -20,10 +20,7 @@ func TestRouter(t *testing.T) {
recorder := httptest.NewRecorder()
request := httptest.NewRequest("GET", "/hello/world", nil)
router.ServeHTTP(recorder, request)
body, err := io.ReadAll(recorder.Result().Body)
if err != nil {
t.Error(err)
}
body, _ := io.ReadAll(recorder.Result().Body)
if !middlecalled {
t.Error("middleware not called")
@ -37,7 +34,7 @@ func TestRouter(t *testing.T) {
}
func TestRouterPaths(t *testing.T) {
router := NewRouter()
router := NewRouter("")
router.For("/path/to/foo", func(c *Context) {
c.Out.Write([]byte("foo"))
})
@ -49,17 +46,14 @@ func TestRouterPaths(t *testing.T) {
request := httptest.NewRequest("GET", "/path/to/bar", nil)
router.ServeHTTP(recorder, request)
body, err := io.ReadAll(recorder.Result().Body)
if err != nil {
t.Error(err)
}
body, _ := io.ReadAll(recorder.Result().Body)
if string(body) != "bar" {
t.Error("expected 2nd route to be called")
}
}
func TestRouterMiddlewareIntercept(t *testing.T) {
router := NewRouter()
router := NewRouter("")
router.Use(func(c *Context) {
c.Out.WriteHeader(404)
})
@ -76,17 +70,14 @@ func TestRouterMiddlewareIntercept(t *testing.T) {
if recorder.Result().StatusCode != 404 {
t.Error("expected 404")
}
body, err := io.ReadAll(recorder.Result().Body)
if err != nil {
t.Error(err)
}
body, _ := io.ReadAll(recorder.Result().Body)
if len(body) != 0 {
t.Errorf("expected empty body, got %v", body)
}
}
func TestRouterMiddlewareOrder(t *testing.T) {
router := NewRouter()
router := NewRouter("")
router.Use(func(c *Context) {
c.Out.Write([]byte("foo"))
@ -113,11 +104,44 @@ func TestRouterMiddlewareOrder(t *testing.T) {
if recorder.Result().StatusCode != 200 {
t.Error("expected 200")
}
body, err := io.ReadAll(recorder.Result().Body)
if err != nil {
t.Error(err)
}
body, _ := io.ReadAll(recorder.Result().Body)
if string(body) != "foobar!!!" {
t.Errorf("invalid body, got %#v", string(body))
}
}
func TestRouterBase(t *testing.T) {
router := NewRouter("/foo")
router.For("/bar", func(c *Context) {
c.Out.Write([]byte("!!!"))
})
recorder := httptest.NewRecorder()
request := httptest.NewRequest("GET", "/foo/bar", nil)
router.ServeHTTP(recorder, request)
if recorder.Result().StatusCode != 200 {
t.Error("expected 200")
}
body, _ := io.ReadAll(recorder.Result().Body)
if string(body) != "!!!" {
t.Errorf("invalid body, got %#v", string(body))
}
}
func TestRouterBaseRedirect(t *testing.T) {
router := NewRouter("/foo")
router.For("/", func(c *Context) {
c.Out.Write([]byte("!!!"))
})
recorder := httptest.NewRecorder()
request := httptest.NewRequest("GET", "/foo", nil)
router.ServeHTTP(recorder, request)
if recorder.Result().StatusCode != 302 {
t.Errorf("expected 302, got %d", recorder.Result().StatusCode)
}
}