You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
302 lines
14 KiB
302 lines
14 KiB
# -*- coding: utf-8 -*-
|
|
#
|
|
import os
|
|
import unittest
|
|
from websocket._url import get_proxy_info, parse_url, _is_address_in_network, _is_no_proxy_host
|
|
|
|
"""
|
|
test_url.py
|
|
websocket - WebSocket client library for Python
|
|
|
|
Copyright 2022 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 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")
|
|
|
|
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.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")))
|
|
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))
|
|
|
|
def testProxyFromEnv(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"] = "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()
|