summaryrefslogtreecommitdiffstats
path: root/contrib/python/websocket-client/py2/websocket/tests/test_url.py
diff options
context:
space:
mode:
authorrobot-piglet <[email protected]>2024-12-09 18:25:21 +0300
committerrobot-piglet <[email protected]>2024-12-09 19:18:57 +0300
commit13374e0884578812cda7697d0c5680122db59a37 (patch)
tree30a022eb841035299deb2b8c393b2902f0c21735 /contrib/python/websocket-client/py2/websocket/tests/test_url.py
parentc7ade6d3bf7cd492235a61b77153351e422a28f3 (diff)
Intermediate changes
commit_hash:034150f557268506d7bc0cbd8b5becf65f765593
Diffstat (limited to 'contrib/python/websocket-client/py2/websocket/tests/test_url.py')
-rw-r--r--contrib/python/websocket-client/py2/websocket/tests/test_url.py309
1 files changed, 0 insertions, 309 deletions
diff --git a/contrib/python/websocket-client/py2/websocket/tests/test_url.py b/contrib/python/websocket-client/py2/websocket/tests/test_url.py
deleted file mode 100644
index b1d8e06f23f..00000000000
--- a/contrib/python/websocket-client/py2/websocket/tests/test_url.py
+++ /dev/null
@@ -1,309 +0,0 @@
-# -*- coding: utf-8 -*-
-#
-"""
-websocket - WebSocket client library for Python
-
-Copyright (C) 2010 Hiroki Ohtani(liris)
-
- This library is free software; you can redistribute it and/or
- modify it under the terms of the GNU Lesser General Public
- License as published by the Free Software Foundation; either
- version 2.1 of the License, or (at your option) any later version.
-
- This library is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- Lesser General Public License for more details.
-
- You should have received a copy of the GNU Lesser General Public
- License along with this library; if not, write to the Free Software
- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
-
-"""
-
-import sys
-import os
-
-from websocket._url import get_proxy_info, parse_url, _is_address_in_network, _is_no_proxy_host
-
-if sys.version_info[0] == 2 and sys.version_info[1] < 7:
- import unittest2 as unittest
-else:
- import unittest
-sys.path[0:0] = [""]
-
-
-class UrlTest(unittest.TestCase):
-
- def test_address_in_network(self):
- self.assertTrue(_is_address_in_network('127.0.0.1', '127.0.0.0/8'))
- self.assertTrue(_is_address_in_network('127.1.0.1', '127.0.0.0/8'))
- self.assertFalse(_is_address_in_network('127.1.0.1', '127.0.0.0/24'))
-
- def testParseUrl(self):
- p = parse_url("ws://www.example.com/r")
- self.assertEqual(p[0], "www.example.com")
- self.assertEqual(p[1], 80)
- self.assertEqual(p[2], "/r")
- self.assertEqual(p[3], False)
-
- p = parse_url("ws://www.example.com/r/")
- self.assertEqual(p[0], "www.example.com")
- self.assertEqual(p[1], 80)
- self.assertEqual(p[2], "/r/")
- self.assertEqual(p[3], False)
-
- p = parse_url("ws://www.example.com/")
- self.assertEqual(p[0], "www.example.com")
- self.assertEqual(p[1], 80)
- self.assertEqual(p[2], "/")
- self.assertEqual(p[3], False)
-
- p = parse_url("ws://www.example.com")
- self.assertEqual(p[0], "www.example.com")
- self.assertEqual(p[1], 80)
- self.assertEqual(p[2], "/")
- self.assertEqual(p[3], False)
-
- p = parse_url("ws://www.example.com:8080/r")
- self.assertEqual(p[0], "www.example.com")
- self.assertEqual(p[1], 8080)
- self.assertEqual(p[2], "/r")
- self.assertEqual(p[3], False)
-
- p = parse_url("ws://www.example.com:8080/")
- self.assertEqual(p[0], "www.example.com")
- self.assertEqual(p[1], 8080)
- self.assertEqual(p[2], "/")
- self.assertEqual(p[3], False)
-
- p = parse_url("ws://www.example.com:8080")
- self.assertEqual(p[0], "www.example.com")
- self.assertEqual(p[1], 8080)
- self.assertEqual(p[2], "/")
- self.assertEqual(p[3], False)
-
- p = parse_url("wss://www.example.com:8080/r")
- self.assertEqual(p[0], "www.example.com")
- self.assertEqual(p[1], 8080)
- self.assertEqual(p[2], "/r")
- self.assertEqual(p[3], True)
-
- p = parse_url("wss://www.example.com:8080/r?key=value")
- self.assertEqual(p[0], "www.example.com")
- self.assertEqual(p[1], 8080)
- self.assertEqual(p[2], "/r?key=value")
- self.assertEqual(p[3], True)
-
- self.assertRaises(ValueError, parse_url, "http://www.example.com/r")
-
- if sys.version_info[0] == 2 and sys.version_info[1] < 7:
- return
-
- p = parse_url("ws://[2a03:4000:123:83::3]/r")
- self.assertEqual(p[0], "2a03:4000:123:83::3")
- self.assertEqual(p[1], 80)
- self.assertEqual(p[2], "/r")
- self.assertEqual(p[3], False)
-
- p = parse_url("ws://[2a03:4000:123:83::3]:8080/r")
- self.assertEqual(p[0], "2a03:4000:123:83::3")
- self.assertEqual(p[1], 8080)
- self.assertEqual(p[2], "/r")
- self.assertEqual(p[3], False)
-
- p = parse_url("wss://[2a03:4000:123:83::3]/r")
- self.assertEqual(p[0], "2a03:4000:123:83::3")
- self.assertEqual(p[1], 443)
- self.assertEqual(p[2], "/r")
- self.assertEqual(p[3], True)
-
- p = parse_url("wss://[2a03:4000:123:83::3]:8080/r")
- self.assertEqual(p[0], "2a03:4000:123:83::3")
- self.assertEqual(p[1], 8080)
- self.assertEqual(p[2], "/r")
- self.assertEqual(p[3], True)
-
-
-class IsNoProxyHostTest(unittest.TestCase):
- def setUp(self):
- self.no_proxy = os.environ.get("no_proxy", None)
- if "no_proxy" in os.environ:
- del os.environ["no_proxy"]
-
- def tearDown(self):
- if self.no_proxy:
- os.environ["no_proxy"] = self.no_proxy
- elif "no_proxy" in os.environ:
- del os.environ["no_proxy"]
-
- def testMatchAll(self):
- self.assertTrue(_is_no_proxy_host("any.websocket.org", ['*']))
- self.assertTrue(_is_no_proxy_host("192.168.0.1", ['*']))
- self.assertTrue(_is_no_proxy_host("any.websocket.org", ['other.websocket.org', '*']))
- os.environ['no_proxy'] = '*'
- self.assertTrue(_is_no_proxy_host("any.websocket.org", None))
- self.assertTrue(_is_no_proxy_host("192.168.0.1", None))
- os.environ['no_proxy'] = 'other.websocket.org, *'
- self.assertTrue(_is_no_proxy_host("any.websocket.org", None))
-
- def testIpAddress(self):
- self.assertTrue(_is_no_proxy_host("127.0.0.1", ['127.0.0.1']))
- self.assertFalse(_is_no_proxy_host("127.0.0.2", ['127.0.0.1']))
- self.assertTrue(_is_no_proxy_host("127.0.0.1", ['other.websocket.org', '127.0.0.1']))
- self.assertFalse(_is_no_proxy_host("127.0.0.2", ['other.websocket.org', '127.0.0.1']))
- os.environ['no_proxy'] = '127.0.0.1'
- self.assertTrue(_is_no_proxy_host("127.0.0.1", None))
- self.assertFalse(_is_no_proxy_host("127.0.0.2", None))
- os.environ['no_proxy'] = 'other.websocket.org, 127.0.0.1'
- self.assertTrue(_is_no_proxy_host("127.0.0.1", None))
- self.assertFalse(_is_no_proxy_host("127.0.0.2", None))
-
- def testIpAddressInRange(self):
- self.assertTrue(_is_no_proxy_host("127.0.0.1", ['127.0.0.0/8']))
- self.assertTrue(_is_no_proxy_host("127.0.0.2", ['127.0.0.0/8']))
- self.assertFalse(_is_no_proxy_host("127.1.0.1", ['127.0.0.0/24']))
- os.environ['no_proxy'] = '127.0.0.0/8'
- self.assertTrue(_is_no_proxy_host("127.0.0.1", None))
- self.assertTrue(_is_no_proxy_host("127.0.0.2", None))
- os.environ['no_proxy'] = '127.0.0.0/24'
- self.assertFalse(_is_no_proxy_host("127.1.0.1", None))
-
- def testHostnameMatch(self):
- self.assertTrue(_is_no_proxy_host("my.websocket.org", ['my.websocket.org']))
- self.assertTrue(_is_no_proxy_host("my.websocket.org", ['other.websocket.org', 'my.websocket.org']))
- self.assertFalse(_is_no_proxy_host("my.websocket.org", ['other.websocket.org']))
- os.environ['no_proxy'] = 'my.websocket.org'
- self.assertTrue(_is_no_proxy_host("my.websocket.org", None))
- self.assertFalse(_is_no_proxy_host("other.websocket.org", None))
- os.environ['no_proxy'] = 'other.websocket.org, my.websocket.org'
- self.assertTrue(_is_no_proxy_host("my.websocket.org", None))
-
- def testHostnameMatchDomain(self):
- self.assertTrue(_is_no_proxy_host("any.websocket.org", ['.websocket.org']))
- self.assertTrue(_is_no_proxy_host("my.other.websocket.org", ['.websocket.org']))
- self.assertTrue(_is_no_proxy_host("any.websocket.org", ['my.websocket.org', '.websocket.org']))
- self.assertFalse(_is_no_proxy_host("any.websocket.com", ['.websocket.org']))
- os.environ['no_proxy'] = '.websocket.org'
- self.assertTrue(_is_no_proxy_host("any.websocket.org", None))
- self.assertTrue(_is_no_proxy_host("my.other.websocket.org", None))
- self.assertFalse(_is_no_proxy_host("any.websocket.com", None))
- os.environ['no_proxy'] = 'my.websocket.org, .websocket.org'
- self.assertTrue(_is_no_proxy_host("any.websocket.org", None))
-
-
-class ProxyInfoTest(unittest.TestCase):
- def setUp(self):
- self.http_proxy = os.environ.get("http_proxy", None)
- self.https_proxy = os.environ.get("https_proxy", None)
- self.no_proxy = os.environ.get("no_proxy", None)
- if "http_proxy" in os.environ:
- del os.environ["http_proxy"]
- if "https_proxy" in os.environ:
- del os.environ["https_proxy"]
- if "no_proxy" in os.environ:
- del os.environ["no_proxy"]
-
- def tearDown(self):
- if self.http_proxy:
- os.environ["http_proxy"] = self.http_proxy
- elif "http_proxy" in os.environ:
- del os.environ["http_proxy"]
-
- if self.https_proxy:
- os.environ["https_proxy"] = self.https_proxy
- elif "https_proxy" in os.environ:
- del os.environ["https_proxy"]
-
- if self.no_proxy:
- os.environ["no_proxy"] = self.no_proxy
- elif "no_proxy" in os.environ:
- del os.environ["no_proxy"]
-
- def testProxyFromArgs(self):
- self.assertEqual(get_proxy_info("echo.websocket.org", False, proxy_host="localhost"), ("localhost", 0, None))
- self.assertEqual(get_proxy_info("echo.websocket.org", False, proxy_host="localhost", proxy_port=3128),
- ("localhost", 3128, None))
- self.assertEqual(get_proxy_info("echo.websocket.org", True, proxy_host="localhost"), ("localhost", 0, None))
- self.assertEqual(get_proxy_info("echo.websocket.org", True, proxy_host="localhost", proxy_port=3128),
- ("localhost", 3128, None))
-
- self.assertEqual(get_proxy_info("echo.websocket.org", False, proxy_host="localhost", proxy_auth=("a", "b")),
- ("localhost", 0, ("a", "b")))
- self.assertEqual(
- get_proxy_info("echo.websocket.org", False, proxy_host="localhost", proxy_port=3128, proxy_auth=("a", "b")),
- ("localhost", 3128, ("a", "b")))
- self.assertEqual(get_proxy_info("echo.websocket.org", True, proxy_host="localhost", proxy_auth=("a", "b")),
- ("localhost", 0, ("a", "b")))
- self.assertEqual(
- get_proxy_info("echo.websocket.org", True, proxy_host="localhost", proxy_port=3128, proxy_auth=("a", "b")),
- ("localhost", 3128, ("a", "b")))
-
- self.assertEqual(get_proxy_info("echo.websocket.org", True, proxy_host="localhost", proxy_port=3128,
- no_proxy=["example.com"], proxy_auth=("a", "b")),
- ("localhost", 3128, ("a", "b")))
- self.assertEqual(get_proxy_info("echo.websocket.org", True, proxy_host="localhost", proxy_port=3128,
- no_proxy=["echo.websocket.org"], proxy_auth=("a", "b")),
- (None, 0, None))
-
- def testProxyFromEnv(self):
- os.environ["http_proxy"] = "http://localhost/"
- self.assertEqual(get_proxy_info("echo.websocket.org", False), ("localhost", None, None))
- os.environ["http_proxy"] = "http://localhost:3128/"
- self.assertEqual(get_proxy_info("echo.websocket.org", False), ("localhost", 3128, None))
-
- os.environ["http_proxy"] = "http://localhost/"
- os.environ["https_proxy"] = "http://localhost2/"
- self.assertEqual(get_proxy_info("echo.websocket.org", False), ("localhost", None, None))
- os.environ["http_proxy"] = "http://localhost:3128/"
- os.environ["https_proxy"] = "http://localhost2:3128/"
- self.assertEqual(get_proxy_info("echo.websocket.org", False), ("localhost", 3128, None))
-
- os.environ["http_proxy"] = "http://localhost/"
- os.environ["https_proxy"] = "http://localhost2/"
- self.assertEqual(get_proxy_info("echo.websocket.org", True), ("localhost2", None, None))
- os.environ["http_proxy"] = "http://localhost:3128/"
- os.environ["https_proxy"] = "http://localhost2:3128/"
- self.assertEqual(get_proxy_info("echo.websocket.org", True), ("localhost2", 3128, None))
-
- os.environ["http_proxy"] = "http://a:b@localhost/"
- self.assertEqual(get_proxy_info("echo.websocket.org", False), ("localhost", None, ("a", "b")))
- os.environ["http_proxy"] = "http://a:b@localhost:3128/"
- self.assertEqual(get_proxy_info("echo.websocket.org", False), ("localhost", 3128, ("a", "b")))
-
- os.environ["http_proxy"] = "http://a:b@localhost/"
- os.environ["https_proxy"] = "http://a:b@localhost2/"
- self.assertEqual(get_proxy_info("echo.websocket.org", False), ("localhost", None, ("a", "b")))
- os.environ["http_proxy"] = "http://a:b@localhost:3128/"
- os.environ["https_proxy"] = "http://a:b@localhost2:3128/"
- self.assertEqual(get_proxy_info("echo.websocket.org", False), ("localhost", 3128, ("a", "b")))
-
- os.environ["http_proxy"] = "http://a:b@localhost/"
- os.environ["https_proxy"] = "http://a:b@localhost2/"
- self.assertEqual(get_proxy_info("echo.websocket.org", True), ("localhost2", None, ("a", "b")))
- os.environ["http_proxy"] = "http://a:b@localhost:3128/"
- os.environ["https_proxy"] = "http://a:b@localhost2:3128/"
- self.assertEqual(get_proxy_info("echo.websocket.org", True), ("localhost2", 3128, ("a", "b")))
-
- os.environ["http_proxy"] = "http://a:b@localhost/"
- os.environ["https_proxy"] = "http://a:b@localhost2/"
- os.environ["no_proxy"] = "example1.com,example2.com"
- self.assertEqual(get_proxy_info("example.1.com", True), ("localhost2", None, ("a", "b")))
- os.environ["http_proxy"] = "http://a:b@localhost:3128/"
- os.environ["https_proxy"] = "http://a:b@localhost2:3128/"
- os.environ["no_proxy"] = "example1.com,example2.com, echo.websocket.org"
- self.assertEqual(get_proxy_info("echo.websocket.org", True), (None, 0, None))
- os.environ["http_proxy"] = "http://a:b@localhost:3128/"
- os.environ["https_proxy"] = "http://a:b@localhost2:3128/"
- os.environ["no_proxy"] = "example1.com,example2.com, .websocket.org"
- self.assertEqual(get_proxy_info("echo.websocket.org", True), (None, 0, None))
-
- os.environ["http_proxy"] = "http://a:b@localhost:3128/"
- os.environ["https_proxy"] = "http://a:b@localhost2:3128/"
- os.environ["no_proxy"] = "127.0.0.0/8, 192.168.0.0/16"
- self.assertEqual(get_proxy_info("127.0.0.1", False), (None, 0, None))
- self.assertEqual(get_proxy_info("192.168.1.1", False), (None, 0, None))
-
-
-if __name__ == "__main__":
- unittest.main()