You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
188 lines
4.2 KiB
188 lines
4.2 KiB
package web
|
|
|
|
import (
|
|
"net/http"
|
|
"reflect"
|
|
"regexp"
|
|
"testing"
|
|
)
|
|
|
|
func pt(url string, match bool, params map[string]string) patternTest {
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return patternTest{
|
|
r: req,
|
|
match: match,
|
|
c: &C{},
|
|
cout: &C{URLParams: params},
|
|
}
|
|
}
|
|
|
|
type patternTest struct {
|
|
r *http.Request
|
|
match bool
|
|
c *C
|
|
cout *C
|
|
}
|
|
|
|
var patternTests = []struct {
|
|
pat Pattern
|
|
prefix string
|
|
tests []patternTest
|
|
}{
|
|
// Regexp tests
|
|
{parseRegexpPattern(regexp.MustCompile("^/hello$")),
|
|
"/hello", []patternTest{
|
|
pt("/hello", true, nil),
|
|
pt("/hell", false, nil),
|
|
pt("/hello/", false, nil),
|
|
pt("/hello/world", false, nil),
|
|
pt("/world", false, nil),
|
|
}},
|
|
{parseRegexpPattern(regexp.MustCompile("^/hello/(?P<name>[a-z]+)$")),
|
|
"/hello/", []patternTest{
|
|
pt("/hello/world", true, map[string]string{
|
|
"name": "world",
|
|
}),
|
|
pt("/hello/", false, nil),
|
|
pt("/hello/my/love", false, nil),
|
|
}},
|
|
{parseRegexpPattern(regexp.MustCompile(`^/a(?P<a>\d+)/b(?P<b>\d+)/?$`)),
|
|
"/a", []patternTest{
|
|
pt("/a1/b2", true, map[string]string{
|
|
"a": "1",
|
|
"b": "2",
|
|
}),
|
|
pt("/a9001/b007/", true, map[string]string{
|
|
"a": "9001",
|
|
"b": "007",
|
|
}),
|
|
pt("/a/b", false, nil),
|
|
pt("/a", false, nil),
|
|
pt("/squirrel", false, nil),
|
|
}},
|
|
{parseRegexpPattern(regexp.MustCompile(`^/hello/([a-z]+)$`)),
|
|
"/hello/", []patternTest{
|
|
pt("/hello/world", true, map[string]string{
|
|
"$1": "world",
|
|
}),
|
|
pt("/hello/", false, nil),
|
|
}},
|
|
{parseRegexpPattern(regexp.MustCompile("/hello")),
|
|
"/hello", []patternTest{
|
|
pt("/hello", true, nil),
|
|
pt("/hell", false, nil),
|
|
pt("/hello/", true, nil),
|
|
pt("/hello/world", true, nil),
|
|
pt("/world/hello", false, nil),
|
|
}},
|
|
|
|
// String pattern tests
|
|
{parseStringPattern("/hello"),
|
|
"/hello", []patternTest{
|
|
pt("/hello", true, nil),
|
|
pt("/hell", false, nil),
|
|
pt("/hello/", false, nil),
|
|
pt("/hello/world", false, nil),
|
|
}},
|
|
{parseStringPattern("/hello/:name"),
|
|
"/hello/", []patternTest{
|
|
pt("/hello/world", true, map[string]string{
|
|
"name": "world",
|
|
}),
|
|
pt("/hello/my.world;wow", true, map[string]string{
|
|
"name": "my.world;wow",
|
|
}),
|
|
pt("/hell", false, nil),
|
|
pt("/hello/", false, nil),
|
|
pt("/hello/my/love", false, nil),
|
|
}},
|
|
{parseStringPattern("/a/:a/b/:b"),
|
|
"/a/", []patternTest{
|
|
pt("/a/1/b/2", true, map[string]string{
|
|
"a": "1",
|
|
"b": "2",
|
|
}),
|
|
pt("/a", false, nil),
|
|
pt("/a//b/", false, nil),
|
|
pt("/a/1/b/2/3", false, nil),
|
|
}},
|
|
{parseStringPattern("/a/:b.:c"),
|
|
"/a/", []patternTest{
|
|
pt("/a/cat.gif", true, map[string]string{
|
|
"b": "cat",
|
|
"c": "gif",
|
|
}),
|
|
pt("/a/cat.tar.gz", true, map[string]string{
|
|
"b": "cat",
|
|
"c": "tar.gz",
|
|
}),
|
|
pt("/a", false, nil),
|
|
pt("/a/cat", false, nil),
|
|
pt("/a/cat/gif", false, nil),
|
|
pt("/a/cat.", false, nil),
|
|
pt("/a/cat/dog.gif", false, nil),
|
|
}},
|
|
|
|
// String prefix tests
|
|
{parseStringPattern("/user/:user/*"),
|
|
"/user/", []patternTest{
|
|
pt("/user/bob/", true, map[string]string{
|
|
"user": "bob",
|
|
"*": "/",
|
|
}),
|
|
pt("/user/bob/friends/123", true, map[string]string{
|
|
"user": "bob",
|
|
"*": "/friends/123",
|
|
}),
|
|
pt("/user/bob", false, nil),
|
|
pt("/user/", false, nil),
|
|
pt("/user//", false, nil),
|
|
}},
|
|
{parseStringPattern("/user/:user/friends/*"),
|
|
"/user/", []patternTest{
|
|
pt("/user/bob/friends/", true, map[string]string{
|
|
"user": "bob",
|
|
"*": "/",
|
|
}),
|
|
pt("/user/bob/friends/123", true, map[string]string{
|
|
"user": "bob",
|
|
"*": "/123",
|
|
}),
|
|
pt("/user/bob/enemies", false, nil),
|
|
}},
|
|
}
|
|
|
|
func TestPatterns(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
for _, pt := range patternTests {
|
|
p := pt.pat.Prefix()
|
|
if p != pt.prefix {
|
|
t.Errorf("Expected prefix %q for %v, got %q", pt.prefix,
|
|
pt.pat, p)
|
|
} else {
|
|
for _, test := range pt.tests {
|
|
runTest(t, pt.pat, test)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func runTest(t *testing.T, p Pattern, test patternTest) {
|
|
result := p.Match(test.r, test.c)
|
|
if result != test.match {
|
|
t.Errorf("Expected match(%v, %#v) to return %v", p,
|
|
test.r.URL.Path, test.match)
|
|
return
|
|
}
|
|
p.Run(test.r, test.c)
|
|
|
|
if !reflect.DeepEqual(test.c, test.cout) {
|
|
t.Errorf("Expected a context of %v, instead got %v", test.cout,
|
|
test.c)
|
|
}
|
|
}
|