diff options
author | Chris Wilcox <crwilcox@google.com> | 2018-04-27 14:23:56 -0700 |
---|---|---|
committer | Chris Wilcox <crwilcox@google.com> | 2018-04-27 14:23:56 -0700 |
commit | fc210c543bfb62ae3f1bbbc5d873fa39bd048ff4 (patch) | |
tree | b847a31ae96bb17a293f75b4e43701e0fb381e3f | |
parent | 6d18a448eff6c3cc150eb7cf1fd60bf94fad2b0c (diff) | |
download | urllib3-fc210c543bfb62ae3f1bbbc5d873fa39bd048ff4.tar.gz |
add unit tests
-rw-r--r-- | test/test_response.py | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/test/test_response.py b/test/test_response.py index 86ac786b..0b347c25 100644 --- a/test/test_response.py +++ b/test/test_response.py @@ -11,7 +11,7 @@ from urllib3.exceptions import ( DecodeError, ResponseNotChunked, ProtocolError, InvalidHeader ) from urllib3.packages.six.moves import http_client as httplib -from urllib3.util.retry import Retry +from urllib3.util.retry import Retry, RequestHistory from urllib3.util.response import is_fp_closed from base64 import b64decode @@ -675,6 +675,24 @@ class TestResponse(object): resp = HTTPResponse(fp, retries=retry) assert resp.retries == retry + def test_geturl(self): + fp = BytesIO(b'') + request_url = 'https://example.com' + resp = HTTPResponse(fp, request_url=request_url) + assert resp.geturl() == request_url + + def test_geturl_retries(self): + fp = BytesIO(b'') + resp = HTTPResponse(fp, request_url='http://example.com') + request_histories = [ + RequestHistory(method='GET', url='http://example.com', error=None, + status=301, redirect_location='https://example.com/'), + RequestHistory(method='GET', url='https://example.com/', error=None, + status=301, redirect_location='https://www.example.com')] + retry = Retry(history=request_histories) + resp = HTTPResponse(fp, retries=retry) + assert resp.geturl() == 'https://www.example.com' + class MockChunkedEncodingResponse(object): |