aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/websocket-client/py3/websocket/tests/test_http.py
diff options
context:
space:
mode:
authorrobot-piglet <robot-piglet@yandex-team.com>2024-05-08 11:10:05 +0300
committerrobot-piglet <robot-piglet@yandex-team.com>2024-05-08 11:21:05 +0300
commit4196c19bc0753c1c2fe4d29f231bb49039430f9f (patch)
treef02ac799e7260353047c65d46fa5ce206f2e9303 /contrib/python/websocket-client/py3/websocket/tests/test_http.py
parentc3d51e170ca39637234e72ae19dcbc53b737a83e (diff)
downloadydb-4196c19bc0753c1c2fe4d29f231bb49039430f9f.tar.gz
Intermediate changes
Diffstat (limited to 'contrib/python/websocket-client/py3/websocket/tests/test_http.py')
-rw-r--r--contrib/python/websocket-client/py3/websocket/tests/test_http.py29
1 files changed, 14 insertions, 15 deletions
diff --git a/contrib/python/websocket-client/py3/websocket/tests/test_http.py b/contrib/python/websocket-client/py3/websocket/tests/test_http.py
index 9df36e43ab..b8b363acc4 100644
--- a/contrib/python/websocket-client/py3/websocket/tests/test_http.py
+++ b/contrib/python/websocket-client/py3/websocket/tests/test_http.py
@@ -7,7 +7,7 @@ import ssl
import unittest
import websocket
-import websocket as ws
+from websocket._exceptions import WebSocketProxyException, WebSocketException
from websocket._http import (
_get_addrinfo_list,
_start_proxied_socket,
@@ -15,13 +15,14 @@ from websocket._http import (
connect,
proxy_info,
read_headers,
+ HAVE_PYTHON_SOCKS,
)
"""
test_http.py
websocket - WebSocket client library for Python
-Copyright 2023 engn33r
+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.
@@ -94,20 +95,18 @@ class OptsList:
class HttpTest(unittest.TestCase):
- def testReadHeader(self):
- status, header, status_message = read_headers(
- HeaderSockMock("data/header01.txt")
- )
+ def test_read_header(self):
+ status, header, _ = read_headers(HeaderSockMock("data/header01.txt"))
self.assertEqual(status, 101)
self.assertEqual(header["connection"], "Upgrade")
# header02.txt is intentionally malformed
self.assertRaises(
- ws.WebSocketException, read_headers, HeaderSockMock("data/header02.txt")
+ WebSocketException, read_headers, HeaderSockMock("data/header02.txt")
)
- def testTunnel(self):
+ def test_tunnel(self):
self.assertRaises(
- ws.WebSocketProxyException,
+ WebSocketProxyException,
_tunnel,
HeaderSockMock("data/header01.txt"),
"example.com",
@@ -115,7 +114,7 @@ class HttpTest(unittest.TestCase):
("username", "password"),
)
self.assertRaises(
- ws.WebSocketProxyException,
+ WebSocketProxyException,
_tunnel,
HeaderSockMock("data/header02.txt"),
"example.com",
@@ -124,9 +123,9 @@ class HttpTest(unittest.TestCase):
)
@unittest.skipUnless(TEST_WITH_INTERNET, "Internet-requiring tests are disabled")
- def testConnect(self):
+ def test_connect(self):
# Not currently testing an actual proxy connection, so just check whether proxy errors are raised. This requires internet for a DNS lookup
- if ws._http.HAVE_PYTHON_SOCKS:
+ if HAVE_PYTHON_SOCKS:
# Need this check, otherwise case where python_socks is not installed triggers
# websocket._exceptions.WebSocketException: Python Socks is needed for SOCKS proxying but is not available
self.assertRaises(
@@ -245,7 +244,7 @@ class HttpTest(unittest.TestCase):
@unittest.skipUnless(
TEST_WITH_LOCAL_SERVER, "Tests using local websocket server are disabled"
)
- def testProxyConnect(self):
+ def test_proxy_connect(self):
ws = websocket.WebSocket()
ws.connect(
f"ws://127.0.0.1:{LOCAL_WS_SERVER_PORT}",
@@ -290,7 +289,7 @@ class HttpTest(unittest.TestCase):
# TODO: Test SOCKS4 and SOCK5 proxies with unit tests
@unittest.skipUnless(TEST_WITH_INTERNET, "Internet-requiring tests are disabled")
- def testSSLopt(self):
+ def test_sslopt(self):
ssloptions = {
"check_hostname": False,
"server_hostname": "ServerName",
@@ -316,7 +315,7 @@ class HttpTest(unittest.TestCase):
ws_ssl2.connect("wss://api.bitfinex.com/ws/2")
ws_ssl2.close
- def testProxyInfo(self):
+ def test_proxy_info(self):
self.assertEqual(
proxy_info(
http_proxy_host="127.0.0.1", http_proxy_port="8080", proxy_type="http"