You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
713 B
35 lines
713 B
package web
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func TestRouterMiddleware(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
m := New()
|
|
ch := make(chan string, 1)
|
|
m.Get("/a", chHandler(ch, "a"))
|
|
m.Get("/b", chHandler(ch, "b"))
|
|
m.Use(m.Router)
|
|
m.Use(func(c *C, h http.Handler) http.Handler {
|
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
|
m := GetMatch(*c)
|
|
if rp := m.RawPattern(); rp != "/a" {
|
|
t.Fatalf("RawPattern was not /a: %v", rp)
|
|
}
|
|
r.URL.Path = "/b"
|
|
h.ServeHTTP(w, r)
|
|
}
|
|
return http.HandlerFunc(fn)
|
|
})
|
|
|
|
r, _ := http.NewRequest("GET", "/a", nil)
|
|
w := httptest.NewRecorder()
|
|
m.ServeHTTP(w, r)
|
|
if v := <-ch; v != "a" {
|
|
t.Errorf("Routing was not frozen! %s", v)
|
|
}
|
|
}
|