May 2, 2023 · Apr 12, 2023 · Apr 12, 2023 · Apr 12, 2023 · Apr 25, 2023 · Apr 25, 2023
diff --git a/Lib/test/test_httplib.py b/Lib/test/test_httplib.py
        lines = output.getvalue().splitlines()
        self.assertIn('header: {}'.format(expected_header), lines)

    def test_tunnel_leak(self):
        sock = None

        def _create_connection(address, timeout=None, source_address=None):
            nonlocal sock
            sock = FakeSocket(
                'HTTP/1.1 404 NOT FOUND\r\n\r\n',
                host=address[0],
                port=address[1],
            )
            return sock

        self.conn._create_connection = _create_connection
        self.conn.set_tunnel('destination.com')
        exc = None
        try:
            self.conn.request('HEAD', '/', '')
        except OSError as e:
            # keeping a reference to exc keeps response alive in the traceback
            exc = e
        self.assertIsNotNone(exc)
        self.assertTrue(sock.file_closed)


 if __name__ == '__main__':
    unittest.main(verbosity=2)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2390,6 +2390,29 @@ def test_tunnel_debuglog(self):
		lines = output.getvalue().splitlines()
		self.assertIn('header: {}'.format(expected_header), lines)

		def test_tunnel_leak(self):
		sock = None

		def _create_connection(address, timeout=None, source_address=None):
		nonlocal sock
		sock = FakeSocket(
		'HTTP/1.1 404 NOT FOUND\r\n\r\n',
		host=address[0],
		port=address[1],
		)
		return sock

		self.conn._create_connection = _create_connection
		self.conn.set_tunnel('destination.com')
gpshead marked this conversation as resolved. Show resolvedHide resolved
		exc = None
		try:
		self.conn.request('HEAD', '/', '')
		except OSError as e:
		# keeping a reference to exc keeps response alive in the traceback
		exc = e
		self.assertIsNotNone(exc)
		self.assertTrue(sock.file_closed)


		if __name__ == '__main__':
		unittest.main(verbosity=2)