aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/websocket-client/py3/websocket/tests/test_url.py
diff options
context:
space:
mode:
authorAlexSm <alex@ydb.tech>2023-12-21 15:05:38 +0100
committerGitHub <noreply@github.com>2023-12-21 15:05:38 +0100
commite98bcbc74422492351c51646dba3849a138a8ffc (patch)
tree38ad7a09b1f9c201ce8a7e3d69f2017388769224 /contrib/python/websocket-client/py3/websocket/tests/test_url.py
parent559d7083cd8378cb25b9e966dedcca21d413e338 (diff)
downloadydb-e98bcbc74422492351c51646dba3849a138a8ffc.tar.gz
Import libs 1 (#590)
* Import libs 1 * Add new file without extension * Add file missed in export config
Diffstat (limited to 'contrib/python/websocket-client/py3/websocket/tests/test_url.py')
-rw-r--r--contrib/python/websocket-client/py3/websocket/tests/test_url.py280
1 files changed, 204 insertions, 76 deletions
diff --git a/contrib/python/websocket-client/py3/websocket/tests/test_url.py b/contrib/python/websocket-client/py3/websocket/tests/test_url.py
index a74dd7669d..792399ebf5 100644
--- a/contrib/python/websocket-client/py3/websocket/tests/test_url.py
+++ b/contrib/python/websocket-client/py3/websocket/tests/test_url.py
@@ -2,7 +2,13 @@
#
import os
import unittest
-from websocket._url import get_proxy_info, parse_url, _is_address_in_network, _is_no_proxy_host
+
+from websocket._url import (
+ _is_address_in_network,
+ _is_no_proxy_host,
+ get_proxy_info,
+ parse_url,
+)
"""
test_url.py
@@ -25,11 +31,10 @@ limitations under the License.
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'))
+ 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")
@@ -126,57 +131,71 @@ class IsNoProxyHostTest(unittest.TestCase):
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", ["*"]))
+ 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, *'
+ 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", ["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'
+ 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", ["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'
+ 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", ["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'
+ 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", [".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'
+ os.environ["no_proxy"] = "my.websocket.org, .websocket.org"
self.assertTrue(_is_no_proxy_host("any.websocket.org", None))
@@ -209,96 +228,205 @@ class ProxyInfoTest(unittest.TestCase):
del os.environ["no_proxy"]
def testProxyFromArgs(self):
- self.assertEqual(get_proxy_info("echo.websocket.events", False, proxy_host="localhost"), ("localhost", 0, None))
- self.assertEqual(get_proxy_info("echo.websocket.events", False, proxy_host="localhost", proxy_port=3128),
- ("localhost", 3128, None))
- self.assertEqual(get_proxy_info("echo.websocket.events", True, proxy_host="localhost"), ("localhost", 0, None))
- self.assertEqual(get_proxy_info("echo.websocket.events", True, proxy_host="localhost", proxy_port=3128),
- ("localhost", 3128, None))
-
- self.assertEqual(get_proxy_info("echo.websocket.events", False, proxy_host="localhost", proxy_auth=("a", "b")),
- ("localhost", 0, ("a", "b")))
self.assertEqual(
- get_proxy_info("echo.websocket.events", False, proxy_host="localhost", proxy_port=3128, proxy_auth=("a", "b")),
- ("localhost", 3128, ("a", "b")))
- self.assertEqual(get_proxy_info("echo.websocket.events", True, proxy_host="localhost", proxy_auth=("a", "b")),
- ("localhost", 0, ("a", "b")))
+ get_proxy_info("echo.websocket.events", False, proxy_host="localhost"),
+ ("localhost", 0, None),
+ )
+ self.assertEqual(
+ get_proxy_info(
+ "echo.websocket.events", False, proxy_host="localhost", proxy_port=3128
+ ),
+ ("localhost", 3128, None),
+ )
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", True, proxy_host="localhost"),
+ ("localhost", 0, None),
+ )
+ self.assertEqual(
+ get_proxy_info(
+ "echo.websocket.events", True, proxy_host="localhost", proxy_port=3128
+ ),
+ ("localhost", 3128, None),
+ )
+
+ self.assertEqual(
+ get_proxy_info(
+ "echo.websocket.events",
+ False,
+ proxy_host="localhost",
+ proxy_auth=("a", "b"),
+ ),
+ ("localhost", 0, ("a", "b")),
+ )
+ self.assertEqual(
+ get_proxy_info(
+ "echo.websocket.events",
+ False,
+ proxy_host="localhost",
+ proxy_port=3128,
+ proxy_auth=("a", "b"),
+ ),
+ ("localhost", 3128, ("a", "b")),
+ )
self.assertEqual(
- get_proxy_info("echo.websocket.events", True, proxy_host="localhost", proxy_port=3128, proxy_auth=("a", "b")),
- ("localhost", 3128, ("a", "b")))
+ get_proxy_info(
+ "echo.websocket.events",
+ True,
+ proxy_host="localhost",
+ proxy_auth=("a", "b"),
+ ),
+ ("localhost", 0, ("a", "b")),
+ )
+ self.assertEqual(
+ get_proxy_info(
+ "echo.websocket.events",
+ True,
+ proxy_host="localhost",
+ proxy_port=3128,
+ proxy_auth=("a", "b"),
+ ),
+ ("localhost", 3128, ("a", "b")),
+ )
- self.assertEqual(get_proxy_info("echo.websocket.events", 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.events", True, proxy_host="localhost", proxy_port=3128,
- no_proxy=["echo.websocket.events"], proxy_auth=("a", "b")),
- (None, 0, None))
+ self.assertEqual(
+ get_proxy_info(
+ "echo.websocket.events",
+ 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.events",
+ True,
+ proxy_host="localhost",
+ proxy_port=3128,
+ no_proxy=["echo.websocket.events"],
+ proxy_auth=("a", "b"),
+ ),
+ (None, 0, None),
+ )
def testProxyFromEnv(self):
os.environ["http_proxy"] = "http://localhost/"
- self.assertEqual(get_proxy_info("echo.websocket.events", False), ("localhost", None, None))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", False), ("localhost", None, None)
+ )
os.environ["http_proxy"] = "http://localhost:3128/"
- self.assertEqual(get_proxy_info("echo.websocket.events", False), ("localhost", 3128, None))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", False), ("localhost", 3128, None)
+ )
os.environ["http_proxy"] = "http://localhost/"
os.environ["https_proxy"] = "http://localhost2/"
- self.assertEqual(get_proxy_info("echo.websocket.events", False), ("localhost", None, None))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", 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.events", False), ("localhost", 3128, None))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", False), ("localhost", 3128, None)
+ )
os.environ["http_proxy"] = "http://localhost/"
os.environ["https_proxy"] = "http://localhost2/"
- self.assertEqual(get_proxy_info("echo.websocket.events", True), ("localhost2", None, None))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", 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.events", True), ("localhost2", 3128, None))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", True), ("localhost2", 3128, None)
+ )
os.environ["http_proxy"] = ""
os.environ["https_proxy"] = "http://localhost2/"
- self.assertEqual(get_proxy_info("echo.websocket.events", True), ("localhost2", None, None))
- self.assertEqual(get_proxy_info("echo.websocket.events", False), (None, 0, None))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", True), ("localhost2", None, None)
+ )
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", False), (None, 0, None)
+ )
os.environ["http_proxy"] = ""
os.environ["https_proxy"] = "http://localhost2:3128/"
- self.assertEqual(get_proxy_info("echo.websocket.events", True), ("localhost2", 3128, None))
- self.assertEqual(get_proxy_info("echo.websocket.events", False), (None, 0, None))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", True), ("localhost2", 3128, None)
+ )
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", False), (None, 0, None)
+ )
os.environ["http_proxy"] = "http://localhost/"
os.environ["https_proxy"] = ""
self.assertEqual(get_proxy_info("echo.websocket.events", True), (None, 0, None))
- self.assertEqual(get_proxy_info("echo.websocket.events", False), ("localhost", None, None))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", False), ("localhost", None, None)
+ )
os.environ["http_proxy"] = "http://localhost:3128/"
os.environ["https_proxy"] = ""
self.assertEqual(get_proxy_info("echo.websocket.events", True), (None, 0, None))
- self.assertEqual(get_proxy_info("echo.websocket.events", False), ("localhost", 3128, None))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", False), ("localhost", 3128, None)
+ )
os.environ["http_proxy"] = "http://a:b@localhost/"
- self.assertEqual(get_proxy_info("echo.websocket.events", False), ("localhost", None, ("a", "b")))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", False),
+ ("localhost", None, ("a", "b")),
+ )
os.environ["http_proxy"] = "http://a:b@localhost:3128/"
- self.assertEqual(get_proxy_info("echo.websocket.events", False), ("localhost", 3128, ("a", "b")))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", 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.events", False), ("localhost", None, ("a", "b")))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", 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.events", False), ("localhost", 3128, ("a", "b")))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", 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.events", True), ("localhost2", None, ("a", "b")))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", 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.events", True), ("localhost2", 3128, ("a", "b")))
-
- os.environ["http_proxy"] = "http://john%40example.com:P%40SSWORD@localhost:3128/"
- os.environ["https_proxy"] = "http://john%40example.com:P%40SSWORD@localhost2:3128/"
- self.assertEqual(get_proxy_info("echo.websocket.events", True), ("localhost2", 3128, ("john@example.com", "P@SSWORD")))
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", True),
+ ("localhost2", 3128, ("a", "b")),
+ )
+
+ os.environ[
+ "http_proxy"
+ ] = "http://john%40example.com:P%40SSWORD@localhost:3128/"
+ os.environ[
+ "https_proxy"
+ ] = "http://john%40example.com:P%40SSWORD@localhost2:3128/"
+ self.assertEqual(
+ get_proxy_info("echo.websocket.events", True),
+ ("localhost2", 3128, ("john@example.com", "P@SSWORD")),
+ )
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")))
+ 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.events"