From 6f96da08c9d838e4ead43425440d6120d4a02d0f Mon Sep 17 00:00:00 2001 From: Shadab Zafar Date: Wed, 17 Feb 2016 08:48:59 +0530 Subject: Handle port numbers in host header from: https://github.com/mitmproxy/netlib/pull/121 --- test/netlib/http/test_request.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'test/netlib') diff --git a/test/netlib/http/test_request.py b/test/netlib/http/test_request.py index b4ecfd4e..4f8a34c2 100644 --- a/test/netlib/http/test_request.py +++ b/test/netlib/http/test_request.py @@ -106,6 +106,8 @@ class TestRequestUtils(object): request = treq() assert request.pretty_host == "address" assert request.host == "address" + request.headers["host"] = "other:22" + assert request.pretty_host == "other" request.headers["host"] = "other" assert request.pretty_host == "other" assert request.host == "address" @@ -123,6 +125,8 @@ class TestRequestUtils(object): assert request.pretty_url == "http://address:22/path" request.headers["host"] = "other" assert request.pretty_url == "http://other:22/path" + request.headers["host"] = "other:33" + assert request.pretty_url == "http://other:33/path" def test_pretty_url_authority(self): request = treq(first_line_format="authority") -- cgit v1.2.3