diff options
author | Serhiy Storchaka <storchaka@gmail.com> | 2014-12-12 09:29:15 +0200 |
---|---|---|
committer | Serhiy Storchaka <storchaka@gmail.com> | 2014-12-12 09:29:15 +0200 |
commit | de45172262270798ee79dba266dced2faf6f58ce (patch) | |
tree | 076940752213084f9e3cc68bbcff4520ff5492b8 /Lib | |
parent | f2b9fb6af9e075b294f79949a4a3d909a8b621c5 (diff) | |
download | cpython-de45172262270798ee79dba266dced2faf6f58ce.tar.gz |
Issue #22095: Fixed HTTPConnection.set_tunnel with default port. The port
value in the host header was set to "None". Patch by Demian Brecht.
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/http/client.py | 8 | ||||
-rw-r--r-- | Lib/test/test_httplib.py | 8 |
2 files changed, 8 insertions, 8 deletions
diff --git a/Lib/http/client.py b/Lib/http/client.py index c0760ddeb6..6de4b0e73c 100644 --- a/Lib/http/client.py +++ b/Lib/http/client.py @@ -771,8 +771,7 @@ class HTTPConnection: if self.sock: raise RuntimeError("Can't set up tunnel for established connection") - self._tunnel_host = host - self._tunnel_port = port + self._tunnel_host, self._tunnel_port = self._get_hostport(host, port) if headers: self._tunnel_headers = headers else: @@ -802,9 +801,8 @@ class HTTPConnection: self.debuglevel = level def _tunnel(self): - (host, port) = self._get_hostport(self._tunnel_host, - self._tunnel_port) - connect_str = "CONNECT %s:%d HTTP/1.0\r\n" % (host, port) + connect_str = "CONNECT %s:%d HTTP/1.0\r\n" % (self._tunnel_host, + self._tunnel_port) connect_bytes = connect_str.encode("ascii") self.send(connect_bytes) for header, value in self._tunnel_headers.items(): diff --git a/Lib/test/test_httplib.py b/Lib/test/test_httplib.py index 49d767dba0..d87433919b 100644 --- a/Lib/test/test_httplib.py +++ b/Lib/test/test_httplib.py @@ -1068,11 +1068,13 @@ class TunnelTests(TestCase): self.assertEqual(conn.sock.host, 'proxy.com') self.assertEqual(conn.sock.port, 80) - self.assertTrue(b'CONNECT destination.com' in conn.sock.data) - self.assertTrue(b'Host: destination.com' in conn.sock.data) + self.assertIn(b'CONNECT destination.com', conn.sock.data) + # issue22095 + self.assertNotIn(b'Host: destination.com:None', conn.sock.data) + self.assertIn(b'Host: destination.com', conn.sock.data) # This test should be removed when CONNECT gets the HTTP/1.1 blessing - self.assertTrue(b'Host: proxy.com' not in conn.sock.data) + self.assertNotIn(b'Host: proxy.com', conn.sock.data) conn.close() conn.request('PUT', '/', '') |