// 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 ( "bufio" "bytes" "errors" "fmt" "net/http" "net/url" "reflect" "strings" "testing" ) 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, regexpType: %v, options: %v, regexp: regexp.MustCompile(%q), reverse: %q, varsN: %v, varsR: %v", r.template, r.regexpType, r.options, 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 scheme string // the expected scheme of the built URL host string // the expected host of the built URL path string // the expected path of the built URL query string // the expected query string of the built URL pathTemplate string // the expected path template of the route hostTemplate string // the expected host template of the route queriesTemplate string // the expected query template of the route methods []string // the expected route methods pathRegexp string // the expected path regexp queriesRegexp string // the expected query regexp 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) { 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, }, { title: "Host route with port, match with request header", route: new(Route).Host("aaa.bbb.ccc:1234"), request: newRequestHost("GET", "/111/222/333", "aaa.bbb.ccc:1234"), vars: map[string]string{}, host: "aaa.bbb.ccc:1234", path: "", shouldMatch: 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 with request header", route: new(Route).Host("aaa.{v1:[a-z]{3}}.ccc:1{v2:(?:23|4)}"), request: newRequestHost("GET", "/111/222/333", "aaa.bbb.ccc:123"), vars: map[string]string{"v1": "bbb", "v2": "23"}, host: "aaa.bbb.ccc:123", path: "", hostTemplate: `aaa.{v1:[a-z]{3}}.ccc:1{v2:(?:23|4)}`, shouldMatch: true, }, { 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: "", hostTemplate: `aaa.{v1:[a-z]{3}}.ccc`, 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: "", hostTemplate: `aaa.{v1:[a-z]{2}(?:b|c)}.ccc`, 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: "", hostTemplate: `aaa.{v1:[a-z]{3}}.ccc`, 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: "", hostTemplate: `{v1:[a-z]{3}}.{v2:[a-z]{3}}.{v3:[a-z]{3}}`, 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: "", hostTemplate: `{v1:[a-z]{3}}.{v2:[a-z]{3}}.{v3:[a-z]{3}}`, 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: "", hostTemplate: `aaa.{v-1:[a-z]{3}}.ccc`, 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: "", hostTemplate: `aaa.{v-1:[a-z]{2}(?:b|c)}.ccc`, 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: "", hostTemplate: `{v-1:[a-z]{3}}.{v-2:[a-z]{3}}.{v-3:[a-z]{3}}`, shouldMatch: true, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(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", pathTemplate: `/111/`, pathRegexp: `^/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/", pathTemplate: `/111`, shouldMatch: false, }, { title: "Path route, match root with no host", route: new(Route).Path("/"), request: newRequest("GET", "/"), vars: map[string]string{}, host: "", path: "/", pathTemplate: `/`, pathRegexp: `^/$`, shouldMatch: true, }, { title: "Path route, match root with no host, App Engine format", route: new(Route).Path("/"), request: func() *http.Request { r := newRequest("GET", "http://localhost/") r.RequestURI = "/" return r }(), vars: map[string]string{}, host: "", path: "/", pathTemplate: `/`, shouldMatch: true, }, { 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", pathTemplate: `/111/{v1:[0-9]{3}}/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", pathTemplate: `/111/{v1:[0-9]{3}}/333`, pathRegexp: `^/111/(?P[0-9]{3})/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", pathTemplate: `/{v1:[0-9]{3}}/{v2:[0-9]{3}}/{v3:[0-9]{3}}`, pathRegexp: `^/(?P[0-9]{3})/(?P[0-9]{3})/(?P[0-9]{3})$`, 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", pathTemplate: `/{v1:[0-9]{3}}/{v2:[0-9]{3}}/{v3:[0-9]{3}}`, pathRegexp: `^/(?P[0-9]{3})/(?P[0-9]{3})/(?P[0-9]{3})$`, 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", pathTemplate: `/{category:a|(?:b/c)}/{product}/{id:[0-9]+}`, pathRegexp: `^/(?Pa|(?:b/c))/(?P[^/]+)/(?P[0-9]+)$`, 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", pathTemplate: `/111/{v-1:[0-9]{3}}/333`, pathRegexp: `^/111/(?P[0-9]{3})/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", pathTemplate: `/{v-1:[0-9]{3}}/{v-2:[0-9]{3}}/{v-3:[0-9]{3}}`, pathRegexp: `^/(?P[0-9]{3})/(?P[0-9]{3})/(?P[0-9]{3})$`, 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", pathTemplate: `/{product-category:a|(?:b/c)}/{product-name}/{product-id:[0-9]+}`, pathRegexp: `^/(?Pa|(?:b/c))/(?P[^/]+)/(?P[0-9]+)$`, shouldMatch: true, }, { title: "Path route with multiple hyphenated names and patterns with pipe and case insensitive, match", route: new(Route).Path("/{type:(?i:daily|mini|variety)}-{date:\\d{4,4}-\\d{2,2}-\\d{2,2}}"), request: newRequest("GET", "http://localhost/daily-2016-01-01"), vars: map[string]string{"type": "daily", "date": "2016-01-01"}, host: "", path: "/daily-2016-01-01", pathTemplate: `/{type:(?i:daily|mini|variety)}-{date:\d{4,4}-\d{2,2}-\d{2,2}}`, pathRegexp: `^/(?P(?i:daily|mini|variety))-(?P\d{4,4}-\d{2,2}-\d{2,2})$`, shouldMatch: true, }, { title: "Path route with empty match right after other match", route: new(Route).Path(`/{v1:[0-9]*}{v2:[a-z]*}/{v3:[0-9]*}`), request: newRequest("GET", "http://localhost/111/222"), vars: map[string]string{"v1": "111", "v2": "", "v3": "222"}, host: "", path: "/111/222", pathTemplate: `/{v1:[0-9]*}{v2:[a-z]*}/{v3:[0-9]*}`, pathRegexp: `^/(?P[0-9]*)(?P[a-z]*)/(?P[0-9]*)$`, 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", pathTemplate: `/{category:a|b/c}`, 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", pathTemplate: `/{category:a|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", pathTemplate: `/{category:a|b/c}/{product}/{id:[0-9]+}`, 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", pathTemplate: `/{category:a|b/c}/{product}/{id:[0-9]+}`, shouldMatch: true, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(t, test) testUseEscapedRoute(t, test) testRegexp(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", pathTemplate: `/111/{v1:[0-9]{3}}`, 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", pathTemplate: `/111/{v1:[0-9]{3}}`, 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", pathTemplate: `/{v1:[0-9]{3}}/{v2:[0-9]{3}}`, 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", pathTemplate: `/{v1:[0-9]{3}}/{v2:[0-9]{3}}`, shouldMatch: false, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(t, test) testUseEscapedRoute(t, test) }) } } func TestSchemeHostPath(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{}, scheme: "http", host: "aaa.bbb.ccc", path: "/111/222/333", pathTemplate: `/111/222/333`, hostTemplate: `aaa.bbb.ccc`, shouldMatch: true, }, { title: "Scheme, Host, and Path route, match", route: new(Route).Schemes("https").Host("aaa.bbb.ccc").Path("/111/222/333"), request: newRequest("GET", "https://aaa.bbb.ccc/111/222/333"), vars: map[string]string{}, scheme: "https", host: "aaa.bbb.ccc", path: "/111/222/333", pathTemplate: `/111/222/333`, hostTemplate: `aaa.bbb.ccc`, 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{}, scheme: "http", host: "aaa.bbb.ccc", path: "/111/222/333", pathTemplate: `/111/222/333`, hostTemplate: `aaa.bbb.ccc`, 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"}, scheme: "http", host: "aaa.bbb.ccc", path: "/111/222/333", pathTemplate: `/111/{v2:[0-9]{3}}/333`, hostTemplate: `aaa.{v1:[a-z]{3}}.ccc`, shouldMatch: true, }, { title: "Scheme, Host, and Path route with host and path patterns, match", route: new(Route).Schemes("ftp", "ssss").Host("aaa.{v1:[a-z]{3}}.ccc").Path("/111/{v2:[0-9]{3}}/333"), request: newRequest("GET", "ssss://aaa.bbb.ccc/111/222/333"), vars: map[string]string{"v1": "bbb", "v2": "222"}, scheme: "ftp", host: "aaa.bbb.ccc", path: "/111/222/333", pathTemplate: `/111/{v2:[0-9]{3}}/333`, hostTemplate: `aaa.{v1:[a-z]{3}}.ccc`, 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"}, scheme: "http", host: "aaa.bbb.ccc", path: "/111/222/333", pathTemplate: `/111/{v2:[0-9]{3}}/333`, hostTemplate: `aaa.{v1:[a-z]{3}}.ccc`, 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"}, scheme: "http", host: "aaa.bbb.ccc", path: "/111/222/333", pathTemplate: `/{v4:[0-9]{3}}/{v5:[0-9]{3}}/{v6:[0-9]{3}}`, hostTemplate: `{v1:[a-z]{3}}.{v2:[a-z]{3}}.{v3:[a-z]{3}}`, 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"}, scheme: "http", host: "aaa.bbb.ccc", path: "/111/222/333", pathTemplate: `/{v4:[0-9]{3}}/{v5:[0-9]{3}}/{v6:[0-9]{3}}`, hostTemplate: `{v1:[a-z]{3}}.{v2:[a-z]{3}}.{v3:[a-z]{3}}`, shouldMatch: false, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(t, test) testUseEscapedRoute(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 { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(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: "", methods: []string{"GET", "POST"}, shouldMatch: true, }, { title: "Methods route, match POST", route: new(Route).Methods("GET", "POST"), request: newRequest("POST", "http://localhost"), vars: map[string]string{}, host: "", path: "", methods: []string{"GET", "POST"}, shouldMatch: true, }, { title: "Methods route, bad method", route: new(Route).Methods("GET", "POST"), request: newRequest("PUT", "http://localhost"), vars: map[string]string{}, host: "", path: "", methods: []string{"GET", "POST"}, shouldMatch: false, }, { title: "Route without methods", route: new(Route), request: newRequest("PUT", "http://localhost"), vars: map[string]string{}, host: "", path: "", methods: []string{}, shouldMatch: true, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(t, test) testMethods(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: "", query: "foo=bar&baz=ding", queriesTemplate: "foo=bar,baz=ding", queriesRegexp: "^foo=bar$,^baz=ding$", 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: "", query: "foo=bar&baz=ding", pathTemplate: `/api`, hostTemplate: `www.example.com`, queriesTemplate: "foo=bar,baz=ding", queriesRegexp: "^foo=bar$,^baz=ding$", 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: "", query: "foo=bar&baz=ding", pathTemplate: `/api`, hostTemplate: `www.example.com`, queriesTemplate: "foo=bar,baz=ding", queriesRegexp: "^foo=bar$,^baz=ding$", 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: "", queriesTemplate: "foo=bar,baz=ding", queriesRegexp: "^foo=bar$,^baz=ding$", 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: "", query: "foo=bar", queriesTemplate: "foo={v1}", queriesRegexp: "^foo=(?P.*)$", 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: "", query: "foo=bar&baz=ding", queriesTemplate: "foo={v1},baz={v2}", queriesRegexp: "^foo=(?P.*)$,^baz=(?P.*)$", 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: "", query: "foo=10", queriesTemplate: "foo={v1:[0-9]+}", queriesRegexp: "^foo=(?P[0-9]+)$", 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: "", queriesTemplate: "foo={v1:[0-9]+}", queriesRegexp: "^foo=(?P[0-9]+)$", 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: "", query: "foo=1", queriesTemplate: "foo={v1:[0-9]{1}}", queriesRegexp: "^foo=(?P[0-9]{1})$", 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: "", query: "foo=1", queriesTemplate: "foo={v1:[0-9]{1}}", queriesRegexp: "^foo=(?P[0-9]{1})$", 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: "", queriesTemplate: "foo={v1:[0-9]{1}}", queriesRegexp: "^foo=(?P[0-9]{1})$", 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: "", query: "foo=1a", queriesTemplate: "foo={v1:[0-9]{1}(?:a|b)}", queriesRegexp: "^foo=(?P[0-9]{1}(?:a|b))$", 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: "", queriesTemplate: "foo={v1:[0-9]{1}}", queriesRegexp: "^foo=(?P[0-9]{1})$", 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: "", query: "foo=bar", queriesTemplate: "foo={v-1}", queriesRegexp: "^foo=(?P.*)$", 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: "", query: "foo=bar&baz=ding", queriesTemplate: "foo={v-1},baz={v-2}", queriesRegexp: "^foo=(?P.*)$,^baz=(?P.*)$", 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: "", query: "foo=10", queriesTemplate: "foo={v-1:[0-9]+}", queriesRegexp: "^foo=(?P[0-9]+)$", 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: "", query: "foo=1a", queriesTemplate: "foo={v-1:[0-9]{1}(?:a|b)}", queriesRegexp: "^foo=(?P[0-9]{1}(?:a|b))$", 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: "", query: "foo=", queriesTemplate: "foo=", queriesRegexp: "^foo=.*$", 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: "", queriesTemplate: "foo=", queriesRegexp: "^foo=.*$", 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: "", query: "foo=", queriesTemplate: "foo=", queriesRegexp: "^foo=.*$", 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: "", queriesTemplate: "foo=bar", queriesRegexp: "^foo=bar$", 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: "", queriesTemplate: "foo={bar}", queriesRegexp: "^foo=(?P.*)$", 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: "", query: "foo=", queriesTemplate: "foo={bar}", queriesRegexp: "^foo=(?P.*)$", 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: "", queriesTemplate: "foo=bar,baz=ding", queriesRegexp: "^foo=bar$,^baz=ding$", shouldMatch: false, }, { title: "Queries route with pattern, match, escaped value", route: new(Route).Queries("foo", "{v1}"), request: newRequest("GET", "http://localhost?foo=%25bar%26%20%2F%3D%3F"), vars: map[string]string{"v1": "%bar& /=?"}, host: "", path: "", query: "foo=%25bar%26+%2F%3D%3F", queriesTemplate: "foo={v1}", queriesRegexp: "^foo=(?P.*)$", shouldMatch: true, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testTemplate(t, test) testQueriesTemplates(t, test) testUseEscapedRoute(t, test) testQueriesRegexp(t, test) }) } } func TestSchemes(t *testing.T) { tests := []routeTest{ // Schemes { title: "Schemes route, default scheme, match http, build http", route: new(Route).Host("localhost"), request: newRequest("GET", "http://localhost"), scheme: "http", host: "localhost", shouldMatch: true, }, { title: "Schemes route, match https, build https", route: new(Route).Schemes("https", "ftp").Host("localhost"), request: newRequest("GET", "https://localhost"), scheme: "https", host: "localhost", shouldMatch: true, }, { title: "Schemes route, match ftp, build https", route: new(Route).Schemes("https", "ftp").Host("localhost"), request: newRequest("GET", "ftp://localhost"), scheme: "https", host: "localhost", shouldMatch: true, }, { title: "Schemes route, match ftp, build ftp", route: new(Route).Schemes("ftp", "https").Host("localhost"), request: newRequest("GET", "ftp://localhost"), scheme: "ftp", host: "localhost", shouldMatch: true, }, { title: "Schemes route, bad scheme", route: new(Route).Schemes("https", "ftp").Host("localhost"), request: newRequest("GET", "http://localhost"), scheme: "https", host: "localhost", shouldMatch: false, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(t, test) }) } } func TestMatcherFunc(t *testing.T) { m := func(r *http.Request, m *RouteMatch) bool { return r.URL.Host == "aaa.bbb.ccc" } 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 { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(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", pathTemplate: `/111/{v1:\d}{v2:.*}`, 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", pathTemplate: `/{v1:\d}/{v2:\w}`, shouldMatch: true, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(t, test) }) } } func TestSubRouter(t *testing.T) { subrouter1 := new(Route).Host("{v1:[a-z]+}.google.com").Subrouter() subrouter2 := new(Route).PathPrefix("/foo/{v1}").Subrouter() subrouter3 := new(Route).PathPrefix("/foo").Subrouter() subrouter4 := new(Route).PathPrefix("/foo/bar").Subrouter() subrouter5 := new(Route).PathPrefix("/{category}").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", pathTemplate: `/{v2:[a-z]+}`, hostTemplate: `{v1:[a-z]+}.google.com`, 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", pathTemplate: `/{v2:[a-z]+}`, hostTemplate: `{v1:[a-z]+}.google.com`, 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", pathTemplate: `/foo/{v1}/baz/{v2}`, 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", pathTemplate: `/foo/{v1}/baz/{v2}`, shouldMatch: false, }, { route: subrouter3.Path("/"), request: newRequest("GET", "http://localhost/foo/"), vars: map[string]string{}, host: "", path: "/foo/", pathTemplate: `/foo/`, shouldMatch: true, }, { route: subrouter3.Path(""), request: newRequest("GET", "http://localhost/foo"), vars: map[string]string{}, host: "", path: "/foo", pathTemplate: `/foo`, shouldMatch: true, }, { route: subrouter4.Path("/"), request: newRequest("GET", "http://localhost/foo/bar/"), vars: map[string]string{}, host: "", path: "/foo/bar/", pathTemplate: `/foo/bar/`, shouldMatch: true, }, { route: subrouter4.Path(""), request: newRequest("GET", "http://localhost/foo/bar"), vars: map[string]string{}, host: "", path: "/foo/bar", pathTemplate: `/foo/bar`, shouldMatch: true, }, { route: subrouter5.Path("/"), request: newRequest("GET", "http://localhost/baz/"), vars: map[string]string{"category": "baz"}, host: "", path: "/baz/", pathTemplate: `/{category}/`, shouldMatch: true, }, { route: subrouter5.Path(""), request: newRequest("GET", "http://localhost/baz"), vars: map[string]string{"category": "baz"}, host: "", path: "/baz", pathTemplate: `/{category}`, shouldMatch: true, }, { title: "Mismatch method specified on parent route", route: new(Route).Methods("POST").PathPrefix("/foo").Subrouter().Path("/"), request: newRequest("GET", "http://localhost/foo/"), vars: map[string]string{}, host: "", path: "/foo/", pathTemplate: `/foo/`, shouldMatch: false, }, { title: "Match method specified on parent route", route: new(Route).Methods("POST").PathPrefix("/foo").Subrouter().Path("/"), request: newRequest("POST", "http://localhost/foo/"), vars: map[string]string{}, host: "", path: "/foo/", pathTemplate: `/foo/`, shouldMatch: true, }, { title: "Mismatch scheme specified on parent route", route: new(Route).Schemes("https").Subrouter().PathPrefix("/"), request: newRequest("GET", "http://localhost/"), vars: map[string]string{}, host: "", path: "/", pathTemplate: `/`, shouldMatch: false, }, { title: "Match scheme specified on parent route", route: new(Route).Schemes("http").Subrouter().PathPrefix("/"), request: newRequest("GET", "http://localhost/"), vars: map[string]string{}, host: "", path: "/", pathTemplate: `/`, shouldMatch: true, }, { title: "No match header specified on parent route", route: new(Route).Headers("X-Forwarded-Proto", "https").Subrouter().PathPrefix("/"), request: newRequest("GET", "http://localhost/"), vars: map[string]string{}, host: "", path: "/", pathTemplate: `/`, shouldMatch: false, }, { title: "Header mismatch value specified on parent route", route: new(Route).Headers("X-Forwarded-Proto", "https").Subrouter().PathPrefix("/"), request: newRequestWithHeaders("GET", "http://localhost/", "X-Forwarded-Proto", "http"), vars: map[string]string{}, host: "", path: "/", pathTemplate: `/`, shouldMatch: false, }, { title: "Header match value specified on parent route", route: new(Route).Headers("X-Forwarded-Proto", "https").Subrouter().PathPrefix("/"), request: newRequestWithHeaders("GET", "http://localhost/", "X-Forwarded-Proto", "https"), vars: map[string]string{}, host: "", path: "/", pathTemplate: `/`, shouldMatch: true, }, { title: "Query specified on parent route not present", route: new(Route).Headers("key", "foobar").Subrouter().PathPrefix("/"), request: newRequest("GET", "http://localhost/"), vars: map[string]string{}, host: "", path: "/", pathTemplate: `/`, shouldMatch: false, }, { title: "Query mismatch value specified on parent route", route: new(Route).Queries("key", "foobar").Subrouter().PathPrefix("/"), request: newRequest("GET", "http://localhost/?key=notfoobar"), vars: map[string]string{}, host: "", path: "/", pathTemplate: `/`, shouldMatch: false, }, { title: "Query match value specified on subroute", route: new(Route).Queries("key", "foobar").Subrouter().PathPrefix("/"), request: newRequest("GET", "http://localhost/?key=foobar"), vars: map[string]string{}, host: "", path: "/", pathTemplate: `/`, shouldMatch: true, }, { title: "Build with scheme on parent router", route: new(Route).Schemes("ftp").Host("google.com").Subrouter().Path("/"), request: newRequest("GET", "ftp://google.com/"), scheme: "ftp", host: "google.com", path: "/", pathTemplate: `/`, hostTemplate: `google.com`, shouldMatch: true, }, { title: "Prefer scheme on child route when building URLs", route: new(Route).Schemes("https", "ftp").Host("google.com").Subrouter().Schemes("ftp").Path("/"), request: newRequest("GET", "ftp://google.com/"), scheme: "ftp", host: "google.com", path: "/", pathTemplate: `/`, hostTemplate: `google.com`, shouldMatch: true, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(t, test) testUseEscapedRoute(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") r3.Name("j") if r1.namedRoutes == nil || len(r1.namedRoutes) != 10 { t.Errorf("Expected 10 named routes, got %v", r1.namedRoutes) } else if r1.Get("j") == nil { t.Errorf("Subroute name not registered") } } func TestNameMultipleCalls(t *testing.T) { r1 := NewRouter() rt := r1.NewRoute().Name("foo").Name("bar") err := rt.GetError() if err == nil { t.Errorf("Expected an error") } } 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 { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(t, test) testUseEscapedRoute(t, test) }) } } func TestUseEncodedPath(t *testing.T) { r := NewRouter() r.UseEncodedPath() tests := []routeTest{ { title: "Router with useEncodedPath, URL with encoded slash does match", route: r.NewRoute().Path("/v1/{v1}/v2"), request: newRequest("GET", "http://localhost/v1/1%2F2/v2"), vars: map[string]string{"v1": "1%2F2"}, host: "", path: "/v1/1%2F2/v2", pathTemplate: `/v1/{v1}/v2`, shouldMatch: true, }, { title: "Router with useEncodedPath, URL with encoded slash doesn't match", route: r.NewRoute().Path("/v1/1/2/v2"), request: newRequest("GET", "http://localhost/v1/1%2F2/v2"), vars: map[string]string{"v1": "1%2F2"}, host: "", path: "/v1/1%2F2/v2", pathTemplate: `/v1/1/2/v2`, shouldMatch: false, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testRoute(t, test) testTemplate(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 "%d"`, 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() routeSubrouter := func(r *Route) (*Route, *Router) { return r, r.Subrouter() } gRoute, g := routeSubrouter(router.Path("/g")) oRoute, o := routeSubrouter(g.PathPrefix("/o")) rRoute, r := routeSubrouter(o.PathPrefix("/r")) iRoute, i := routeSubrouter(r.PathPrefix("/i")) l1Route, l1 := routeSubrouter(i.PathPrefix("/l")) l2Route, l2 := routeSubrouter(l1.PathPrefix("/l")) l2.Path("/a") testCases := []struct { path string ancestors []*Route }{ {"/g", []*Route{}}, {"/g/o", []*Route{gRoute}}, {"/g/o/r", []*Route{gRoute, oRoute}}, {"/g/o/r/i", []*Route{gRoute, oRoute, rRoute}}, {"/g/o/r/i/l", []*Route{gRoute, oRoute, rRoute, iRoute}}, {"/g/o/r/i/l/l", []*Route{gRoute, oRoute, rRoute, iRoute, l1Route}}, {"/g/o/r/i/l/l/a", []*Route{gRoute, oRoute, rRoute, iRoute, l1Route, l2Route}}, } idx := 0 err := router.Walk(func(route *Route, router *Router, ancestors []*Route) error { path := testCases[idx].path tpl := route.regexp.path.template if tpl != path { t.Errorf(`Expected %s got %s`, path, tpl) } currWantAncestors := testCases[idx].ancestors if !reflect.DeepEqual(currWantAncestors, ancestors) { t.Errorf(`Expected %+v got %+v`, currWantAncestors, ancestors) } idx++ return nil }) if err != nil { panic(err) } if idx != len(testCases) { t.Errorf("Expected %d routes, found %d", len(testCases), idx) } } func TestWalkSubrouters(t *testing.T) { router := NewRouter() g := router.Path("/g").Subrouter() o := g.PathPrefix("/o").Subrouter() o.Methods("GET") o.Methods("PUT") // all 4 routes should be matched paths := []string{"/g", "/g/o", "/g/o", "/g/o"} idx := 0 err := router.Walk(func(route *Route, router *Router, ancestors []*Route) error { path := paths[idx] tpl, _ := route.GetPathTemplate() 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) } } func TestWalkErrorRoute(t *testing.T) { router := NewRouter() router.Path("/g") expectedError := errors.New("error") err := router.Walk(func(route *Route, router *Router, ancestors []*Route) error { return expectedError }) if err != expectedError { t.Errorf("Expected %v routes, found %v", expectedError, err) } } func TestWalkErrorMatcher(t *testing.T) { router := NewRouter() expectedError := router.Path("/g").Subrouter().Path("").GetError() err := router.Walk(func(route *Route, router *Router, ancestors []*Route) error { return route.GetError() }) if err != expectedError { t.Errorf("Expected %v routes, found %v", expectedError, err) } } func TestWalkErrorHandler(t *testing.T) { handler := NewRouter() expectedError := handler.Path("/path").Subrouter().Path("").GetError() router := NewRouter() router.Path("/g").Handler(handler) err := router.Walk(func(route *Route, router *Router, ancestors []*Route) error { return route.GetError() }) if err != expectedError { t.Errorf("Expected %v routes, found %v", expectedError, err) } } func TestSubrouterErrorHandling(t *testing.T) { superRouterCalled := false subRouterCalled := false router := NewRouter() router.NotFoundHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { superRouterCalled = true }) subRouter := router.PathPrefix("/bign8").Subrouter() subRouter.NotFoundHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { subRouterCalled = true }) req, _ := http.NewRequest("GET", "http://localhost/bign8/was/here", nil) router.ServeHTTP(NewRecorder(), req) if superRouterCalled { t.Error("Super router 404 handler called when sub-router 404 handler is available.") } if !subRouterCalled { t.Error("Sub-router 404 handler was not called.") } } // See: https://github.com/gorilla/mux/issues/200 func TestPanicOnCapturingGroups(t *testing.T) { defer func() { if recover() == nil { t.Errorf("(Test that capturing groups now fail fast) Expected panic, however test completed successfully.\n") } }() NewRouter().NewRoute().Path("/{type:(promo|special)}/{promoId}.json") } // ---------------------------------------------------------------------------- // Helpers // ---------------------------------------------------------------------------- func getRouteTemplate(route *Route) string { host, err := route.GetHostTemplate() if err != nil { host = "none" } path, err := route.GetPathTemplate() if err != nil { path = "none" } 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 query := test.query shouldRedirect := test.shouldRedirect uri := url.URL{ Scheme: test.scheme, Host: test.host, Path: test.path, } if uri.Scheme == "" { uri.Scheme = "http" } 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 vars != nil && !stringMapEqual(vars, match.Vars) { t.Errorf("(%v) Vars not equal: expected %v, got %v", test.title, vars, match.Vars) return } if test.scheme != "" { u, err := route.URL(mapToPairs(match.Vars)...) if err != nil { t.Fatalf("(%v) URL error: %v -- %v", test.title, err, getRouteTemplate(route)) } if uri.Scheme != u.Scheme { t.Errorf("(%v) URLScheme not equal: expected %v, got %v", test.title, uri.Scheme, u.Scheme) return } } if test.host != "" { u, err := test.route.URLHost(mapToPairs(match.Vars)...) if err != nil { t.Fatalf("(%v) URLHost error: %v -- %v", test.title, err, getRouteTemplate(route)) } if uri.Scheme != u.Scheme { t.Errorf("(%v) URLHost scheme not equal: expected %v, got %v -- %v", test.title, uri.Scheme, u.Scheme, getRouteTemplate(route)) return } if uri.Host != u.Host { t.Errorf("(%v) URLHost host not equal: expected %v, got %v -- %v", test.title, uri.Host, u.Host, getRouteTemplate(route)) return } } if test.path != "" { u, err := route.URLPath(mapToPairs(match.Vars)...) if err != nil { t.Fatalf("(%v) URLPath error: %v -- %v", test.title, err, getRouteTemplate(route)) } if uri.Path != u.Path { t.Errorf("(%v) URLPath not equal: expected %v, got %v -- %v", test.title, uri.Path, u.Path, getRouteTemplate(route)) return } } if test.host != "" && test.path != "" { u, err := route.URL(mapToPairs(match.Vars)...) if err != nil { t.Fatalf("(%v) URL error: %v -- %v", test.title, err, getRouteTemplate(route)) } if expected, got := uri.String(), u.String(); expected != got { t.Errorf("(%v) URL not equal: expected %v, got %v -- %v", test.title, expected, got, getRouteTemplate(route)) return } } if query != "" { u, err := route.URL(mapToPairs(match.Vars)...) if err != nil { t.Errorf("(%v) erred while creating url: %v", test.title, err) return } if query != u.RawQuery { t.Errorf("(%v) URL query not equal: expected %v, got %v", test.title, query, u.RawQuery) 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 } } } func testUseEscapedRoute(t *testing.T, test routeTest) { test.route.useEncodedPath = true testRoute(t, test) } func testTemplate(t *testing.T, test routeTest) { route := test.route pathTemplate := test.pathTemplate if len(pathTemplate) == 0 { pathTemplate = test.path } hostTemplate := test.hostTemplate if len(hostTemplate) == 0 { hostTemplate = test.host } routePathTemplate, pathErr := route.GetPathTemplate() if pathErr == nil && routePathTemplate != pathTemplate { t.Errorf("(%v) GetPathTemplate not equal: expected %v, got %v", test.title, pathTemplate, routePathTemplate) } routeHostTemplate, hostErr := route.GetHostTemplate() if hostErr == nil && routeHostTemplate != hostTemplate { t.Errorf("(%v) GetHostTemplate not equal: expected %v, got %v", test.title, hostTemplate, routeHostTemplate) } } func testMethods(t *testing.T, test routeTest) { route := test.route methods, _ := route.GetMethods() if strings.Join(methods, ",") != strings.Join(test.methods, ",") { t.Errorf("(%v) GetMethods not equal: expected %v, got %v", test.title, test.methods, methods) } } func testRegexp(t *testing.T, test routeTest) { route := test.route routePathRegexp, regexpErr := route.GetPathRegexp() if test.pathRegexp != "" && regexpErr == nil && routePathRegexp != test.pathRegexp { t.Errorf("(%v) GetPathRegexp not equal: expected %v, got %v", test.title, test.pathRegexp, routePathRegexp) } } func testQueriesRegexp(t *testing.T, test routeTest) { route := test.route queries, queriesErr := route.GetQueriesRegexp() gotQueries := strings.Join(queries, ",") if test.queriesRegexp != "" && queriesErr == nil && gotQueries != test.queriesRegexp { t.Errorf("(%v) GetQueriesRegexp not equal: expected %v, got %v", test.title, test.queriesRegexp, gotQueries) } } func testQueriesTemplates(t *testing.T, test routeTest) { route := test.route queries, queriesErr := route.GetQueriesTemplates() gotQueries := strings.Join(queries, ",") if test.queriesTemplate != "" && queriesErr == nil && gotQueries != test.queriesTemplate { t.Errorf("(%v) GetQueriesTemplates not equal: expected %v, got %v", test.title, test.queriesTemplate, gotQueries) } } 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") } } func TestSkipClean(t *testing.T) { func1 := func(w http.ResponseWriter, r *http.Request) {} func2 := func(w http.ResponseWriter, r *http.Request) {} r := NewRouter() r.SkipClean(true) r.HandleFunc("/api/", func2).Name("func2") r.HandleFunc("/", func1).Name("func1") req, _ := http.NewRequest("GET", "http://localhost//api/?abc=def", nil) res := NewRecorder() r.ServeHTTP(res, req) if len(res.HeaderMap["Location"]) != 0 { t.Errorf("Shouldn't redirect since skip clean is disabled") } } // 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) } } func TestNoMatchMethodErrorHandler(t *testing.T) { func1 := func(w http.ResponseWriter, r *http.Request) {} r := NewRouter() r.HandleFunc("/", func1).Methods("GET", "POST") req, _ := http.NewRequest("PUT", "http://localhost/", nil) match := new(RouteMatch) matched := r.Match(req, match) if matched { t.Error("Should not have matched route for methods") } if match.MatchErr != ErrMethodMismatch { t.Error("Should get ErrMethodMismatch error") } resp := NewRecorder() r.ServeHTTP(resp, req) if resp.Code != 405 { t.Errorf("Expecting code %v", 405) } // Add matching route r.HandleFunc("/", func1).Methods("PUT") match = new(RouteMatch) matched = r.Match(req, match) if !matched { t.Error("Should have matched route for methods") } if match.MatchErr != nil { t.Error("Should not have any matching error. Found:", match.MatchErr) } } func TestErrMatchNotFound(t *testing.T) { emptyHandler := func(w http.ResponseWriter, r *http.Request) {} r := NewRouter() r.HandleFunc("/", emptyHandler) s := r.PathPrefix("/sub/").Subrouter() s.HandleFunc("/", emptyHandler) // Regular 404 not found req, _ := http.NewRequest("GET", "/sub/whatever", nil) match := new(RouteMatch) matched := r.Match(req, match) if matched { t.Errorf("Subrouter should not have matched that, got %v", match.Route) } // Even without a custom handler, MatchErr is set to ErrNotFound if match.MatchErr != ErrNotFound { t.Errorf("Expected ErrNotFound MatchErr, but was %v", match.MatchErr) } // Now lets add a 404 handler to subrouter s.NotFoundHandler = http.NotFoundHandler() req, _ = http.NewRequest("GET", "/sub/whatever", nil) // Test the subrouter first match = new(RouteMatch) matched = s.Match(req, match) // Now we should get a match if !matched { t.Errorf("Subrouter should have matched %s", req.RequestURI) } // But MatchErr should be set to ErrNotFound anyway if match.MatchErr != ErrNotFound { t.Errorf("Expected ErrNotFound MatchErr, but was %v", match.MatchErr) } // Now test the parent (MatchErr should propagate) match = new(RouteMatch) matched = r.Match(req, match) // Now we should get a match if !matched { t.Errorf("Router should have matched %s via subrouter", req.RequestURI) } // But MatchErr should be set to ErrNotFound anyway if match.MatchErr != ErrNotFound { t.Errorf("Expected ErrNotFound MatchErr, but was %v", match.MatchErr) } } // methodsSubrouterTest models the data necessary for testing handler // matching for subrouters created after HTTP methods matcher registration. type methodsSubrouterTest struct { title string wantCode int router *Router // method is the input into the request and expected response method string // input request path path string // redirectTo is the expected location path for strict-slash matches redirectTo string } // methodHandler writes the method string in response. func methodHandler(method string) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { w.Write([]byte(method)) } } // TestMethodsSubrouterCatchall matches handlers for subrouters where a // catchall handler is set for a mis-matching method. func TestMethodsSubrouterCatchall(t *testing.T) { t.Parallel() router := NewRouter() router.Methods("PATCH").Subrouter().PathPrefix("/").HandlerFunc(methodHandler("PUT")) router.Methods("GET").Subrouter().HandleFunc("/foo", methodHandler("GET")) router.Methods("POST").Subrouter().HandleFunc("/foo", methodHandler("POST")) router.Methods("DELETE").Subrouter().HandleFunc("/foo", methodHandler("DELETE")) tests := []methodsSubrouterTest{ { title: "match GET handler", router: router, path: "http://localhost/foo", method: "GET", wantCode: http.StatusOK, }, { title: "match POST handler", router: router, method: "POST", path: "http://localhost/foo", wantCode: http.StatusOK, }, { title: "match DELETE handler", router: router, method: "DELETE", path: "http://localhost/foo", wantCode: http.StatusOK, }, { title: "disallow PUT method", router: router, method: "PUT", path: "http://localhost/foo", wantCode: http.StatusMethodNotAllowed, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testMethodsSubrouter(t, test) }) } } // TestMethodsSubrouterStrictSlash matches handlers on subrouters with // strict-slash matchers. func TestMethodsSubrouterStrictSlash(t *testing.T) { t.Parallel() router := NewRouter() sub := router.PathPrefix("/").Subrouter() sub.StrictSlash(true).Path("/foo").Methods("GET").Subrouter().HandleFunc("", methodHandler("GET")) sub.StrictSlash(true).Path("/foo/").Methods("PUT").Subrouter().HandleFunc("/", methodHandler("PUT")) sub.StrictSlash(true).Path("/foo/").Methods("POST").Subrouter().HandleFunc("/", methodHandler("POST")) tests := []methodsSubrouterTest{ { title: "match POST handler", router: router, method: "POST", path: "http://localhost/foo/", wantCode: http.StatusOK, }, { title: "match GET handler", router: router, method: "GET", path: "http://localhost/foo", wantCode: http.StatusOK, }, { title: "match POST handler, redirect strict-slash", router: router, method: "POST", path: "http://localhost/foo", redirectTo: "http://localhost/foo/", wantCode: http.StatusMovedPermanently, }, { title: "match GET handler, redirect strict-slash", router: router, method: "GET", path: "http://localhost/foo/", redirectTo: "http://localhost/foo", wantCode: http.StatusMovedPermanently, }, { title: "disallow DELETE method", router: router, method: "DELETE", path: "http://localhost/foo", wantCode: http.StatusMethodNotAllowed, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testMethodsSubrouter(t, test) }) } } // TestMethodsSubrouterPathPrefix matches handlers on subrouters created // on a router with a path prefix matcher and method matcher. func TestMethodsSubrouterPathPrefix(t *testing.T) { t.Parallel() router := NewRouter() router.PathPrefix("/1").Methods("POST").Subrouter().HandleFunc("/2", methodHandler("POST")) router.PathPrefix("/1").Methods("DELETE").Subrouter().HandleFunc("/2", methodHandler("DELETE")) router.PathPrefix("/1").Methods("PUT").Subrouter().HandleFunc("/2", methodHandler("PUT")) router.PathPrefix("/1").Methods("POST").Subrouter().HandleFunc("/2", methodHandler("POST2")) tests := []methodsSubrouterTest{ { title: "match first POST handler", router: router, method: "POST", path: "http://localhost/1/2", wantCode: http.StatusOK, }, { title: "match DELETE handler", router: router, method: "DELETE", path: "http://localhost/1/2", wantCode: http.StatusOK, }, { title: "match PUT handler", router: router, method: "PUT", path: "http://localhost/1/2", wantCode: http.StatusOK, }, { title: "disallow PATCH method", router: router, method: "PATCH", path: "http://localhost/1/2", wantCode: http.StatusMethodNotAllowed, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testMethodsSubrouter(t, test) }) } } // TestMethodsSubrouterSubrouter matches handlers on subrouters produced // from method matchers registered on a root subrouter. func TestMethodsSubrouterSubrouter(t *testing.T) { t.Parallel() router := NewRouter() sub := router.PathPrefix("/1").Subrouter() sub.Methods("POST").Subrouter().HandleFunc("/2", methodHandler("POST")) sub.Methods("GET").Subrouter().HandleFunc("/2", methodHandler("GET")) sub.Methods("PATCH").Subrouter().HandleFunc("/2", methodHandler("PATCH")) sub.HandleFunc("/2", methodHandler("PUT")).Subrouter().Methods("PUT") sub.HandleFunc("/2", methodHandler("POST2")).Subrouter().Methods("POST") tests := []methodsSubrouterTest{ { title: "match first POST handler", router: router, method: "POST", path: "http://localhost/1/2", wantCode: http.StatusOK, }, { title: "match GET handler", router: router, method: "GET", path: "http://localhost/1/2", wantCode: http.StatusOK, }, { title: "match PATCH handler", router: router, method: "PATCH", path: "http://localhost/1/2", wantCode: http.StatusOK, }, { title: "match PUT handler", router: router, method: "PUT", path: "http://localhost/1/2", wantCode: http.StatusOK, }, { title: "disallow DELETE method", router: router, method: "DELETE", path: "http://localhost/1/2", wantCode: http.StatusMethodNotAllowed, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testMethodsSubrouter(t, test) }) } } // TestMethodsSubrouterPathVariable matches handlers on matching paths // with path variables in them. func TestMethodsSubrouterPathVariable(t *testing.T) { t.Parallel() router := NewRouter() router.Methods("GET").Subrouter().HandleFunc("/foo", methodHandler("GET")) router.Methods("POST").Subrouter().HandleFunc("/{any}", methodHandler("POST")) router.Methods("DELETE").Subrouter().HandleFunc("/1/{any}", methodHandler("DELETE")) router.Methods("PUT").Subrouter().HandleFunc("/1/{any}", methodHandler("PUT")) tests := []methodsSubrouterTest{ { title: "match GET handler", router: router, method: "GET", path: "http://localhost/foo", wantCode: http.StatusOK, }, { title: "match POST handler", router: router, method: "POST", path: "http://localhost/foo", wantCode: http.StatusOK, }, { title: "match DELETE handler", router: router, method: "DELETE", path: "http://localhost/1/foo", wantCode: http.StatusOK, }, { title: "match PUT handler", router: router, method: "PUT", path: "http://localhost/1/foo", wantCode: http.StatusOK, }, { title: "disallow PATCH method", router: router, method: "PATCH", path: "http://localhost/1/foo", wantCode: http.StatusMethodNotAllowed, }, } for _, test := range tests { t.Run(test.title, func(t *testing.T) { testMethodsSubrouter(t, test) }) } } func ExampleSetURLVars() { req, _ := http.NewRequest("GET", "/foo", nil) req = SetURLVars(req, map[string]string{"foo": "bar"}) fmt.Println(Vars(req)["foo"]) // Output: bar } // testMethodsSubrouter runs an individual methodsSubrouterTest. func testMethodsSubrouter(t *testing.T, test methodsSubrouterTest) { // Execute request req, _ := http.NewRequest(test.method, test.path, nil) resp := NewRecorder() test.router.ServeHTTP(resp, req) switch test.wantCode { case http.StatusMethodNotAllowed: if resp.Code != http.StatusMethodNotAllowed { t.Errorf(`(%s) Expected "405 Method Not Allowed", but got %d code`, test.title, resp.Code) } else if matchedMethod := resp.Body.String(); matchedMethod != "" { t.Errorf(`(%s) Expected "405 Method Not Allowed", but %q handler was called`, test.title, matchedMethod) } case http.StatusMovedPermanently: if gotLocation := resp.HeaderMap.Get("Location"); gotLocation != test.redirectTo { t.Errorf("(%s) Expected %q route-match to redirect to %q, but got %q", test.title, test.method, test.redirectTo, gotLocation) } case http.StatusOK: if matchedMethod := resp.Body.String(); matchedMethod != test.method { t.Errorf("(%s) Expected %q handler to be called, but %q handler was called", test.title, test.method, matchedMethod) } default: expectedCodes := []int{http.StatusMethodNotAllowed, http.StatusMovedPermanently, http.StatusOK} t.Errorf("(%s) Expected wantCode to be one of: %v, but got %d", test.title, expectedCodes, test.wantCode) } } func TestSubrouterMatching(t *testing.T) { const ( none, stdOnly, subOnly uint8 = 0, 1 << 0, 1 << 1 both = subOnly | stdOnly ) type request struct { Name string Request *http.Request Flags uint8 } cases := []struct { Name string Standard, Subrouter func(*Router) Requests []request }{ { "pathPrefix", func(r *Router) { r.PathPrefix("/before").PathPrefix("/after") }, func(r *Router) { r.PathPrefix("/before").Subrouter().PathPrefix("/after") }, []request{ {"no match final path prefix", newRequest("GET", "/after"), none}, {"no match parent path prefix", newRequest("GET", "/before"), none}, {"matches append", newRequest("GET", "/before/after"), both}, {"matches as prefix", newRequest("GET", "/before/after/1234"), both}, }, }, { "path", func(r *Router) { r.Path("/before").Path("/after") }, func(r *Router) { r.Path("/before").Subrouter().Path("/after") }, []request{ {"no match subroute path", newRequest("GET", "/after"), none}, {"no match parent path", newRequest("GET", "/before"), none}, {"no match as prefix", newRequest("GET", "/before/after/1234"), none}, {"no match append", newRequest("GET", "/before/after"), none}, }, }, { "host", func(r *Router) { r.Host("before.com").Host("after.com") }, func(r *Router) { r.Host("before.com").Subrouter().Host("after.com") }, []request{ {"no match before", newRequestHost("GET", "/", "before.com"), none}, {"no match other", newRequestHost("GET", "/", "other.com"), none}, {"matches after", newRequestHost("GET", "/", "after.com"), none}, }, }, { "queries variant keys", func(r *Router) { r.Queries("foo", "bar").Queries("cricket", "baseball") }, func(r *Router) { r.Queries("foo", "bar").Subrouter().Queries("cricket", "baseball") }, []request{ {"matches with all", newRequest("GET", "/?foo=bar&cricket=baseball"), both}, {"matches with more", newRequest("GET", "/?foo=bar&cricket=baseball&something=else"), both}, {"no match with none", newRequest("GET", "/"), none}, {"no match with some", newRequest("GET", "/?cricket=baseball"), none}, }, }, { "queries overlapping keys", func(r *Router) { r.Queries("foo", "bar").Queries("foo", "baz") }, func(r *Router) { r.Queries("foo", "bar").Subrouter().Queries("foo", "baz") }, []request{ {"no match old value", newRequest("GET", "/?foo=bar"), none}, {"no match diff value", newRequest("GET", "/?foo=bak"), none}, {"no match with none", newRequest("GET", "/"), none}, {"matches override", newRequest("GET", "/?foo=baz"), none}, }, }, { "header variant keys", func(r *Router) { r.Headers("foo", "bar").Headers("cricket", "baseball") }, func(r *Router) { r.Headers("foo", "bar").Subrouter().Headers("cricket", "baseball") }, []request{ { "matches with all", newRequestWithHeaders("GET", "/", "foo", "bar", "cricket", "baseball"), both, }, { "matches with more", newRequestWithHeaders("GET", "/", "foo", "bar", "cricket", "baseball", "something", "else"), both, }, {"no match with none", newRequest("GET", "/"), none}, {"no match with some", newRequestWithHeaders("GET", "/", "cricket", "baseball"), none}, }, }, { "header overlapping keys", func(r *Router) { r.Headers("foo", "bar").Headers("foo", "baz") }, func(r *Router) { r.Headers("foo", "bar").Subrouter().Headers("foo", "baz") }, []request{ {"no match old value", newRequestWithHeaders("GET", "/", "foo", "bar"), none}, {"no match diff value", newRequestWithHeaders("GET", "/", "foo", "bak"), none}, {"no match with none", newRequest("GET", "/"), none}, {"matches override", newRequestWithHeaders("GET", "/", "foo", "baz"), none}, }, }, { "method", func(r *Router) { r.Methods("POST").Methods("GET") }, func(r *Router) { r.Methods("POST").Subrouter().Methods("GET") }, []request{ {"matches before", newRequest("POST", "/"), none}, {"no match other", newRequest("HEAD", "/"), none}, {"matches override", newRequest("GET", "/"), none}, }, }, { "schemes", func(r *Router) { r.Schemes("http").Schemes("https") }, func(r *Router) { r.Schemes("http").Subrouter().Schemes("https") }, []request{ {"matches overrides", newRequest("GET", "https://www.example.com/"), none}, {"matches original", newRequest("GET", "http://www.example.com/"), none}, {"no match other", newRequest("GET", "ftp://www.example.com/"), none}, }, }, } // case -> request -> router for _, c := range cases { t.Run(c.Name, func(t *testing.T) { for _, req := range c.Requests { t.Run(req.Name, func(t *testing.T) { for _, v := range []struct { Name string Config func(*Router) Expected bool }{ {"subrouter", c.Subrouter, (req.Flags & subOnly) != 0}, {"standard", c.Standard, (req.Flags & stdOnly) != 0}, } { r := NewRouter() v.Config(r) if r.Match(req.Request, &RouteMatch{}) != v.Expected { if v.Expected { t.Errorf("expected %v match", v.Name) } else { t.Errorf("expected %v no match", v.Name) } } } }) } }) } } // verify that copyRouteConf copies fields as expected. func Test_copyRouteConf(t *testing.T) { var ( m MatcherFunc = func(*http.Request, *RouteMatch) bool { return true } b BuildVarsFunc = func(i map[string]string) map[string]string { return i } r, _ = newRouteRegexp("hi", regexpTypeHost, routeRegexpOptions{}) ) tests := []struct { name string args routeConf want routeConf }{ { "empty", routeConf{}, routeConf{}, }, { "full", routeConf{ useEncodedPath: true, strictSlash: true, skipClean: true, regexp: routeRegexpGroup{host: r, path: r, queries: []*routeRegexp{r}}, matchers: []matcher{m}, buildScheme: "https", buildVarsFunc: b, }, routeConf{ useEncodedPath: true, strictSlash: true, skipClean: true, regexp: routeRegexpGroup{host: r, path: r, queries: []*routeRegexp{r}}, matchers: []matcher{m}, buildScheme: "https", buildVarsFunc: b, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { // special case some incomparable fields of routeConf before delegating to reflect.DeepEqual got := copyRouteConf(tt.args) // funcs not comparable, just compare length of slices if len(got.matchers) != len(tt.want.matchers) { t.Errorf("matchers different lengths: %v %v", len(got.matchers), len(tt.want.matchers)) } got.matchers, tt.want.matchers = nil, nil // deep equal treats nil slice differently to empty slice so check for zero len first { bothZero := len(got.regexp.queries) == 0 && len(tt.want.regexp.queries) == 0 if !bothZero && !reflect.DeepEqual(got.regexp.queries, tt.want.regexp.queries) { t.Errorf("queries unequal: %v %v", got.regexp.queries, tt.want.regexp.queries) } got.regexp.queries, tt.want.regexp.queries = nil, nil } // funcs not comparable, just compare nullity if (got.buildVarsFunc == nil) != (tt.want.buildVarsFunc == nil) { t.Errorf("build vars funcs unequal: %v %v", got.buildVarsFunc == nil, tt.want.buildVarsFunc == nil) } got.buildVarsFunc, tt.want.buildVarsFunc = nil, nil // finish the deal if !reflect.DeepEqual(got, tt.want) { t.Errorf("route confs unequal: %v %v", got, tt.want) } }) } } func TestMethodNotAllowed(t *testing.T) { handler := func(w http.ResponseWriter, r *http.Request) { w.WriteHeader(http.StatusOK) } router := NewRouter() router.HandleFunc("/thing", handler).Methods(http.MethodGet) router.HandleFunc("/something", handler).Methods(http.MethodGet) w := NewRecorder() req := newRequest(http.MethodPut, "/thing") router.ServeHTTP(w, req) if w.Code != 405 { t.Fatalf("Expected status code 405 (got %d)", w.Code) } } func TestSubrouterNotFound(t *testing.T) { handler := func(w http.ResponseWriter, r *http.Request) { w.WriteHeader(http.StatusOK) } router := NewRouter() router.Path("/a").Subrouter().HandleFunc("/thing", handler).Methods(http.MethodGet) router.Path("/b").Subrouter().HandleFunc("/something", handler).Methods(http.MethodGet) w := NewRecorder() req := newRequest(http.MethodPut, "/not-present") router.ServeHTTP(w, req) if w.Code != 404 { t.Fatalf("Expected status code 404 (got %d)", w.Code) } } // 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 } // stringHandler returns a handler func that writes a message 's' to the // http.ResponseWriter. func stringHandler(s string) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { w.Write([]byte(s)) } } // newRequest is a helper function to create a new request with a method and url. // The request returned is a 'server' request as opposed to a 'client' one through // simulated write onto the wire and read off of the wire. // The differences between requests are detailed in the net/http package. func newRequest(method, url string) *http.Request { req, err := http.NewRequest(method, url, nil) if err != nil { panic(err) } // extract the escaped original host+path from url // http://localhost/path/here?v=1#frag -> //localhost/path/here opaque := "" if i := len(req.URL.Scheme); i > 0 { opaque = url[i+1:] } if i := strings.LastIndex(opaque, "?"); i > -1 { opaque = opaque[:i] } if i := strings.LastIndex(opaque, "#"); i > -1 { opaque = opaque[:i] } // Escaped host+path workaround as detailed in https://golang.org/pkg/net/url/#URL // for < 1.5 client side workaround req.URL.Opaque = opaque // Simulate writing to wire var buff bytes.Buffer req.Write(&buff) ioreader := bufio.NewReader(&buff) // Parse request off of 'wire' req, err = http.ReadRequest(ioreader) if err != nil { panic(err) } return req } // create a new request with the provided headers func newRequestWithHeaders(method, url string, headers ...string) *http.Request { req := newRequest(method, url) if len(headers)%2 != 0 { panic(fmt.Sprintf("Expected headers length divisible by 2 but got %v", len(headers))) } for i := 0; i < len(headers); i += 2 { req.Header.Set(headers[i], headers[i+1]) } return req } // newRequestHost a new request with a method, url, and host header func newRequestHost(method, url, host string) *http.Request { req, err := http.NewRequest(method, url, nil) if err != nil { panic(err) } req.Host = host return req }