add a test for max_redirects in safe/normal modes

This commit is contained in:
Shivaram Lingamneni
2012-05-03 01:21:37 -07:00
parent a9033912b3
commit 41d1a427b0
+10
View File
@@ -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()