router_test.go

package lion

import (
	"context"
	"crypto/rand"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"path/filepath"
	"testing"

	"github.com/celrenheit/htest"
	"github.com/fatih/color"
)

var (
	emptyParams = mss{}
)

func TestRouteMatching(t *testing.T) {
	helloHandler := fakeHandler()
	helloNameEscapedParamHandler := fakeHandler()
	helloNameNestedEscapedParamHandler := fakeHandler()
	helloNameHandler := fakeHandler()
	helloNameTweetsHandler := fakeHandler()
	helloNameGetTweetHandler := fakeHandler()
	cartsHandler := fakeHandler()
	getCartHandler := fakeHandler()
	helloContactHandler := fakeHandler()
	helloContactNamedHandler := fakeHandler()
	helloContactNamedDeeperHandler := fakeHandler()
	helloContactNamedSubParamHandler := fakeHandler()
	helloContactByPersonHandler := fakeHandler()
	helloContactByPersonStaticHandler := fakeHandler()
	helloContactByPersonStaticSubHandler := fakeHandler()
	helloContactByPersonToPersonHandler := fakeHandler()
	helloContactByPersonToPersonEscapedHandler := fakeHandler()
	helloContactByPersonAndPathHandler := fakeHandler()
	extensionHandler := fakeHandler()
	usernameHandler := fakeHandler()
	mailAtHandler := fakeHandler()
	wildcardHandler := fakeHandler()
	userProfileHandler := fakeHandler()
	userSuperHandler := fakeHandler()
	userMainWildcard := fakeHandler()
	emptywildcardHandler := fakeHandler()
	unicodeAlphaHandler := fakeHandler()
	regexpRoot := fakeHandler()
	regexpStatic := fakeHandler()
	regexpStaticPrefix := fakeHandler()
	regexpParam3 := fakeHandler()
	regexpStatic2 := fakeHandler()
	regexpOnlyNumbers := fakeHandler()
	regexpABC := fakeHandler()
	regexpABCN := fakeHandler()
	regexpABCAny := fakeHandler()

	routes := []struct {
		Method  string
		Pattern string
		Handler http.Handler
	}{
		{Pattern: "/hello", Handler: helloHandler},
		{Pattern: "/hello/contact", Handler: helloContactHandler},
		{Pattern: `/hello/\:name`, Handler: helloNameEscapedParamHandler},
		{Pattern: "/hello/\\:name/\\:nested/\\:escaped", Handler: helloNameNestedEscapedParamHandler},
		{Pattern: "/hello/:name", Handler: helloNameHandler},
		{Pattern: "/hello/:name/tweets", Handler: helloNameTweetsHandler},
		{Pattern: "/hello/:name/tweets/:id", Handler: helloNameGetTweetHandler},
		{Pattern: "/carts", Handler: cartsHandler},
		{Pattern: "/carts/:cartid", Handler: getCartHandler},
		{Pattern: "/hello/contact/named", Handler: helloContactNamedHandler},
		{Pattern: "/hello/contact/named/deeper", Handler: helloContactNamedDeeperHandler},
		{Pattern: "/hello/contact/named/:param", Handler: helloContactNamedSubParamHandler},
		{Pattern: "/hello/contact/:dest", Handler: helloContactByPersonHandler},
		{Pattern: "/hello/contact/:dest/static", Handler: helloContactByPersonStaticHandler},
		{Pattern: "/hello/contact/:dest/static/sub", Handler: helloContactByPersonStaticSubHandler},
		{Pattern: "/hello/contact/:dest/:from", Handler: helloContactByPersonToPersonHandler},
		{Pattern: "/hello/contact/\\:dest/\\:from", Handler: helloContactByPersonToPersonEscapedHandler},
		{Pattern: "/hello/contact/:dest/*path", Handler: helloContactByPersonAndPathHandler},
		{Pattern: "/extension/:file.:ext", Handler: extensionHandler},
		{Pattern: "/@:username", Handler: usernameHandler},
		{Pattern: "/mail@:domain", Handler: mailAtHandler},
		{Pattern: "/static/*", Handler: wildcardHandler},
		{Pattern: "/users/:userID/profile", Handler: userProfileHandler},
		{Pattern: "/users/super/*", Handler: userSuperHandler},
		{Pattern: "/users/*", Handler: userMainWildcard},
		{Pattern: "/empty/*", Handler: emptywildcardHandler},
		{Pattern: "/α", Handler: unicodeAlphaHandler},
		{Pattern: "/regexp", Handler: regexpRoot},
		{Pattern: "/regexp/static", Handler: regexpStatic},
		{Pattern: "/regexp/bb", Handler: regexpStaticPrefix},
		{Pattern: "/regexp/bbbb", Handler: regexpStatic2},
		{Pattern: "/regexp/:param([a-z]{3})", Handler: regexpParam3},
		{Pattern: "/regexp/n/:n([0-9]+)", Handler: regexpOnlyNumbers},
		{Pattern: "/regexp/abc/:p(a|b/c)", Handler: regexpABC},
		{Pattern: "/regexp/abc/:p(a|b/c)/:n([0-9]+)", Handler: regexpABCN},
		{Pattern: "/regexp/abc/*any", Handler: regexpABCAny},
	}

	tests := []struct {
		Method          string
		Input           string
		ExpectedHandler http.Handler
		ExpectedParams  mss
		ExpectedStatus  int
	}{
		{Input: "/hello", ExpectedHandler: helloHandler, ExpectedParams: emptyParams},
		{Input: "/hello/:name", ExpectedHandler: helloNameEscapedParamHandler, ExpectedParams: emptyParams},
		{Input: "/hello/:name/:nested/:escaped", ExpectedHandler: helloNameNestedEscapedParamHandler, ExpectedParams: emptyParams},
		{Input: "/hello/batman", ExpectedHandler: helloNameHandler, ExpectedParams: mss{"name": "batman"}},
		{Input: "/hello/dot.inthemiddle", ExpectedHandler: helloNameHandler, ExpectedParams: mss{"name": "dot.inthemiddle"}},
		{Input: "/hello/batman/tweets", ExpectedHandler: helloNameTweetsHandler, ExpectedParams: mss{"name": "batman"}},
		{Input: "/hello/batman/tweets/123", ExpectedHandler: helloNameGetTweetHandler, ExpectedParams: mss{"name": "batman", "id": "123"}},
		{Input: "/carts", ExpectedHandler: cartsHandler, ExpectedParams: emptyParams},
		{Input: "/carts/123456", ExpectedHandler: getCartHandler, ExpectedParams: mss{"cartid": "123456"}},
		{Input: "/hello/contact", ExpectedHandler: helloContactHandler, ExpectedParams: emptyParams},
		{Input: "/hello/contact/named", ExpectedHandler: helloContactNamedHandler, ExpectedParams: emptyParams},
		{Input: "/hello/contact/named/deeper", ExpectedHandler: helloContactNamedDeeperHandler, ExpectedParams: emptyParams},
		{Input: "/hello/contact/named/batman", ExpectedHandler: helloContactNamedSubParamHandler, ExpectedParams: mss{"param": "batman"}},
		{Input: "/hello/contact/nameddd", ExpectedHandler: helloContactByPersonHandler, ExpectedParams: mss{"dest": "nameddd"}},
		{Input: "/hello/contact/nameddd/static", ExpectedHandler: helloContactByPersonStaticHandler, ExpectedParams: mss{"dest": "nameddd"}},
		{Input: "/hello/contact/nameddd/static/a/b/c/d", ExpectedHandler: helloContactByPersonAndPathHandler, ExpectedParams: mss{"dest": "nameddd", "path": "static/a/b/c/d"}},
		{Input: "/hello/contact/nameddd/static/sub", ExpectedHandler: helloContactByPersonStaticSubHandler, ExpectedParams: mss{"dest": "nameddd"}},
		{Input: "/hello/contact/nameddd/staticcc", ExpectedHandler: helloContactByPersonToPersonHandler, ExpectedParams: mss{"dest": "nameddd", "from": "staticcc"}},
		{Input: "/hello/contact/batman", ExpectedHandler: helloContactByPersonHandler, ExpectedParams: mss{"dest": "batman"}},
		{Input: "/hello/contact/batman/static", ExpectedHandler: helloContactByPersonStaticHandler, ExpectedParams: mss{"dest": "batman"}},
		{Input: "/hello/contact/batman/robin", ExpectedHandler: helloContactByPersonToPersonHandler, ExpectedParams: mss{"dest": "batman", "from": "robin"}},
		{Input: "/hello/contact/:dest/:from", ExpectedHandler: helloContactByPersonToPersonEscapedHandler, ExpectedParams: emptyParams},
		{Input: "/hello/contact/batman/folder/subfolder/file", ExpectedHandler: helloContactByPersonAndPathHandler, ExpectedParams: mss{"dest": "batman", "path": "folder/subfolder/file"}},
		{Input: "/extension/batman.jpg", ExpectedHandler: extensionHandler, ExpectedParams: mss{"file": "batman", "ext": "jpg"}},
		{Input: "/@celrenheit", ExpectedHandler: usernameHandler, ExpectedParams: mss{"username": "celrenheit"}},
		{Input: "/mail@test.com", ExpectedHandler: mailAtHandler, ExpectedParams: mss{"domain": "test.com"}},
		{Input: "/static/unkownpath/subfolder", ExpectedHandler: wildcardHandler, ExpectedParams: mss{"*": "unkownpath/subfolder"}},
		{Input: "/users/123/profile", ExpectedHandler: userProfileHandler, ExpectedParams: mss{"userID": "123"}},
		{Input: "/users/super/123/okay/yes", ExpectedHandler: userSuperHandler, ExpectedParams: mss{"*": "123/okay/yes"}},
		{Input: "/users/123/okay/yes", ExpectedHandler: userMainWildcard, ExpectedParams: mss{"*": "123/okay/yes"}},
		{Input: "/empty/", ExpectedHandler: emptywildcardHandler, ExpectedParams: mss{"*": ""}},
		{Input: "/carts404", ExpectedHandler: nil, ExpectedParams: emptyParams, ExpectedStatus: http.StatusNotFound},
		{Input: "/α", ExpectedHandler: unicodeAlphaHandler, ExpectedParams: emptyParams},
		{Input: "/hello/أسد", ExpectedHandler: helloNameHandler, ExpectedParams: mss{"name": "أسد"}},
		{Input: "/hello/أسد/tweets", ExpectedHandler: helloNameTweetsHandler, ExpectedParams: mss{"name": "أسد"}},
		{Input: "/regexp", ExpectedHandler: regexpRoot, ExpectedParams: emptyParams},
		{Input: "/regexp/static", ExpectedHandler: regexpStatic, ExpectedParams: emptyParams},
		{Input: "/regexp/aaa", ExpectedHandler: regexpParam3, ExpectedParams: mss{"param": "aaa"}},
		{Input: "/regexp/bb", ExpectedHandler: regexpStaticPrefix, ExpectedParams: emptyParams},
		{Input: "/regexp/bbb", ExpectedHandler: regexpParam3, ExpectedParams: mss{"param": "bbb"}},
		{Input: "/regexp/bbbb", ExpectedHandler: regexpStatic2, ExpectedParams: emptyParams},
		{Input: "/regexp/aaaa", ExpectedHandler: nil, ExpectedParams: emptyParams, ExpectedStatus: http.StatusNotFound},
		{Input: "/regexp/n/123456", ExpectedHandler: regexpOnlyNumbers, ExpectedParams: mss{"n": "123456"}},
		{Input: "/regexp/n/hello", ExpectedHandler: nil, ExpectedParams: emptyParams, ExpectedStatus: http.StatusNotFound},
		{Input: "/regexp/abc/a", ExpectedHandler: regexpABC, ExpectedParams: mss{"p": "a"}},
		{Input: "/regexp/abc/b/c", ExpectedHandler: regexpABC, ExpectedParams: mss{"p": "b/c"}},
		{Input: "/regexp/abc/a/123456", ExpectedHandler: regexpABCN, ExpectedParams: mss{"p": "a", "n": "123456"}},
		{Input: "/regexp/abc/b/c/123456", ExpectedHandler: regexpABCN, ExpectedParams: mss{"p": "b/c", "n": "123456"}},
		{Input: "/regexp/abc/b/c/123456/test", ExpectedHandler: regexpABCAny, ExpectedParams: mss{"any": "b/c/123456/test"}},
	}

	mux := New()
	for _, r := range routes {
		method := "GET"
		if r.Method != "" {
			method = r.Method
		}
		mux.Handle(method, r.Pattern, r.Handler)
	}

	tester := htest.New(t, mux)

	for _, test := range tests {
		method := "GET"
		if test.Method != "" {
			method = test.Method
		}

		req, _ := http.NewRequest(method, test.Input, nil)
		c := newContextWithParent(context.TODO())
		h := mux.hostrm.Match(c, req)
		req = setParamContext(req, c)

		if len(test.ExpectedParams) != len(c.params) {
			t.Errorf("Length missmatch: expected %d but got %d (%v) for path %s", len(test.ExpectedParams), len(c.params), c.toMap(), test.Input)
		}

		// Compare params
		for k, v := range test.ExpectedParams {
			actual := Param(req, k)
			if actual != v {
				t.Errorf("Expected key %s to equal %s but got %s for url: %s", cyan(k), green(v), red(actual), test.Input)
			}
		}

		// Compare handlers
		if fmt.Sprintf("%v", h) != fmt.Sprintf("%v", test.ExpectedHandler) {
			t.Errorf("Handler not match for %s", test.Input)
		}

		expectedStatus := http.StatusOK
		if test.ExpectedStatus != 0 {
			expectedStatus = test.ExpectedStatus
		}

		tester.Request(method, test.Input).Do().
			ExpectStatus(expectedStatus)
	}

	// fmt.Println(matcher.Print(mux.hostrm.defaultRM.(*pathMatcher).matcher))
}

type anyHandler struct{}

func (a anyHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	w.WriteHeader(http.StatusUnauthorized)
	fmt.Fprintf(w, "Any::%s", r.Method)
}

func TestAnyMethod(t *testing.T) {

	l := New()
	l.AnyFunc("/apifunc", func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusForbidden)
		fmt.Fprintf(w, "AnyFunc::%s", r.Method)
	})

	l.Any("/api", anyHandler{})

	test := htest.New(t, l)
	for _, m := range allowedHTTPMethods {

		test.Request(m, "/api").Do().
			ExpectBody("Any::" + m).
			ExpectStatus(http.StatusUnauthorized)

		test.Request(m, "/apifunc").Do().
			ExpectBody("AnyFunc::" + m).
			ExpectStatus(http.StatusForbidden)

	}
}

type testmw struct{}

func (m testmw) ServeNext(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Test-Key", "Test-Value")
		next.ServeHTTP(w, r)
	})
}

func TestMiddleware(t *testing.T) {
	mux := New()
	mux.Use(testmw{})
	mux.Get("/hi", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("Hi!"))
	}))
	htest.New(t, mux).Get("/hi").Do().ExpectHeader("Test-Key", "Test-Value")
}

func TestMiddlewareFunc(t *testing.T) {
	mux := New()
	mux.UseFunc(func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.Header().Set("Test-Key", "Test-Value")
			next.ServeHTTP(w, r)
		})
	})
	mux.Get("/hi", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("Hi!"))
	}))

	htest.New(t, mux).Get("/hi").Do().ExpectHeader("Test-Key", "Test-Value")
}

func TestMiddlewareChain(t *testing.T) {
	mux := New()
	mux.UseFunc(func(next http.Handler) http.Handler {
		return nil
	})
}

func TestMountingSubrouter(t *testing.T) {
	mux := New()

	adminrouter := New()
	adminrouter.GetFunc("/:id", func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("admin", "id")
	})

	mux.Mount("/admin", adminrouter)

	htest.New(t, mux).Get("/admin/123").Do().ExpectHeader("admin", "id")
}

func TestGroupSubGroup(t *testing.T) {
	s := New()

	admin := s.Group("/admin")
	sub := admin.Subrouter()
	sub.UseFunc(func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.Header().Set("Test-Key", "Get")
			next.ServeHTTP(w, r)
		})
	})

	sub.GetFunc("/", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("Get"))
	})

	sub2 := admin.Subrouter()
	sub2.UseFunc(func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.Header().Set("Test-Key", "Put")
			next.ServeHTTP(w, r)
		})
	})

	sub2.PutFunc("/", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("Put"))
	})

	test := htest.New(t, s)
	test.Get("/admin").Do().ExpectHeader("Test-Key", "Get")
	test.Put("/admin").Do().ExpectHeader("Test-Key", "Put")
}

func TestSubrouter(t *testing.T) {
	r := New()
	r.Use(fakeMW("Global", "true"))
	r.Get("/", fakeHandler())

	sub := r.Subrouter()
	sub.Use(fakeMW("Sub", "true"))
	sub.Get("/hello", fakeHandler())

	test := htest.New(t, r)
	test.Get("/").Do().
		ExpectStatus(http.StatusOK).
		ExpectHeader("Global", "true")

	test.Get("/hello").Do().
		ExpectStatus(http.StatusOK).
		ExpectHeader("Global", "true").
		ExpectHeader("Sub", "true")

	// Mounting
	m := New()
	api := m.Group("/api")
	api.Mount("/", sub)
	test = htest.New(t, m)
	test.Get("/api").Do().
		ExpectStatus(http.StatusNotFound)

	test.Get("/api/hello").Do().
		ExpectStatus(http.StatusOK).
		ExpectHeader("Global", "true").
		ExpectHeader("Sub", "true")
}

func TestNamedMiddlewares(t *testing.T) {
	l := New()
	l.DefineFunc("admin", func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.Header().Set("Test-Key", "admin")
			next.ServeHTTP(w, r)
		})
	})

	l.DefineFunc("public", func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.Header().Set("Test-Key", "public")
			next.ServeHTTP(w, r)
		})
	})

	g := l.Group("/admin")
	g.UseNamed("admin")
	g.GetFunc("/test", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("admintest"))
	})

	p := l.Group("/public")
	p.UseNamed("public")
	p.GetFunc("/test", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("publictest"))
	})

	test := htest.New(t, l)
	test.Get("/admin/test").Do().
		ExpectHeader("Test-Key", "admin").
		ExpectBody("admintest")

	test.Get("/public/test").Do().
		ExpectHeader("Test-Key", "public").
		ExpectBody("publictest")
}

func TestEmptyRouter(t *testing.T) {
	l := New()
	htest.New(t, l).Get("/").Do().ExpectStatus(http.StatusNotFound)
}

func TestConflictingParamNames(t *testing.T) {
	l := New()

	l.Get("/artistas/:Anything/discografia/:DNSDiscography/", fakeHandler())
	recv := catchPanic(func() {
		l.Get("/artistas/:DNS/", fakeHandler())
	})
	if recv == nil {
		t.Error("Should have detected a conflicting parameter name")
	}

	l = New()

	l.Get("/artistas/:DNS/", fakeHandler())
	recv = catchPanic(func() {
		l.Get("/artistas/:Anything/discografia/:DNSDiscography/", fakeHandler())
	})
	if recv == nil {
		t.Error("Should have detected a conflicting parameter name")
	}

	l.Get("/wild/*test", fakeHandler())
	recv = catchPanic(func() {
		l.Get("/wild/*different", fakeHandler())
	})
	if recv == nil {
		t.Error("Should have detected a conflicting parameter name")
	}

	l.Get("/wildstar/*", fakeHandler())
	recv = catchPanic(func() {
		l.Get("/wildstar/*different", fakeHandler())
	})
	if recv == nil {
		t.Error("Should have detected a conflicting parameter name")
	}
}

func TestServeFiles(t *testing.T) {
	cwd, _ := os.Getwd()
	// Temporary directory
	dir, err := ioutil.TempDir(cwd, "test_serve")
	if err != nil {
		t.Error(err)
	}
	defer os.RemoveAll(dir)

	// Temporary file in the Temporary directory created previously
	f, err := ioutil.TempFile(dir, "")
	if err != nil {
		t.Error(err)
	}
	f.WriteString("Lion")
	f.Close()

	_, filename := filepath.Split(f.Name())

	l := New()
	test := htest.New(t, l)

	// ServeFiles
	l.ServeFiles("/public", http.Dir(dir))

	// Tests
	test.Get("/public/" + filename).Do().
		ExpectBody("Lion").
		ExpectStatus(http.StatusOK)
	test.Head("/public/"+filename).Do().
		ExpectHeader("Content-type", "text/plain; charset=utf-8").
		ExpectStatus(http.StatusOK)

	// ServeFile
	l.ServeFile("/file", f.Name())

	// Tests
	test.Get("/file").Do().
		ExpectBody("Lion").
		ExpectStatus(http.StatusOK)
	test.Head("/file").Do().
		ExpectHeader("Content-type", "text/plain; charset=utf-8").
		ExpectStatus(http.StatusOK)
}

func TestRouterShouldPanic(t *testing.T) {
	l := New()
	recv := catchPanic(func() {
		l.Get("path", fakeHandler())
	})

	if recv == nil {
		t.Error("Should panic when path does not start with '/'")
	}

	recv = catchPanic(func() {
		l.UseNamed("unknow middleware")
	})

	if recv == nil {
		t.Error("Should panic when using an unknown named middleware")
	}
}

func TestStaticAndWildcardTriggersPanic(t *testing.T) {
	l := New()
	l.Get("/api", fakeHandler())
	l.Get("/*wild", fakeHandler())
	recv := catchPanic(func() {
		htest.New(t, l).Get("/").Do().ExpectStatus(http.StatusOK)
	})
	if recv != nil {
		t.Errorf("Panic triggers")
	}
}

func TestAutomaticOptions(t *testing.T) {
	l := New()
	l.Post("/api", fakeHandler())
	l.Put("/api", fakeHandler())
	l.Patch("/api", fakeHandler())
	l.Trace("/api", fakeHandler())
	test := htest.New(t, l)
	test.Options("/api").Do().
		ExpectStatus(http.StatusOK).
		ExpectHeader("Accept", "POST,PUT,TRACE,PATCH,OPTIONS")

	test.Get("/api").Do().
		ExpectStatus(http.StatusMethodNotAllowed)

	test.Options("/404").Do().
		ExpectStatus(http.StatusNotFound).
		ExpectHeader("Accept", "")

	// Allow custom options handler
	l.Options("/api", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Batman", "Robin")
		w.WriteHeader(http.StatusFound)
	}))
	test.Options("/api").Do().
		ExpectStatus(http.StatusFound).
		ExpectHeader("Batman", "Robin")
}

func TestValidation(t *testing.T) {
	l := New()
	l.Get("/api/:key", fakeHandler())
	recv := catchPanic(func() {
		l.Get("/api/:key/picture/:key", fakeHandler())
	})
	if recv == nil {
		t.Error("Should panic for duplicated parameter names")
	}
	l = New()
	recv = catchPanic(func() {
		l.Get("api2", fakeHandler())
	})
	if recv == nil {
		t.Error("Should panic for path not starting with '/'")
	}

	recv = catchPanic(func() {
		l.Group("api3")
	})
	if recv == nil {
		t.Error("Should panic for group's pattern not starting with '/'")
	}

	recv = catchPanic(func() {
		l.Handle("BATMAN", "/api", fakeHandler())
	})
	if recv == nil {
		t.Error("Should panic for invalid http method")
	}

	recv = catchPanic(func() {
		l.Handle("", "/api2", fakeHandler())
	})
	if recv == nil {
		t.Error("Should panic for empty http method")
	}

	recv = catchPanic(func() {
		l.Handle("GET", "", fakeHandler())
	})
	if recv == nil {
		t.Error("Should panic for empty pattern")
	}

	recv = catchPanic(func() {
		l.Get("/emptypname/:", fakeHandler())
	})

	if recv == nil {
		t.Error("Should panic for an unnamed parameter")
	}
}

func TestTrailingSlashRedirect(t *testing.T) {
	router := New()
	router.Get("/a", fakeHandler())
	router.Get("/b/", fakeHandler())
	router.Get("/noslash/:param", fakeHandler())
	router.Get("/withslash/:param/", fakeHandler())
	test := htest.New(t, router)

	tests := map[string]string{
		"/a/":              "/a",
		"/b":               "/b/",
		"/noslash/here/":   "/noslash/here",
		"/withslash/here/": "/withslash/here",
		"/withslash/":      "",
		"/withslash":       "",
		"/":                "",
	}

	for input, expectedRedirect := range tests {
		code := http.StatusMovedPermanently
		if expectedRedirect == "" {
			code = 404
		}
		test.Get(input).Do().
			ExpectStatus(code).
			ExpectHeader("Location", expectedRedirect)
	}
}

func TestUSEContext(t *testing.T) {
	r := New()
	r.USE(func(next func(Context)) func(Context) {
		return func(c Context) {
			c.WithHeader("FOO", "BAR")
			next(c)
		}
	})
	r.GET("/", func(c Context) {
		c.WriteHeader(http.StatusOK)
	})

	r.USE(func(next func(Context)) func(Context) {
		return func(c Context) {
			if c.GetHeader("NEXT") == "true" {
				next(c)
				return
			}

			c.Error(ErrorUnauthorized)
		}
	})

	r.GET("/shouldnotpass", func(c Context) {
		c.WriteHeader(http.StatusOK)
	})

	test := htest.New(t, r)
	test.Get("/").Do().
		ExpectStatus(http.StatusOK).
		ExpectHeader("FOO", "BAR")

	test.Get("/shouldnotpass").Do().
		ExpectStatus(http.StatusUnauthorized).
		ExpectHeader("FOO", "BAR")

	test.Get("/shouldnotpass").
		SetHeader("NEXT", "true").
		Do().
		ExpectStatus(http.StatusOK).
		ExpectHeader("FOO", "BAR")
}

// Issue #20
func TestMountingIssue20(t *testing.T) {
	l := New()

	sub := New()
	//normal return
	sub.GET("/", func(ctx Context) {
		ctx.Write([]byte("/"))
	})

	//404 page not found ？
	sub.GET("/a", func(ctx Context) {
		ctx.Write([]byte("a"))
	})
	//normal return
	sub.GET("/b", func(ctx Context) {
		ctx.Write([]byte("b"))
	})

	//404 page not found ？
	sub.GET("/c", func(ctx Context) {
		ctx.Write([]byte("c"))
	})
	//normal return
	sub.GET("/d", func(ctx Context) {
		ctx.Write([]byte("d"))
	})

	l.Mount("/api", sub)

	test := htest.New(t, l)
	test.Get("/api").Do().ExpectStatus(http.StatusOK)
	test.Get("/api/a").Do().ExpectStatus(http.StatusOK)
	test.Get("/api/b").Do().ExpectStatus(http.StatusOK)
	test.Get("/api/c").Do().ExpectStatus(http.StatusOK)
	test.Get("/api/d").Do().ExpectStatus(http.StatusOK)
}

func catchPanic(fn func()) (recv interface{}) {
	defer func() {
		if r := recover(); r != nil {
			recv = r
		}
	}()
	fn()
	return
}

// TODO: Find a better way to compare handlers that using a random token
type fakeHandlerType struct{ t string }

func (f *fakeHandlerType) ServeHTTP(w http.ResponseWriter, r *http.Request) {}

func fakeHandler() http.Handler {
	return &fakeHandlerType{t: randToken()}
}

type fakemw struct{ key, val string }

func (f *fakemw) ServeNext(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set(f.key, f.val)
		next.ServeHTTP(w, r)
	})
}

func fakeMW(key, val string) Middleware {
	return &fakemw{key, val}
}

func randToken() string {
	b := make([]byte, 16)
	rand.Read(b)
	return fmt.Sprintf("%x", b)
}

// Test utils

var red = color.New(color.FgRed).SprintFunc()
var green = color.New(color.FgGreen).SprintFunc()
var cyan = color.New(color.FgCyan).SprintFunc()