diff options
author | robot-piglet <robot-piglet@yandex-team.com> | 2024-12-09 18:25:21 +0300 |
---|---|---|
committer | robot-piglet <robot-piglet@yandex-team.com> | 2024-12-09 19:18:57 +0300 |
commit | 13374e0884578812cda7697d0c5680122db59a37 (patch) | |
tree | 30a022eb841035299deb2b8c393b2902f0c21735 /contrib/python/websocket-client/py3/websocket/tests/test_url.py | |
parent | c7ade6d3bf7cd492235a61b77153351e422a28f3 (diff) | |
download | ydb-13374e0884578812cda7697d0c5680122db59a37.tar.gz |
Intermediate changes
commit_hash:034150f557268506d7bc0cbd8b5becf65f765593
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.py | 464 |
1 files changed, 0 insertions, 464 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 deleted file mode 100644 index 110fdfad70a..00000000000 --- a/contrib/python/websocket-client/py3/websocket/tests/test_url.py +++ /dev/null @@ -1,464 +0,0 @@ -# -*- coding: utf-8 -*- -# -import os -import unittest - -from websocket._url import ( - _is_address_in_network, - _is_no_proxy_host, - get_proxy_info, - parse_url, -) -from websocket._exceptions import WebSocketProxyException - -""" -test_url.py -websocket - WebSocket client library for Python - -Copyright 2024 engn33r - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -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")) - - def test_parse_url(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") - - 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 test_match_all(self): - self.assertTrue(_is_no_proxy_host("any.websocket.org", ["*"])) - self.assertTrue(_is_no_proxy_host("192.168.0.1", ["*"])) - self.assertFalse(_is_no_proxy_host("192.168.0.1", ["192.168.1.1"])) - self.assertFalse( - _is_no_proxy_host("any.websocket.org", ["other.websocket.org"]) - ) - 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 test_ip_address(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 test_ip_address_in_range(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 test_hostname_match(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 test_hostname_match_domain(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 test_proxy_from_args(self): - self.assertRaises( - WebSocketProxyException, - get_proxy_info, - "echo.websocket.events", - False, - proxy_host="localhost", - ) - 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", proxy_port=3128 - ), - ("localhost", 3128, None), - ) - - self.assertEqual( - get_proxy_info( - "echo.websocket.events", - False, - proxy_host="localhost", - proxy_port=9001, - proxy_auth=("a", "b"), - ), - ("localhost", 9001, ("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=8765, - proxy_auth=("a", "b"), - ), - ("localhost", 8765, ("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=[".websocket.events"], - ), - (None, 0, None), - ) - - def test_proxy_from_env(self): - os.environ["http_proxy"] = "http://localhost/" - 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) - ) - - os.environ["http_proxy"] = "http://localhost/" - os.environ["https_proxy"] = "http://localhost2/" - 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) - ) - - os.environ["http_proxy"] = "http://localhost/" - os.environ["https_proxy"] = "http://localhost2/" - 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) - ) - - 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) - ) - 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) - ) - - 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) - ) - 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) - ) - - os.environ["http_proxy"] = "http://a:b@localhost/" - 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")), - ) - - 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")), - ) - 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")), - ) - - 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")), - ) - 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")), - ) - - 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.events" - self.assertEqual(get_proxy_info("echo.websocket.events", 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.events" - self.assertEqual(get_proxy_info("echo.websocket.events", 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() |