1 files changed, 58 insertions, 0 deletions
diff --git a/vendor/gopkg.in/throttled/throttled.v2/varyby_test.go b/vendor/gopkg.in/throttled/throttled.v2/varyby_test.go
new file mode 100644
index 000000000..66a5f4e98
--- /dev/null
+++ b/vendor/gopkg.in/throttled/throttled.v2/varyby_test.go
@@ -0,0 +1,58 @@
+package throttled_test
+
+import (
+	"net/http"
+	"net/url"
+	"testing"
+
+	"gopkg.in/throttled/throttled.v2"
+)
+
+func TestVaryBy(t *testing.T) {
+	u, err := url.Parse("http://localhost/test/path?q=s")
+	if err != nil {
+		panic(err)
+	}
+	ck := &http.Cookie{Name: "ssn", Value: "test"}
+	cases := []struct {
+		vb *throttled.VaryBy
+		r  *http.Request
+		k  string
+	}{
+		0: {nil, &http.Request{}, ""},
+		1: {&throttled.VaryBy{RemoteAddr: true}, &http.Request{RemoteAddr: "::"}, "::\n"},
+		2: {
+			&throttled.VaryBy{Method: true, Path: true},
+			&http.Request{Method: "POST", URL: u},
+			"post\n/test/path\n",
+		},
+		3: {
+			&throttled.VaryBy{Headers: []string{"Content-length"}},
+			&http.Request{Header: http.Header{"Content-Type": []string{"text/plain"}, "Content-Length": []string{"123"}}},
+			"123\n",
+		},
+		4: {
+			&throttled.VaryBy{Separator: ",", Method: true, Headers: []string{"Content-length"}, Params: []string{"q", "user"}},
+			&http.Request{Method: "GET", Header: http.Header{"Content-Type": []string{"text/plain"}, "Content-Length": []string{"123"}}, Form: url.Values{"q": []string{"s"}, "pwd": []string{"secret"}, "user": []string{"test"}}},
+			"get,123,s,test,",
+		},
+		5: {
+			&throttled.VaryBy{Cookies: []string{"ssn"}},
+			&http.Request{Header: http.Header{"Cookie": []string{ck.String()}}},
+			"test\n",
+		},
+		6: {
+			&throttled.VaryBy{Cookies: []string{"ssn"}, RemoteAddr: true, Custom: func(r *http.Request) string {
+				return "blah"
+			}},
+			&http.Request{Header: http.Header{"Cookie": []string{ck.String()}}},
+			"blah",
+		},
+	}
+	for i, c := range cases {
+		got := c.vb.Key(c.r)
+		if got != c.k {
+			t.Errorf("%d: expected '%s' (%d), got '%s' (%d)", i, c.k, len(c.k), got, len(got))
+		}
+	}
+}