Merge pull request #904 from haikuginger/approved-session-managersessionmanager

Approved SessionManager changes

1 files changed, 22 insertions, 0 deletions

diff --git a/test/test_retry.py b/test/test_retry.py
index b6015c13..86cbb348 100644
--- a/test/test_retry.py
+++ b/test/test_retry.py
@@ -239,3 +239,25 @@ class RetryTest(unittest.TestCase):
         self.assertEqual(retry.history, (RequestHistory('GET', '/test1', connection_error, None, None),
                                          RequestHistory('POST', '/test2', read_error, None, None),
                                          RequestHistory('GET', '/test3', None, 500, None)))
+
+
+class RedirectTest(unittest.TestCase):
+
+    def setUp(self):
+        self.retries = Retry()
+
+    def test_redirect_method(self):
+        tests = [
+            ('GET', 303, 'GET'),
+            ('HEAD', 303, 'HEAD'),
+            ('PUT', 303, 'GET'),
+            ('DELETE', 303, 'GET'),
+            ('POST', 303, 'GET'),
+            ('OPTIONS', 303, 'GET'),
+            ('POST', 301, 'GET'),
+            ('POST', 302, 'GET'),
+            ('OPTIONS', 301, 'OPTIONS'),
+            ('DELETE', 302, 'DELETE')
+        ]
+        for test in tests:
+            self.assertEqual(test[2], self.retries.redirect_method(*test[:2]))