Fixup more tests

1 files changed, 7 insertions, 4 deletions

diff --git a/test/netlib/http/test_request.py b/test/netlib/http/test_request.py
index 350b5405..5672259e 100644
--- a/test/netlib/http/test_request.py
+++ b/test/netlib/http/test_request.py
@@ -110,7 +110,7 @@ class TestRequestUtils(object):
         # Same port as self.port (22)
         request.headers["host"] = "other:22"
         assert request.pretty_host == "other"
-        # Different Ports
+        # Different ports
         request.headers["host"] = "other"
         assert request.pretty_host == "address"
         assert request.host == "address"
@@ -125,12 +125,15 @@ class TestRequestUtils(object):
 
     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"
-        request.headers["host"] = "other:33"
-        assert request.pretty_url == "http://other:33/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")