From 83930ac113bf9f30f69ad6cbb33878d75f9684fe Mon Sep 17 00:00:00 2001
From: Brian Waldon <brian@waldon.cc>
Date: Thu, 29 Jan 2015 15:53:19 -0800
Subject: [PATCH] client: test DefaultCheckRedirect

---
 client/client_test.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/client/client_test.go b/client/client_test.go
index 43c02ac90..0b2564bc9 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -544,3 +544,23 @@ func TestRedirectFollowingHTTPClient(t *testing.T) {
 		}
 	}
 }
+
+func TestDefaultCheckRedirect(t *testing.T) {
+	tests := []struct {
+		num int
+		err error
+	}{
+		{0, nil},
+		{5, nil},
+		{10, nil},
+		{11, ErrTooManyRedirects},
+		{29, ErrTooManyRedirects},
+	}
+
+	for i, tt := range tests {
+		err := DefaultCheckRedirect(tt.num)
+		if !reflect.DeepEqual(tt.err, err) {
+			t.Errorf("#%d: want=%#v got=%#v", i, tt.err, err)
+		}
+	}
+}