1335 lines
41 KiB
Go
1335 lines
41 KiB
Go
// Copyright 2012 The Gorilla Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package mux
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/gorilla/context"
|
|
)
|
|
|
|
func (r *Route) GoString() string {
|
|
matchers := make([]string, len(r.matchers))
|
|
for i, m := range r.matchers {
|
|
matchers[i] = fmt.Sprintf("%#v", m)
|
|
}
|
|
return fmt.Sprintf("&Route{matchers:[]matcher{%s}}", strings.Join(matchers, ", "))
|
|
}
|
|
|
|
func (r *routeRegexp) GoString() string {
|
|
return fmt.Sprintf("&routeRegexp{template: %q, matchHost: %t, matchQuery: %t, strictSlash: %t, regexp: regexp.MustCompile(%q), reverse: %q, varsN: %v, varsR: %v", r.template, r.matchHost, r.matchQuery, r.strictSlash, r.regexp.String(), r.reverse, r.varsN, r.varsR)
|
|
}
|
|
|
|
type routeTest struct {
|
|
title string // title of the test
|
|
route *Route // the route being tested
|
|
request *http.Request // a request to test the route
|
|
vars map[string]string // the expected vars of the match
|
|
host string // the expected host of the match
|
|
path string // the expected path of the match
|
|
shouldMatch bool // whether the request is expected to match the route at all
|
|
shouldRedirect bool // whether the request should result in a redirect
|
|
}
|
|
|
|
func TestHost(t *testing.T) {
|
|
// newRequestHost a new request with a method, url, and host header
|
|
newRequestHost := func(method, url, host string) *http.Request {
|
|
req, err := http.NewRequest(method, url, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
req.Host = host
|
|
return req
|
|
}
|
|
|
|
tests := []routeTest{
|
|
{
|
|
title: "Host route match",
|
|
route: new(Route).Host("aaa.bbb.ccc"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc/111/222/333"),
|
|
vars: map[string]string{},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host route, wrong host in request URL",
|
|
route: new(Route).Host("aaa.bbb.ccc"),
|
|
request: newRequest("GET", "http://aaa.222.ccc/111/222/333"),
|
|
vars: map[string]string{},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Host route with port, match",
|
|
route: new(Route).Host("aaa.bbb.ccc:1234"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc:1234/111/222/333"),
|
|
vars: map[string]string{},
|
|
host: "aaa.bbb.ccc:1234",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host route with port, wrong port in request URL",
|
|
route: new(Route).Host("aaa.bbb.ccc:1234"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc:9999/111/222/333"),
|
|
vars: map[string]string{},
|
|
host: "aaa.bbb.ccc:1234",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Host route, match with host in request header",
|
|
route: new(Route).Host("aaa.bbb.ccc"),
|
|
request: newRequestHost("GET", "/111/222/333", "aaa.bbb.ccc"),
|
|
vars: map[string]string{},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host route, wrong host in request header",
|
|
route: new(Route).Host("aaa.bbb.ccc"),
|
|
request: newRequestHost("GET", "/111/222/333", "aaa.222.ccc"),
|
|
vars: map[string]string{},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
// BUG {new(Route).Host("aaa.bbb.ccc:1234"), newRequestHost("GET", "/111/222/333", "aaa.bbb.ccc:1234"), map[string]string{}, "aaa.bbb.ccc:1234", "", true},
|
|
{
|
|
title: "Host route with port, wrong host in request header",
|
|
route: new(Route).Host("aaa.bbb.ccc:1234"),
|
|
request: newRequestHost("GET", "/111/222/333", "aaa.bbb.ccc:9999"),
|
|
vars: map[string]string{},
|
|
host: "aaa.bbb.ccc:1234",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Host route with pattern, match",
|
|
route: new(Route).Host("aaa.{v1:[a-z]{3}}.ccc"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc/111/222/333"),
|
|
vars: map[string]string{"v1": "bbb"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host route with pattern, additional capturing group, match",
|
|
route: new(Route).Host("aaa.{v1:[a-z]{2}(b|c)}.ccc"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc/111/222/333"),
|
|
vars: map[string]string{"v1": "bbb"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host route with pattern, wrong host in request URL",
|
|
route: new(Route).Host("aaa.{v1:[a-z]{3}}.ccc"),
|
|
request: newRequest("GET", "http://aaa.222.ccc/111/222/333"),
|
|
vars: map[string]string{"v1": "bbb"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Host route with multiple patterns, match",
|
|
route: new(Route).Host("{v1:[a-z]{3}}.{v2:[a-z]{3}}.{v3:[a-z]{3}}"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc/111/222/333"),
|
|
vars: map[string]string{"v1": "aaa", "v2": "bbb", "v3": "ccc"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host route with multiple patterns, wrong host in request URL",
|
|
route: new(Route).Host("{v1:[a-z]{3}}.{v2:[a-z]{3}}.{v3:[a-z]{3}}"),
|
|
request: newRequest("GET", "http://aaa.222.ccc/111/222/333"),
|
|
vars: map[string]string{"v1": "aaa", "v2": "bbb", "v3": "ccc"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Host route with hyphenated name and pattern, match",
|
|
route: new(Route).Host("aaa.{v-1:[a-z]{3}}.ccc"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc/111/222/333"),
|
|
vars: map[string]string{"v-1": "bbb"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host route with hyphenated name and pattern, additional capturing group, match",
|
|
route: new(Route).Host("aaa.{v-1:[a-z]{2}(b|c)}.ccc"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc/111/222/333"),
|
|
vars: map[string]string{"v-1": "bbb"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host route with multiple hyphenated names and patterns, match",
|
|
route: new(Route).Host("{v-1:[a-z]{3}}.{v-2:[a-z]{3}}.{v-3:[a-z]{3}}"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc/111/222/333"),
|
|
vars: map[string]string{"v-1": "aaa", "v-2": "bbb", "v-3": "ccc"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route with single pattern with pipe, match",
|
|
route: new(Route).Path("/{category:a|b/c}"),
|
|
request: newRequest("GET", "http://localhost/a"),
|
|
vars: map[string]string{"category": "a"},
|
|
host: "",
|
|
path: "/a",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route with single pattern with pipe, match",
|
|
route: new(Route).Path("/{category:a|b/c}"),
|
|
request: newRequest("GET", "http://localhost/b/c"),
|
|
vars: map[string]string{"category": "b/c"},
|
|
host: "",
|
|
path: "/b/c",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route with multiple patterns with pipe, match",
|
|
route: new(Route).Path("/{category:a|b/c}/{product}/{id:[0-9]+}"),
|
|
request: newRequest("GET", "http://localhost/a/product_name/1"),
|
|
vars: map[string]string{"category": "a", "product": "product_name", "id": "1"},
|
|
host: "",
|
|
path: "/a/product_name/1",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route with multiple patterns with pipe, match",
|
|
route: new(Route).Path("/{category:a|b/c}/{product}/{id:[0-9]+}"),
|
|
request: newRequest("GET", "http://localhost/b/c/product_name/1"),
|
|
vars: map[string]string{"category": "b/c", "product": "product_name", "id": "1"},
|
|
host: "",
|
|
path: "/b/c/product_name/1",
|
|
shouldMatch: true,
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestPath(t *testing.T) {
|
|
tests := []routeTest{
|
|
{
|
|
title: "Path route, match",
|
|
route: new(Route).Path("/111/222/333"),
|
|
request: newRequest("GET", "http://localhost/111/222/333"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111/222/333",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route, match with trailing slash in request and path",
|
|
route: new(Route).Path("/111/"),
|
|
request: newRequest("GET", "http://localhost/111/"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111/",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route, do not match with trailing slash in path",
|
|
route: new(Route).Path("/111/"),
|
|
request: newRequest("GET", "http://localhost/111"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Path route, do not match with trailing slash in request",
|
|
route: new(Route).Path("/111"),
|
|
request: newRequest("GET", "http://localhost/111/"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111/",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Path route, wrong path in request in request URL",
|
|
route: new(Route).Path("/111/222/333"),
|
|
request: newRequest("GET", "http://localhost/1/2/3"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111/222/333",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Path route with pattern, match",
|
|
route: new(Route).Path("/111/{v1:[0-9]{3}}/333"),
|
|
request: newRequest("GET", "http://localhost/111/222/333"),
|
|
vars: map[string]string{"v1": "222"},
|
|
host: "",
|
|
path: "/111/222/333",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route with pattern, URL in request does not match",
|
|
route: new(Route).Path("/111/{v1:[0-9]{3}}/333"),
|
|
request: newRequest("GET", "http://localhost/111/aaa/333"),
|
|
vars: map[string]string{"v1": "222"},
|
|
host: "",
|
|
path: "/111/222/333",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Path route with multiple patterns, match",
|
|
route: new(Route).Path("/{v1:[0-9]{3}}/{v2:[0-9]{3}}/{v3:[0-9]{3}}"),
|
|
request: newRequest("GET", "http://localhost/111/222/333"),
|
|
vars: map[string]string{"v1": "111", "v2": "222", "v3": "333"},
|
|
host: "",
|
|
path: "/111/222/333",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route with multiple patterns, URL in request does not match",
|
|
route: new(Route).Path("/{v1:[0-9]{3}}/{v2:[0-9]{3}}/{v3:[0-9]{3}}"),
|
|
request: newRequest("GET", "http://localhost/111/aaa/333"),
|
|
vars: map[string]string{"v1": "111", "v2": "222", "v3": "333"},
|
|
host: "",
|
|
path: "/111/222/333",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Path route with multiple patterns with pipe, match",
|
|
route: new(Route).Path("/{category:a|(b/c)}/{product}/{id:[0-9]+}"),
|
|
request: newRequest("GET", "http://localhost/a/product_name/1"),
|
|
vars: map[string]string{"category": "a", "product": "product_name", "id": "1"},
|
|
host: "",
|
|
path: "/a/product_name/1",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route with hyphenated name and pattern, match",
|
|
route: new(Route).Path("/111/{v-1:[0-9]{3}}/333"),
|
|
request: newRequest("GET", "http://localhost/111/222/333"),
|
|
vars: map[string]string{"v-1": "222"},
|
|
host: "",
|
|
path: "/111/222/333",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route with multiple hyphenated names and patterns, match",
|
|
route: new(Route).Path("/{v-1:[0-9]{3}}/{v-2:[0-9]{3}}/{v-3:[0-9]{3}}"),
|
|
request: newRequest("GET", "http://localhost/111/222/333"),
|
|
vars: map[string]string{"v-1": "111", "v-2": "222", "v-3": "333"},
|
|
host: "",
|
|
path: "/111/222/333",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Path route with multiple hyphenated names and patterns with pipe, match",
|
|
route: new(Route).Path("/{product-category:a|(b/c)}/{product-name}/{product-id:[0-9]+}"),
|
|
request: newRequest("GET", "http://localhost/a/product_name/1"),
|
|
vars: map[string]string{"product-category": "a", "product-name": "product_name", "product-id": "1"},
|
|
host: "",
|
|
path: "/a/product_name/1",
|
|
shouldMatch: true,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestPathPrefix(t *testing.T) {
|
|
tests := []routeTest{
|
|
{
|
|
title: "PathPrefix route, match",
|
|
route: new(Route).PathPrefix("/111"),
|
|
request: newRequest("GET", "http://localhost/111/222/333"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "PathPrefix route, match substring",
|
|
route: new(Route).PathPrefix("/1"),
|
|
request: newRequest("GET", "http://localhost/111/222/333"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/1",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "PathPrefix route, URL prefix in request does not match",
|
|
route: new(Route).PathPrefix("/111"),
|
|
request: newRequest("GET", "http://localhost/1/2/3"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "PathPrefix route with pattern, match",
|
|
route: new(Route).PathPrefix("/111/{v1:[0-9]{3}}"),
|
|
request: newRequest("GET", "http://localhost/111/222/333"),
|
|
vars: map[string]string{"v1": "222"},
|
|
host: "",
|
|
path: "/111/222",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "PathPrefix route with pattern, URL prefix in request does not match",
|
|
route: new(Route).PathPrefix("/111/{v1:[0-9]{3}}"),
|
|
request: newRequest("GET", "http://localhost/111/aaa/333"),
|
|
vars: map[string]string{"v1": "222"},
|
|
host: "",
|
|
path: "/111/222",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "PathPrefix route with multiple patterns, match",
|
|
route: new(Route).PathPrefix("/{v1:[0-9]{3}}/{v2:[0-9]{3}}"),
|
|
request: newRequest("GET", "http://localhost/111/222/333"),
|
|
vars: map[string]string{"v1": "111", "v2": "222"},
|
|
host: "",
|
|
path: "/111/222",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "PathPrefix route with multiple patterns, URL prefix in request does not match",
|
|
route: new(Route).PathPrefix("/{v1:[0-9]{3}}/{v2:[0-9]{3}}"),
|
|
request: newRequest("GET", "http://localhost/111/aaa/333"),
|
|
vars: map[string]string{"v1": "111", "v2": "222"},
|
|
host: "",
|
|
path: "/111/222",
|
|
shouldMatch: false,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestHostPath(t *testing.T) {
|
|
tests := []routeTest{
|
|
{
|
|
title: "Host and Path route, match",
|
|
route: new(Route).Host("aaa.bbb.ccc").Path("/111/222/333"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc/111/222/333"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host and Path route, wrong host in request URL",
|
|
route: new(Route).Host("aaa.bbb.ccc").Path("/111/222/333"),
|
|
request: newRequest("GET", "http://aaa.222.ccc/111/222/333"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Host and Path route with pattern, match",
|
|
route: new(Route).Host("aaa.{v1:[a-z]{3}}.ccc").Path("/111/{v2:[0-9]{3}}/333"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc/111/222/333"),
|
|
vars: map[string]string{"v1": "bbb", "v2": "222"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "/111/222/333",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host and Path route with pattern, URL in request does not match",
|
|
route: new(Route).Host("aaa.{v1:[a-z]{3}}.ccc").Path("/111/{v2:[0-9]{3}}/333"),
|
|
request: newRequest("GET", "http://aaa.222.ccc/111/222/333"),
|
|
vars: map[string]string{"v1": "bbb", "v2": "222"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "/111/222/333",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Host and Path route with multiple patterns, match",
|
|
route: new(Route).Host("{v1:[a-z]{3}}.{v2:[a-z]{3}}.{v3:[a-z]{3}}").Path("/{v4:[0-9]{3}}/{v5:[0-9]{3}}/{v6:[0-9]{3}}"),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc/111/222/333"),
|
|
vars: map[string]string{"v1": "aaa", "v2": "bbb", "v3": "ccc", "v4": "111", "v5": "222", "v6": "333"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "/111/222/333",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Host and Path route with multiple patterns, URL in request does not match",
|
|
route: new(Route).Host("{v1:[a-z]{3}}.{v2:[a-z]{3}}.{v3:[a-z]{3}}").Path("/{v4:[0-9]{3}}/{v5:[0-9]{3}}/{v6:[0-9]{3}}"),
|
|
request: newRequest("GET", "http://aaa.222.ccc/111/222/333"),
|
|
vars: map[string]string{"v1": "aaa", "v2": "bbb", "v3": "ccc", "v4": "111", "v5": "222", "v6": "333"},
|
|
host: "aaa.bbb.ccc",
|
|
path: "/111/222/333",
|
|
shouldMatch: false,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestHeaders(t *testing.T) {
|
|
// newRequestHeaders creates a new request with a method, url, and headers
|
|
newRequestHeaders := func(method, url string, headers map[string]string) *http.Request {
|
|
req, err := http.NewRequest(method, url, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
for k, v := range headers {
|
|
req.Header.Add(k, v)
|
|
}
|
|
return req
|
|
}
|
|
|
|
tests := []routeTest{
|
|
{
|
|
title: "Headers route, match",
|
|
route: new(Route).Headers("foo", "bar", "baz", "ding"),
|
|
request: newRequestHeaders("GET", "http://localhost", map[string]string{"foo": "bar", "baz": "ding"}),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Headers route, bad header values",
|
|
route: new(Route).Headers("foo", "bar", "baz", "ding"),
|
|
request: newRequestHeaders("GET", "http://localhost", map[string]string{"foo": "bar", "baz": "dong"}),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Headers route, regex header values to match",
|
|
route: new(Route).Headers("foo", "ba[zr]"),
|
|
request: newRequestHeaders("GET", "http://localhost", map[string]string{"foo": "bar"}),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Headers route, regex header values to match",
|
|
route: new(Route).HeadersRegexp("foo", "ba[zr]"),
|
|
request: newRequestHeaders("GET", "http://localhost", map[string]string{"foo": "baz"}),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
|
|
}
|
|
|
|
func TestMethods(t *testing.T) {
|
|
tests := []routeTest{
|
|
{
|
|
title: "Methods route, match GET",
|
|
route: new(Route).Methods("GET", "POST"),
|
|
request: newRequest("GET", "http://localhost"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Methods route, match POST",
|
|
route: new(Route).Methods("GET", "POST"),
|
|
request: newRequest("POST", "http://localhost"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Methods route, bad method",
|
|
route: new(Route).Methods("GET", "POST"),
|
|
request: newRequest("PUT", "http://localhost"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestQueries(t *testing.T) {
|
|
tests := []routeTest{
|
|
{
|
|
title: "Queries route, match",
|
|
route: new(Route).Queries("foo", "bar", "baz", "ding"),
|
|
request: newRequest("GET", "http://localhost?foo=bar&baz=ding"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route, match with a query string",
|
|
route: new(Route).Host("www.example.com").Path("/api").Queries("foo", "bar", "baz", "ding"),
|
|
request: newRequest("GET", "http://www.example.com/api?foo=bar&baz=ding"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route, match with a query string out of order",
|
|
route: new(Route).Host("www.example.com").Path("/api").Queries("foo", "bar", "baz", "ding"),
|
|
request: newRequest("GET", "http://www.example.com/api?baz=ding&foo=bar"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route, bad query",
|
|
route: new(Route).Queries("foo", "bar", "baz", "ding"),
|
|
request: newRequest("GET", "http://localhost?foo=bar&baz=dong"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Queries route with pattern, match",
|
|
route: new(Route).Queries("foo", "{v1}"),
|
|
request: newRequest("GET", "http://localhost?foo=bar"),
|
|
vars: map[string]string{"v1": "bar"},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with multiple patterns, match",
|
|
route: new(Route).Queries("foo", "{v1}", "baz", "{v2}"),
|
|
request: newRequest("GET", "http://localhost?foo=bar&baz=ding"),
|
|
vars: map[string]string{"v1": "bar", "v2": "ding"},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with regexp pattern, match",
|
|
route: new(Route).Queries("foo", "{v1:[0-9]+}"),
|
|
request: newRequest("GET", "http://localhost?foo=10"),
|
|
vars: map[string]string{"v1": "10"},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with regexp pattern, regexp does not match",
|
|
route: new(Route).Queries("foo", "{v1:[0-9]+}"),
|
|
request: newRequest("GET", "http://localhost?foo=a"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Queries route with regexp pattern with quantifier, match",
|
|
route: new(Route).Queries("foo", "{v1:[0-9]{1}}"),
|
|
request: newRequest("GET", "http://localhost?foo=1"),
|
|
vars: map[string]string{"v1": "1"},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with regexp pattern with quantifier, additional variable in query string, match",
|
|
route: new(Route).Queries("foo", "{v1:[0-9]{1}}"),
|
|
request: newRequest("GET", "http://localhost?bar=2&foo=1"),
|
|
vars: map[string]string{"v1": "1"},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with regexp pattern with quantifier, regexp does not match",
|
|
route: new(Route).Queries("foo", "{v1:[0-9]{1}}"),
|
|
request: newRequest("GET", "http://localhost?foo=12"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Queries route with regexp pattern with quantifier, additional capturing group",
|
|
route: new(Route).Queries("foo", "{v1:[0-9]{1}(a|b)}"),
|
|
request: newRequest("GET", "http://localhost?foo=1a"),
|
|
vars: map[string]string{"v1": "1a"},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with regexp pattern with quantifier, additional variable in query string, regexp does not match",
|
|
route: new(Route).Queries("foo", "{v1:[0-9]{1}}"),
|
|
request: newRequest("GET", "http://localhost?foo=12"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Queries route with hyphenated name, match",
|
|
route: new(Route).Queries("foo", "{v-1}"),
|
|
request: newRequest("GET", "http://localhost?foo=bar"),
|
|
vars: map[string]string{"v-1": "bar"},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with multiple hyphenated names, match",
|
|
route: new(Route).Queries("foo", "{v-1}", "baz", "{v-2}"),
|
|
request: newRequest("GET", "http://localhost?foo=bar&baz=ding"),
|
|
vars: map[string]string{"v-1": "bar", "v-2": "ding"},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with hyphenate name and pattern, match",
|
|
route: new(Route).Queries("foo", "{v-1:[0-9]+}"),
|
|
request: newRequest("GET", "http://localhost?foo=10"),
|
|
vars: map[string]string{"v-1": "10"},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with hyphenated name and pattern with quantifier, additional capturing group",
|
|
route: new(Route).Queries("foo", "{v-1:[0-9]{1}(a|b)}"),
|
|
request: newRequest("GET", "http://localhost?foo=1a"),
|
|
vars: map[string]string{"v-1": "1a"},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with empty value, should match",
|
|
route: new(Route).Queries("foo", ""),
|
|
request: newRequest("GET", "http://localhost?foo=bar"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with empty value and no parameter in request, should not match",
|
|
route: new(Route).Queries("foo", ""),
|
|
request: newRequest("GET", "http://localhost"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Queries route with empty value and empty parameter in request, should match",
|
|
route: new(Route).Queries("foo", ""),
|
|
request: newRequest("GET", "http://localhost?foo="),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route with overlapping value, should not match",
|
|
route: new(Route).Queries("foo", "bar"),
|
|
request: newRequest("GET", "http://localhost?foo=barfoo"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Queries route with no parameter in request, should not match",
|
|
route: new(Route).Queries("foo", "{bar}"),
|
|
request: newRequest("GET", "http://localhost"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
title: "Queries route with empty parameter in request, should match",
|
|
route: new(Route).Queries("foo", "{bar}"),
|
|
request: newRequest("GET", "http://localhost?foo="),
|
|
vars: map[string]string{"foo": ""},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Queries route, bad submatch",
|
|
route: new(Route).Queries("foo", "bar", "baz", "ding"),
|
|
request: newRequest("GET", "http://localhost?fffoo=bar&baz=dingggg"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestSchemes(t *testing.T) {
|
|
tests := []routeTest{
|
|
// Schemes
|
|
{
|
|
title: "Schemes route, match https",
|
|
route: new(Route).Schemes("https", "ftp"),
|
|
request: newRequest("GET", "https://localhost"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Schemes route, match ftp",
|
|
route: new(Route).Schemes("https", "ftp"),
|
|
request: newRequest("GET", "ftp://localhost"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "Schemes route, bad scheme",
|
|
route: new(Route).Schemes("https", "ftp"),
|
|
request: newRequest("GET", "http://localhost"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestMatcherFunc(t *testing.T) {
|
|
m := func(r *http.Request, m *RouteMatch) bool {
|
|
if r.URL.Host == "aaa.bbb.ccc" {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
tests := []routeTest{
|
|
{
|
|
title: "MatchFunc route, match",
|
|
route: new(Route).MatcherFunc(m),
|
|
request: newRequest("GET", "http://aaa.bbb.ccc"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "MatchFunc route, non-match",
|
|
route: new(Route).MatcherFunc(m),
|
|
request: newRequest("GET", "http://aaa.222.ccc"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "",
|
|
shouldMatch: false,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestBuildVarsFunc(t *testing.T) {
|
|
tests := []routeTest{
|
|
{
|
|
title: "BuildVarsFunc set on route",
|
|
route: new(Route).Path(`/111/{v1:\d}{v2:.*}`).BuildVarsFunc(func(vars map[string]string) map[string]string {
|
|
vars["v1"] = "3"
|
|
vars["v2"] = "a"
|
|
return vars
|
|
}),
|
|
request: newRequest("GET", "http://localhost/111/2"),
|
|
path: "/111/3a",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
title: "BuildVarsFunc set on route and parent route",
|
|
route: new(Route).PathPrefix(`/{v1:\d}`).BuildVarsFunc(func(vars map[string]string) map[string]string {
|
|
vars["v1"] = "2"
|
|
return vars
|
|
}).Subrouter().Path(`/{v2:\w}`).BuildVarsFunc(func(vars map[string]string) map[string]string {
|
|
vars["v2"] = "b"
|
|
return vars
|
|
}),
|
|
request: newRequest("GET", "http://localhost/1/a"),
|
|
path: "/2/b",
|
|
shouldMatch: true,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestSubRouter(t *testing.T) {
|
|
subrouter1 := new(Route).Host("{v1:[a-z]+}.google.com").Subrouter()
|
|
subrouter2 := new(Route).PathPrefix("/foo/{v1}").Subrouter()
|
|
|
|
tests := []routeTest{
|
|
{
|
|
route: subrouter1.Path("/{v2:[a-z]+}"),
|
|
request: newRequest("GET", "http://aaa.google.com/bbb"),
|
|
vars: map[string]string{"v1": "aaa", "v2": "bbb"},
|
|
host: "aaa.google.com",
|
|
path: "/bbb",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
route: subrouter1.Path("/{v2:[a-z]+}"),
|
|
request: newRequest("GET", "http://111.google.com/111"),
|
|
vars: map[string]string{"v1": "aaa", "v2": "bbb"},
|
|
host: "aaa.google.com",
|
|
path: "/bbb",
|
|
shouldMatch: false,
|
|
},
|
|
{
|
|
route: subrouter2.Path("/baz/{v2}"),
|
|
request: newRequest("GET", "http://localhost/foo/bar/baz/ding"),
|
|
vars: map[string]string{"v1": "bar", "v2": "ding"},
|
|
host: "",
|
|
path: "/foo/bar/baz/ding",
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
route: subrouter2.Path("/baz/{v2}"),
|
|
request: newRequest("GET", "http://localhost/foo/bar"),
|
|
vars: map[string]string{"v1": "bar", "v2": "ding"},
|
|
host: "",
|
|
path: "/foo/bar/baz/ding",
|
|
shouldMatch: false,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestNamedRoutes(t *testing.T) {
|
|
r1 := NewRouter()
|
|
r1.NewRoute().Name("a")
|
|
r1.NewRoute().Name("b")
|
|
r1.NewRoute().Name("c")
|
|
|
|
r2 := r1.NewRoute().Subrouter()
|
|
r2.NewRoute().Name("d")
|
|
r2.NewRoute().Name("e")
|
|
r2.NewRoute().Name("f")
|
|
|
|
r3 := r2.NewRoute().Subrouter()
|
|
r3.NewRoute().Name("g")
|
|
r3.NewRoute().Name("h")
|
|
r3.NewRoute().Name("i")
|
|
|
|
if r1.namedRoutes == nil || len(r1.namedRoutes) != 9 {
|
|
t.Errorf("Expected 9 named routes, got %v", r1.namedRoutes)
|
|
} else if r1.Get("i") == nil {
|
|
t.Errorf("Subroute name not registered")
|
|
}
|
|
}
|
|
|
|
func TestStrictSlash(t *testing.T) {
|
|
r := NewRouter()
|
|
r.StrictSlash(true)
|
|
|
|
tests := []routeTest{
|
|
{
|
|
title: "Redirect path without slash",
|
|
route: r.NewRoute().Path("/111/"),
|
|
request: newRequest("GET", "http://localhost/111"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111/",
|
|
shouldMatch: true,
|
|
shouldRedirect: true,
|
|
},
|
|
{
|
|
title: "Do not redirect path with slash",
|
|
route: r.NewRoute().Path("/111/"),
|
|
request: newRequest("GET", "http://localhost/111/"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111/",
|
|
shouldMatch: true,
|
|
shouldRedirect: false,
|
|
},
|
|
{
|
|
title: "Redirect path with slash",
|
|
route: r.NewRoute().Path("/111"),
|
|
request: newRequest("GET", "http://localhost/111/"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111",
|
|
shouldMatch: true,
|
|
shouldRedirect: true,
|
|
},
|
|
{
|
|
title: "Do not redirect path without slash",
|
|
route: r.NewRoute().Path("/111"),
|
|
request: newRequest("GET", "http://localhost/111"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/111",
|
|
shouldMatch: true,
|
|
shouldRedirect: false,
|
|
},
|
|
{
|
|
title: "Propagate StrictSlash to subrouters",
|
|
route: r.NewRoute().PathPrefix("/static/").Subrouter().Path("/images/"),
|
|
request: newRequest("GET", "http://localhost/static/images"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/static/images/",
|
|
shouldMatch: true,
|
|
shouldRedirect: true,
|
|
},
|
|
{
|
|
title: "Ignore StrictSlash for path prefix",
|
|
route: r.NewRoute().PathPrefix("/static/"),
|
|
request: newRequest("GET", "http://localhost/static/logo.png"),
|
|
vars: map[string]string{},
|
|
host: "",
|
|
path: "/static/",
|
|
shouldMatch: true,
|
|
shouldRedirect: false,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
testRoute(t, test)
|
|
}
|
|
}
|
|
|
|
func TestWalkSingleDepth(t *testing.T) {
|
|
r0 := NewRouter()
|
|
r1 := NewRouter()
|
|
r2 := NewRouter()
|
|
|
|
r0.Path("/g")
|
|
r0.Path("/o")
|
|
r0.Path("/d").Handler(r1)
|
|
r0.Path("/r").Handler(r2)
|
|
r0.Path("/a")
|
|
|
|
r1.Path("/z")
|
|
r1.Path("/i")
|
|
r1.Path("/l")
|
|
r1.Path("/l")
|
|
|
|
r2.Path("/i")
|
|
r2.Path("/l")
|
|
r2.Path("/l")
|
|
|
|
paths := []string{"g", "o", "r", "i", "l", "l", "a"}
|
|
depths := []int{0, 0, 0, 1, 1, 1, 0}
|
|
i := 0
|
|
err := r0.Walk(func(route *Route, router *Router, ancestors []*Route) error {
|
|
matcher := route.matchers[0].(*routeRegexp)
|
|
if matcher.template == "/d" {
|
|
return SkipRouter
|
|
}
|
|
if len(ancestors) != depths[i] {
|
|
t.Errorf(`Expected depth of %d at i = %d; got "%s"`, depths[i], i, len(ancestors))
|
|
}
|
|
if matcher.template != "/"+paths[i] {
|
|
t.Errorf(`Expected "/%s" at i = %d; got "%s"`, paths[i], i, matcher.template)
|
|
}
|
|
i++
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if i != len(paths) {
|
|
t.Errorf("Expected %d routes, found %d", len(paths), i)
|
|
}
|
|
}
|
|
|
|
func TestWalkNested(t *testing.T) {
|
|
router := NewRouter()
|
|
|
|
g := router.Path("/g").Subrouter()
|
|
o := g.PathPrefix("/o").Subrouter()
|
|
r := o.PathPrefix("/r").Subrouter()
|
|
i := r.PathPrefix("/i").Subrouter()
|
|
l1 := i.PathPrefix("/l").Subrouter()
|
|
l2 := l1.PathPrefix("/l").Subrouter()
|
|
l2.Path("/a")
|
|
|
|
paths := []string{"/g", "/g/o", "/g/o/r", "/g/o/r/i", "/g/o/r/i/l", "/g/o/r/i/l/l", "/g/o/r/i/l/l/a"}
|
|
idx := 0
|
|
err := router.Walk(func(route *Route, router *Router, ancestors []*Route) error {
|
|
path := paths[idx]
|
|
tpl := route.regexp.path.template
|
|
if tpl != path {
|
|
t.Errorf(`Expected %s got %s`, path, tpl)
|
|
}
|
|
idx++
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if idx != len(paths) {
|
|
t.Errorf("Expected %d routes, found %d", len(paths), idx)
|
|
}
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
// Helpers
|
|
// ----------------------------------------------------------------------------
|
|
|
|
func getRouteTemplate(route *Route) string {
|
|
host, path := "none", "none"
|
|
if route.regexp != nil {
|
|
if route.regexp.host != nil {
|
|
host = route.regexp.host.template
|
|
}
|
|
if route.regexp.path != nil {
|
|
path = route.regexp.path.template
|
|
}
|
|
}
|
|
return fmt.Sprintf("Host: %v, Path: %v", host, path)
|
|
}
|
|
|
|
func testRoute(t *testing.T, test routeTest) {
|
|
request := test.request
|
|
route := test.route
|
|
vars := test.vars
|
|
shouldMatch := test.shouldMatch
|
|
host := test.host
|
|
path := test.path
|
|
url := test.host + test.path
|
|
shouldRedirect := test.shouldRedirect
|
|
|
|
var match RouteMatch
|
|
ok := route.Match(request, &match)
|
|
if ok != shouldMatch {
|
|
msg := "Should match"
|
|
if !shouldMatch {
|
|
msg = "Should not match"
|
|
}
|
|
t.Errorf("(%v) %v:\nRoute: %#v\nRequest: %#v\nVars: %v\n", test.title, msg, route, request, vars)
|
|
return
|
|
}
|
|
if shouldMatch {
|
|
if test.vars != nil && !stringMapEqual(test.vars, match.Vars) {
|
|
t.Errorf("(%v) Vars not equal: expected %v, got %v", test.title, vars, match.Vars)
|
|
return
|
|
}
|
|
if host != "" {
|
|
u, _ := test.route.URLHost(mapToPairs(match.Vars)...)
|
|
if host != u.Host {
|
|
t.Errorf("(%v) URLHost not equal: expected %v, got %v -- %v", test.title, host, u.Host, getRouteTemplate(route))
|
|
return
|
|
}
|
|
}
|
|
if path != "" {
|
|
u, _ := route.URLPath(mapToPairs(match.Vars)...)
|
|
if path != u.Path {
|
|
t.Errorf("(%v) URLPath not equal: expected %v, got %v -- %v", test.title, path, u.Path, getRouteTemplate(route))
|
|
return
|
|
}
|
|
}
|
|
if url != "" {
|
|
u, _ := route.URL(mapToPairs(match.Vars)...)
|
|
if url != u.Host+u.Path {
|
|
t.Errorf("(%v) URL not equal: expected %v, got %v -- %v", test.title, url, u.Host+u.Path, getRouteTemplate(route))
|
|
return
|
|
}
|
|
}
|
|
if shouldRedirect && match.Handler == nil {
|
|
t.Errorf("(%v) Did not redirect", test.title)
|
|
return
|
|
}
|
|
if !shouldRedirect && match.Handler != nil {
|
|
t.Errorf("(%v) Unexpected redirect", test.title)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// Tests that the context is cleared or not cleared properly depending on
|
|
// the configuration of the router
|
|
func TestKeepContext(t *testing.T) {
|
|
func1 := func(w http.ResponseWriter, r *http.Request) {}
|
|
|
|
r := NewRouter()
|
|
r.HandleFunc("/", func1).Name("func1")
|
|
|
|
req, _ := http.NewRequest("GET", "http://localhost/", nil)
|
|
context.Set(req, "t", 1)
|
|
|
|
res := new(http.ResponseWriter)
|
|
r.ServeHTTP(*res, req)
|
|
|
|
if _, ok := context.GetOk(req, "t"); ok {
|
|
t.Error("Context should have been cleared at end of request")
|
|
}
|
|
|
|
r.KeepContext = true
|
|
|
|
req, _ = http.NewRequest("GET", "http://localhost/", nil)
|
|
context.Set(req, "t", 1)
|
|
|
|
r.ServeHTTP(*res, req)
|
|
if _, ok := context.GetOk(req, "t"); !ok {
|
|
t.Error("Context should NOT have been cleared at end of request")
|
|
}
|
|
|
|
}
|
|
|
|
type TestA301ResponseWriter struct {
|
|
hh http.Header
|
|
status int
|
|
}
|
|
|
|
func (ho TestA301ResponseWriter) Header() http.Header {
|
|
return http.Header(ho.hh)
|
|
}
|
|
|
|
func (ho TestA301ResponseWriter) Write(b []byte) (int, error) {
|
|
return 0, nil
|
|
}
|
|
|
|
func (ho TestA301ResponseWriter) WriteHeader(code int) {
|
|
ho.status = code
|
|
}
|
|
|
|
func Test301Redirect(t *testing.T) {
|
|
m := make(http.Header)
|
|
|
|
func1 := func(w http.ResponseWriter, r *http.Request) {}
|
|
func2 := func(w http.ResponseWriter, r *http.Request) {}
|
|
|
|
r := NewRouter()
|
|
r.HandleFunc("/api/", func2).Name("func2")
|
|
r.HandleFunc("/", func1).Name("func1")
|
|
|
|
req, _ := http.NewRequest("GET", "http://localhost//api/?abc=def", nil)
|
|
|
|
res := TestA301ResponseWriter{
|
|
hh: m,
|
|
status: 0,
|
|
}
|
|
r.ServeHTTP(&res, req)
|
|
|
|
if "http://localhost/api/?abc=def" != res.hh["Location"][0] {
|
|
t.Errorf("Should have complete URL with query string")
|
|
}
|
|
}
|
|
|
|
// https://plus.google.com/101022900381697718949/posts/eWy6DjFJ6uW
|
|
func TestSubrouterHeader(t *testing.T) {
|
|
expected := "func1 response"
|
|
func1 := func(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Fprint(w, expected)
|
|
}
|
|
func2 := func(http.ResponseWriter, *http.Request) {}
|
|
|
|
r := NewRouter()
|
|
s := r.Headers("SomeSpecialHeader", "").Subrouter()
|
|
s.HandleFunc("/", func1).Name("func1")
|
|
r.HandleFunc("/", func2).Name("func2")
|
|
|
|
req, _ := http.NewRequest("GET", "http://localhost/", nil)
|
|
req.Header.Add("SomeSpecialHeader", "foo")
|
|
match := new(RouteMatch)
|
|
matched := r.Match(req, match)
|
|
if !matched {
|
|
t.Errorf("Should match request")
|
|
}
|
|
if match.Route.GetName() != "func1" {
|
|
t.Errorf("Expecting func1 handler, got %s", match.Route.GetName())
|
|
}
|
|
resp := NewRecorder()
|
|
match.Handler.ServeHTTP(resp, req)
|
|
if resp.Body.String() != expected {
|
|
t.Errorf("Expecting %q", expected)
|
|
}
|
|
}
|
|
|
|
// mapToPairs converts a string map to a slice of string pairs
|
|
func mapToPairs(m map[string]string) []string {
|
|
var i int
|
|
p := make([]string, len(m)*2)
|
|
for k, v := range m {
|
|
p[i] = k
|
|
p[i+1] = v
|
|
i += 2
|
|
}
|
|
return p
|
|
}
|
|
|
|
// stringMapEqual checks the equality of two string maps
|
|
func stringMapEqual(m1, m2 map[string]string) bool {
|
|
nil1 := m1 == nil
|
|
nil2 := m2 == nil
|
|
if nil1 != nil2 || len(m1) != len(m2) {
|
|
return false
|
|
}
|
|
for k, v := range m1 {
|
|
if v != m2[k] {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
// newRequest is a helper function to create a new request with a method and url
|
|
func newRequest(method, url string) *http.Request {
|
|
req, err := http.NewRequest(method, url, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return req
|
|
}
|