diff options
Diffstat (limited to 'test/netlib')
| -rw-r--r-- | test/netlib/http/test_request.py | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/test/netlib/http/test_request.py b/test/netlib/http/test_request.py index b4ecfd4e..5672259e 100644 --- a/test/netlib/http/test_request.py +++ b/test/netlib/http/test_request.py @@ -104,25 +104,36 @@ class TestRequestUtils(object): def test_pretty_host(self): request = treq() + # Without host header assert request.pretty_host == "address" assert request.host == "address" - request.headers["host"] = "other" + # Same port as self.port (22) + request.headers["host"] = "other:22" assert request.pretty_host == "other" + # Different ports + request.headers["host"] = "other" + assert request.pretty_host == "address" assert request.host == "address" + # Empty host request.host = None assert request.pretty_host is None assert request.host is None # Invalid IDNA - request.headers["host"] = ".disqus.com" + request.headers["host"] = ".disqus.com:22" assert request.pretty_host == ".disqus.com" def test_pretty_url(self): request = treq() + # Without host header assert request.url == "http://address:22/path" assert request.pretty_url == "http://address:22/path" - request.headers["host"] = "other" + # Same port as self.port (22) + request.headers["host"] = "other:22" assert request.pretty_url == "http://other:22/path" + # Different ports + request.headers["host"] = "other" + assert request.pretty_url == "http://address:22/path" def test_pretty_url_authority(self): request = treq(first_line_format="authority") |