From 857d8eda3a4055d92626f71ff5b34ce8cc6ec167 Mon Sep 17 00:00:00 2001 From: Shivaram Lingamneni Date: Mon, 6 Aug 2012 13:57:04 -0700 Subject: [PATCH] red tests derived from @dhagrow's examples in #760 --- tests/test_requests.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/tests/test_requests.py b/tests/test_requests.py index 83ca69f..233daed 100755 --- a/tests/test_requests.py +++ b/tests/test_requests.py @@ -979,6 +979,22 @@ class RequestsTestSuite(TestSetup, TestBaseMixin, unittest.TestCase): t = json.loads(r.text) self.assertEqual(t.get('headers').get('Content-Type'), '') + def test_prefetch_redirect_bug(self): + """Test that prefetch persists across redirections.""" + res = get(httpbin('redirect/2'), prefetch=False) + # prefetch should persist across the redirect; if it doesn't, + # this attempt to iterate will crash because the content has already + # been read. + first_line = next(res.iter_lines()) + self.assertTrue(first_line.strip().startswith('{')) + + def test_prefetch_return_response_interaction(self): + """Test that prefetch can be overridden as a kwarg to `send`.""" + req = requests.get(httpbin('get'), return_response=False) + req.send(prefetch=False) + # content should not have been prefetched, and iter_lines should succeed + first_line = next(req.response.iter_lines()) + self.assertTrue(first_line.strip().startswith('{')) if __name__ == '__main__': unittest.main() -- 2.34.1