From: Shivaram Lingamneni <slingamn@cs.stanford.edu>
Date: Thu, 3 May 2012 08:21:37 +0000 (-0700)
Subject: add a test for max_redirects in safe/normal modes
X-Git-Tag: v0.12.1~11^2~1
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=41d1a427b06acd33ae69728f4dc63e56078719b0;p=services%2Fpython-requests.git

add a test for max_redirects in safe/normal modes
---

diff --git a/tests/test_requests.py b/tests/test_requests.py
index fb0d75c..db176f5 100755
--- a/tests/test_requests.py
+++ b/tests/test_requests.py
@@ -847,5 +847,15 @@ class RequestsTestSuite(TestSetup, TestBaseMixin, unittest.TestCase):
         r = requests.get(httpbin('status', '404'))
         r.text
 
+    def test_max_redirects(self):
+        def unsafe_callable():
+            requests.get("http://httpbin.org/redirect/3", config=dict(max_redirects=2))
+        self.assertRaises(requests.exceptions.TooManyRedirects, unsafe_callable)
+
+        # add safe mode
+        response = requests.get("http://httpbin.org/redirect/3", config=dict(safe_mode=True, max_redirects=2))
+        self.assertTrue(response.content is None)
+        self.assertTrue(isinstance(response.error, requests.exceptions.TooManyRedirects))
+
 if __name__ == '__main__':
     unittest.main()