Vendor dependencies

vendor/goji.io/router_test.go

@@ -0,0 +1,153 @@
+package goji
+
+import (
+	"context"
+	"net/http"
+	"reflect"
+	"testing"
+
+	"goji.io/internal"
+	"goji.io/pattern"
+)
+
+func TestNoMatch(t *testing.T) {
+	t.Parallel()
+
+	var rt router
+	rt.add(boolPattern(false), http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		t.Fatal("did not expect handler to be called")
+	}))
+	_, r := wr()
+	ctx := context.Background()
+	ctx = context.WithValue(ctx, internal.Pattern, boolPattern(true))
+	ctx = context.WithValue(ctx, internal.Pattern, boolPattern(true))
+	ctx = context.WithValue(ctx, pattern.Variable("answer"), 42)
+	ctx = context.WithValue(ctx, internal.Path, "/")
+
+	r = r.WithContext(ctx)
+	r = rt.route(r)
+	ctx = r.Context()
+
+	if p := ctx.Value(internal.Pattern); p != nil {
+		t.Errorf("unexpected pattern %v", p)
+	}
+	if h := ctx.Value(internal.Handler); h != nil {
+		t.Errorf("unexpected handler %v", h)
+	}
+	if h := ctx.Value(pattern.Variable("answer")); h != 42 {
+		t.Errorf("context didn't work: got %v, wanted %v", h, 42)
+	}
+}
+
+/*
+These are meant to be end-to-end torture tests of Goji's routing semantics. We
+generate a list of patterns that can be turned off incrementally with a global
+"high water mark." We then run a sequence of requests through the router N
+times, incrementing the mark each time. The net effect is that we can compile
+the entire set of routes Goji would attempt for every request, ensuring that the
+router is picking routes in the correct order.
+*/
+
+var TestRoutes = []testPattern{
+	testPattern{methods: nil, prefix: "/"},
+	testPattern{methods: nil, prefix: "/a"},
+	testPattern{methods: []string{"POST", "PUT"}, prefix: "/a"},
+	testPattern{methods: []string{"GET", "POST"}, prefix: "/a"},
+	testPattern{methods: []string{"GET"}, prefix: "/b"},
+	testPattern{methods: nil, prefix: "/ab"},
+	testPattern{methods: []string{"POST", "PUT"}, prefix: "/"},
+	testPattern{methods: nil, prefix: "/ba"},
+	testPattern{methods: nil, prefix: "/"},
+	testPattern{methods: []string{}, prefix: "/"},
+	testPattern{methods: nil, prefix: "/carl"},
+	testPattern{methods: []string{"PUT"}, prefix: "/car"},
+	testPattern{methods: nil, prefix: "/cake"},
+	testPattern{methods: nil, prefix: "/car"},
+	testPattern{methods: []string{"GET"}, prefix: "/c"},
+	testPattern{methods: []string{"POST"}, prefix: "/"},
+	testPattern{methods: []string{"PUT"}, prefix: "/"},
+}
+
+var RouterTests = []struct {
+	method, path string
+	results      []int
+}{
+	{"GET", "/", []int{0, 8, 8, 8, 8, 8, 8, 8, 8, -1, -1, -1, -1, -1, -1, -1, -1}},
+	{"POST", "/", []int{0, 6, 6, 6, 6, 6, 6, 8, 8, 15, 15, 15, 15, 15, 15, 15, -1}},
+	{"PUT", "/", []int{0, 6, 6, 6, 6, 6, 6, 8, 8, 16, 16, 16, 16, 16, 16, 16, 16}},
+	{"HEAD", "/", []int{0, 8, 8, 8, 8, 8, 8, 8, 8, -1, -1, -1, -1, -1, -1, -1, -1}},
+	{"GET", "/a", []int{0, 1, 3, 3, 8, 8, 8, 8, 8, -1, -1, -1, -1, -1, -1, -1, -1}},
+	{"POST", "/a", []int{0, 1, 2, 3, 6, 6, 6, 8, 8, 15, 15, 15, 15, 15, 15, 15, -1}},
+	{"PUT", "/a", []int{0, 1, 2, 6, 6, 6, 6, 8, 8, 16, 16, 16, 16, 16, 16, 16, 16}},
+	{"HEAD", "/a", []int{0, 1, 8, 8, 8, 8, 8, 8, 8, -1, -1, -1, -1, -1, -1, -1, -1}},
+	{"GET", "/b", []int{0, 4, 4, 4, 4, 8, 8, 8, 8, -1, -1, -1, -1, -1, -1, -1, -1}},
+	{"POST", "/b", []int{0, 6, 6, 6, 6, 6, 6, 8, 8, 15, 15, 15, 15, 15, 15, 15, -1}},
+	{"GET", "/ba", []int{0, 4, 4, 4, 4, 7, 7, 7, 8, -1, -1, -1, -1, -1, -1, -1, -1}},
+	{"GET", "/c", []int{0, 8, 8, 8, 8, 8, 8, 8, 8, 14, 14, 14, 14, 14, 14, -1, -1}},
+	{"POST", "/c", []int{0, 6, 6, 6, 6, 6, 6, 8, 8, 15, 15, 15, 15, 15, 15, 15, -1}},
+	{"GET", "/ab", []int{0, 1, 3, 3, 5, 5, 8, 8, 8, -1, -1, -1, -1, -1, -1, -1, -1}},
+	{"POST", "/ab", []int{0, 1, 2, 3, 5, 5, 6, 8, 8, 15, 15, 15, 15, 15, 15, 15, -1}},
+	{"GET", "/carl", []int{0, 8, 8, 8, 8, 8, 8, 8, 8, 10, 10, 13, 13, 13, 14, -1, -1}},
+	{"POST", "/carl", []int{0, 6, 6, 6, 6, 6, 6, 8, 8, 10, 10, 13, 13, 13, 15, 15, -1}},
+	{"HEAD", "/carl", []int{0, 8, 8, 8, 8, 8, 8, 8, 8, 10, 10, 13, 13, 13, -1, -1, -1}},
+	{"PUT", "/carl", []int{0, 6, 6, 6, 6, 6, 6, 8, 8, 10, 10, 11, 13, 13, 16, 16, 16}},
+	{"GET", "/cake", []int{0, 8, 8, 8, 8, 8, 8, 8, 8, 12, 12, 12, 12, 14, 14, -1, -1}},
+	{"PUT", "/cake", []int{0, 6, 6, 6, 6, 6, 6, 8, 8, 12, 12, 12, 12, 16, 16, 16, 16}},
+	{"OHAI", "/carl", []int{0, 8, 8, 8, 8, 8, 8, 8, 8, 10, 10, 13, 13, 13, -1, -1, -1}},
+}
+
+func TestRouter(t *testing.T) {
+	t.Parallel()
+
+	var rt router
+	mark := new(int)
+	for i, p := range TestRoutes {
+		i := i
+		p.index = i
+		p.mark = mark
+		rt.add(p, intHandler(i))
+	}
+
+	for i, test := range RouterTests {
+		r, err := http.NewRequest(test.method, test.path, nil)
+		if err != nil {
+			panic(err)
+		}
+		ctx := context.WithValue(context.Background(), internal.Path, test.path)
+		r = r.WithContext(ctx)
+
+		var out []int
+		for *mark = 0; *mark < len(TestRoutes); *mark++ {
+			r := rt.route(r)
+			ctx := r.Context()
+			if h := ctx.Value(internal.Handler); h != nil {
+				out = append(out, int(h.(intHandler)))
+			} else {
+				out = append(out, -1)
+			}
+		}
+		if !reflect.DeepEqual(out, test.results) {
+			t.Errorf("[%d] expected %v got %v", i, test.results, out)
+		}
+	}
+}
+
+type contextPattern struct{}
+
+func (contextPattern) Match(r *http.Request) *http.Request {
+	return r.WithContext(context.WithValue(r.Context(), pattern.Variable("hello"), "world"))
+}
+
+func TestRouterContextPropagation(t *testing.T) {
+	t.Parallel()
+
+	var rt router
+	rt.add(contextPattern{}, intHandler(0))
+	_, r := wr()
+	r = r.WithContext(context.WithValue(r.Context(), internal.Path, "/"))
+	r2 := rt.route(r)
+	ctx := r2.Context()
+	if hello := ctx.Value(pattern.Variable("hello")).(string); hello != "world" {
+		t.Fatalf("routed request didn't include correct key from pattern: %q", hello)
+	}
+}