summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorChris Wilcox <crwilcox@google.com>2018-04-27 14:23:56 -0700
committerChris Wilcox <crwilcox@google.com>2018-04-27 14:23:56 -0700
commitfc210c543bfb62ae3f1bbbc5d873fa39bd048ff4 (patch)
treeb847a31ae96bb17a293f75b4e43701e0fb381e3f
parent6d18a448eff6c3cc150eb7cf1fd60bf94fad2b0c (diff)
downloadurllib3-fc210c543bfb62ae3f1bbbc5d873fa39bd048ff4.tar.gz
add unit tests
-rw-r--r--test/test_response.py20
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):