From e67cd15bc81943cba10786cb3f5de0473d9e8550 Mon Sep 17 00:00:00 2001 From: Cory Benfield Date: Thu, 5 Nov 2015 18:53:14 +0000 Subject: [PATCH] Fix breakage introduced by 8f591682 --- requests/adapters.py | 4 ++-- test_requests.py | 7 +++++++ 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/requests/adapters.py b/requests/adapters.py index c69c082e..fbe6b661 100644 --- a/requests/adapters.py +++ b/requests/adapters.py @@ -398,10 +398,10 @@ class HTTPAdapter(BaseAdapter): try: # For Python 2.7+ versions, use buffering of HTTP # responses - r = conn.getresponse(buffering=True) + r = low_conn.getresponse(buffering=True) except TypeError: # For compatibility with Python 2.6 versions and back - r = conn.getresponse() + r = low_conn.getresponse() resp = HTTPResponse.from_httplib( r, diff --git a/test_requests.py b/test_requests.py index 08d2f148..3a8e4b05 100755 --- a/test_requests.py +++ b/test_requests.py @@ -1133,6 +1133,13 @@ class TestRequests(object): next(r.iter_lines()) assert len(list(r.iter_lines())) == 3 + def test_chunked_upload(self, httpbin): + """Can safely send generators.""" + data = (c for c in 'abc') + r = requests.post(httpbin('post'), data=data, stream=True) + assert r.status_code == 200 + assert r.request.headers['Transfer-Encoding'] == 'chunked' + class TestContentEncodingDetection(unittest.TestCase):