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.
175 lines
3.9 KiB
175 lines
3.9 KiB
package web
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"regexp"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// These tests can probably be DRY'd up a bunch
|
|
|
|
func makeRouter() *router {
|
|
return &router{
|
|
routes: make([]route, 0),
|
|
notFound: parseHandler(http.NotFound),
|
|
}
|
|
}
|
|
|
|
func chHandler(ch chan string, s string) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
ch <- s
|
|
})
|
|
}
|
|
|
|
var methods = []string{"CONNECT", "DELETE", "GET", "HEAD", "OPTIONS", "PATCH",
|
|
"POST", "PUT", "TRACE", "OTHER"}
|
|
|
|
func TestMethods(t *testing.T) {
|
|
t.Parallel()
|
|
rt := makeRouter()
|
|
ch := make(chan string, 1)
|
|
|
|
rt.Connect("/", chHandler(ch, "CONNECT"))
|
|
rt.Delete("/", chHandler(ch, "DELETE"))
|
|
rt.Get("/", chHandler(ch, "GET"))
|
|
rt.Head("/", chHandler(ch, "HEAD"))
|
|
rt.Options("/", chHandler(ch, "OPTIONS"))
|
|
rt.Patch("/", chHandler(ch, "PATCH"))
|
|
rt.Post("/", chHandler(ch, "POST"))
|
|
rt.Put("/", chHandler(ch, "PUT"))
|
|
rt.Trace("/", chHandler(ch, "TRACE"))
|
|
rt.Handle("/", chHandler(ch, "OTHER"))
|
|
|
|
for _, method := range methods {
|
|
r, _ := http.NewRequest(method, "/", nil)
|
|
w := httptest.NewRecorder()
|
|
rt.route(C{}, w, r)
|
|
select {
|
|
case val := <-ch:
|
|
if val != method {
|
|
t.Error("Got %q, expected %q", val, method)
|
|
}
|
|
case <-time.After(5 * time.Millisecond):
|
|
t.Errorf("Timeout waiting for method %q", method)
|
|
}
|
|
}
|
|
}
|
|
|
|
type testPattern struct{}
|
|
|
|
func (t testPattern) Prefix() string {
|
|
return ""
|
|
}
|
|
|
|
func (t testPattern) Match(r *http.Request, c *C) bool {
|
|
return true
|
|
}
|
|
|
|
func TestPatternTypes(t *testing.T) {
|
|
t.Parallel()
|
|
rt := makeRouter()
|
|
|
|
rt.Get("/hello/carl", http.NotFound)
|
|
rt.Get("/hello/:name", http.NotFound)
|
|
rt.Get(regexp.MustCompile(`^/hello/(?P<name>.+)$`), http.NotFound)
|
|
rt.Get(testPattern{}, http.NotFound)
|
|
}
|
|
|
|
type testHandler chan string
|
|
|
|
func (t testHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
t <- "http"
|
|
}
|
|
func (t testHandler) ServeHTTPC(c C, w http.ResponseWriter, r *http.Request) {
|
|
t <- "httpc"
|
|
}
|
|
|
|
var testHandlerTable = map[string]string{
|
|
"/a": "http fn",
|
|
"/b": "http handler",
|
|
"/c": "web fn",
|
|
"/d": "web handler",
|
|
"/e": "httpc",
|
|
}
|
|
|
|
func TestHandlerTypes(t *testing.T) {
|
|
t.Parallel()
|
|
rt := makeRouter()
|
|
ch := make(chan string, 1)
|
|
|
|
rt.Get("/a", func(w http.ResponseWriter, r *http.Request) {
|
|
ch <- "http fn"
|
|
})
|
|
rt.Get("/b", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
ch <- "http handler"
|
|
}))
|
|
rt.Get("/c", func(c C, w http.ResponseWriter, r *http.Request) {
|
|
ch <- "web fn"
|
|
})
|
|
rt.Get("/d", HandlerFunc(func(c C, w http.ResponseWriter, r *http.Request) {
|
|
ch <- "web handler"
|
|
}))
|
|
rt.Get("/e", testHandler(ch))
|
|
|
|
for route, response := range testHandlerTable {
|
|
r, _ := http.NewRequest("gEt", route, nil)
|
|
w := httptest.NewRecorder()
|
|
rt.route(C{}, w, r)
|
|
select {
|
|
case resp := <-ch:
|
|
if resp != response {
|
|
t.Errorf("Got %q, expected %q", resp, response)
|
|
}
|
|
case <-time.After(5 * time.Millisecond):
|
|
t.Errorf("Timeout waiting for path %q", route)
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
func TestNotFound(t *testing.T) {
|
|
t.Parallel()
|
|
rt := makeRouter()
|
|
|
|
r, _ := http.NewRequest("post", "/", nil)
|
|
w := httptest.NewRecorder()
|
|
rt.route(C{}, w, r)
|
|
if w.Code != 404 {
|
|
t.Errorf("Expected 404, got %d", w.Code)
|
|
}
|
|
|
|
rt.NotFound(func(w http.ResponseWriter, r *http.Request) {
|
|
http.Error(w, "I'm a teapot!", http.StatusTeapot)
|
|
})
|
|
|
|
r, _ = http.NewRequest("post", "/", nil)
|
|
w = httptest.NewRecorder()
|
|
rt.route(C{}, w, r)
|
|
if w.Code != http.StatusTeapot {
|
|
t.Errorf("Expected a teapot, got %d", w.Code)
|
|
}
|
|
}
|
|
|
|
func TestSub(t *testing.T) {
|
|
t.Parallel()
|
|
rt := makeRouter()
|
|
ch := make(chan string, 1)
|
|
|
|
rt.Sub("/hello", func(w http.ResponseWriter, r *http.Request) {
|
|
ch <- r.URL.Path
|
|
})
|
|
|
|
r, _ := http.NewRequest("GET", "/hello/world", nil)
|
|
w := httptest.NewRecorder()
|
|
rt.route(C{}, w, r)
|
|
select {
|
|
case val := <-ch:
|
|
if val != "/hello/world" {
|
|
t.Error("Got %q, expected /hello/world", val)
|
|
}
|
|
case <-time.After(5 * time.Millisecond):
|
|
t.Errorf("Timeout waiting for hello")
|
|
}
|
|
}
|