diff options
author | Белавин Денис <luckydenis@yandex-team.ru> | 2019-05-14 12:11:56 +0300 |
---|---|---|
committer | Белавин Денис <luckydenis@yandex-team.ru> | 2019-05-14 12:11:56 +0300 |
commit | bd100472443dcf4189630f6b1ed3ec0a61259dd8 (patch) | |
tree | 30a3c951bec6b71f605cd2c3ee5dab23cbab2736 /tests | |
parent | d91fe00983f1c66b9a428fe1ad15bf6ea8238972 (diff) | |
download | python-requests-bd100472443dcf4189630f6b1ed3ec0a61259dd8.tar.gz |
#4965 fix: Accessing response.content twice removes forgets read error.
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_lowlevel.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/tests/test_lowlevel.py b/tests/test_lowlevel.py index 82c3b25a..59c43220 100644 --- a/tests/test_lowlevel.py +++ b/tests/test_lowlevel.py @@ -3,6 +3,7 @@ import pytest import threading import requests +from requests.exceptions import ChunkedEncodingError from tests.testserver.server import Server, consume_socket_content @@ -307,3 +308,43 @@ def test_fragment_update_on_redirect(): assert r.url == 'http://{}:{}/final-url/#relevant-section'.format(host, port) close_server.set() + + +def test_response_content_retains_error(): + """Verify that accessing response.content retains an error. + + See https://github.com/kennethreitz/requests/issues/4965 + """ + + data = "Some random stuff to read from remove server.\n" + + def response_handler(sock): + req = consume_socket_content(sock, timeout=0.5) + + # Send invalid chunked data (length mismatch) + sock.send( + b'HTTP/1.1 200 OK\r\n' + b'Transfer-Encoding: chunked\r\n' + b'\r\n2\r\n42\r\n8\r\n123\r\n' # 5 bytes missing + ) + + close_server = threading.Event() + server = Server(response_handler, wait_to_close_event=close_server) + + with server as (host, port): + url = 'http://{}:{}/path'.format(host, port) + r = requests.post(url, stream=True) + with pytest.raises(ChunkedEncodingError): + r.content + + # Access the bad response data again, I would expect the same + # error again. + + try: + content = r.content + except ChunkedEncodingError: + pass # fine, same exception + else: + assert False, "error response has content: {0!r}".format(content) + close_server.set() + |