aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/websocket-client/py3/websocket/tests/test_abnf.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_abnf.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_abnf.py')
-rw-r--r--contrib/python/websocket-client/py3/websocket/tests/test_abnf.py98
1 files changed, 67 insertions, 31 deletions
diff --git a/contrib/python/websocket-client/py3/websocket/tests/test_abnf.py b/contrib/python/websocket-client/py3/websocket/tests/test_abnf.py
index dbf9b636a3..df5bc6bd71 100644
--- a/contrib/python/websocket-client/py3/websocket/tests/test_abnf.py
+++ b/contrib/python/websocket-client/py3/websocket/tests/test_abnf.py
@@ -1,8 +1,9 @@
# -*- coding: utf-8 -*-
#
+import unittest
+
import websocket as ws
from websocket._abnf import *
-import unittest
"""
test_abnf.py
@@ -25,54 +26,89 @@ limitations under the License.
class ABNFTest(unittest.TestCase):
-
def testInit(self):
- a = ABNF(0,0,0,0, opcode=ABNF.OPCODE_PING)
+ a = ABNF(0, 0, 0, 0, opcode=ABNF.OPCODE_PING)
self.assertEqual(a.fin, 0)
self.assertEqual(a.rsv1, 0)
self.assertEqual(a.rsv2, 0)
self.assertEqual(a.rsv3, 0)
self.assertEqual(a.opcode, 9)
- self.assertEqual(a.data, '')
- a_bad = ABNF(0,1,0,0, opcode=77)
+ self.assertEqual(a.data, "")
+ a_bad = ABNF(0, 1, 0, 0, opcode=77)
self.assertEqual(a_bad.rsv1, 1)
self.assertEqual(a_bad.opcode, 77)
def testValidate(self):
- a_invalid_ping = ABNF(0,0,0,0, opcode=ABNF.OPCODE_PING)
- self.assertRaises(ws._exceptions.WebSocketProtocolException, a_invalid_ping.validate, skip_utf8_validation=False)
- a_bad_rsv_value = ABNF(0,1,0,0, opcode=ABNF.OPCODE_TEXT)
- self.assertRaises(ws._exceptions.WebSocketProtocolException, a_bad_rsv_value.validate, skip_utf8_validation=False)
- a_bad_opcode = ABNF(0,0,0,0, opcode=77)
- self.assertRaises(ws._exceptions.WebSocketProtocolException, a_bad_opcode.validate, skip_utf8_validation=False)
- a_bad_close_frame = ABNF(0,0,0,0, opcode=ABNF.OPCODE_CLOSE, data=b'\x01')
- self.assertRaises(ws._exceptions.WebSocketProtocolException, a_bad_close_frame.validate, skip_utf8_validation=False)
- a_bad_close_frame_2 = ABNF(0,0,0,0, opcode=ABNF.OPCODE_CLOSE, data=b'\x01\x8a\xaa\xff\xdd')
- self.assertRaises(ws._exceptions.WebSocketProtocolException, a_bad_close_frame_2.validate, skip_utf8_validation=False)
- a_bad_close_frame_3 = ABNF(0,0,0,0, opcode=ABNF.OPCODE_CLOSE, data=b'\x03\xe7')
- self.assertRaises(ws._exceptions.WebSocketProtocolException, a_bad_close_frame_3.validate, skip_utf8_validation=True)
+ a_invalid_ping = ABNF(0, 0, 0, 0, opcode=ABNF.OPCODE_PING)
+ self.assertRaises(
+ ws._exceptions.WebSocketProtocolException,
+ a_invalid_ping.validate,
+ skip_utf8_validation=False,
+ )
+ a_bad_rsv_value = ABNF(0, 1, 0, 0, opcode=ABNF.OPCODE_TEXT)
+ self.assertRaises(
+ ws._exceptions.WebSocketProtocolException,
+ a_bad_rsv_value.validate,
+ skip_utf8_validation=False,
+ )
+ a_bad_opcode = ABNF(0, 0, 0, 0, opcode=77)
+ self.assertRaises(
+ ws._exceptions.WebSocketProtocolException,
+ a_bad_opcode.validate,
+ skip_utf8_validation=False,
+ )
+ a_bad_close_frame = ABNF(0, 0, 0, 0, opcode=ABNF.OPCODE_CLOSE, data=b"\x01")
+ self.assertRaises(
+ ws._exceptions.WebSocketProtocolException,
+ a_bad_close_frame.validate,
+ skip_utf8_validation=False,
+ )
+ a_bad_close_frame_2 = ABNF(
+ 0, 0, 0, 0, opcode=ABNF.OPCODE_CLOSE, data=b"\x01\x8a\xaa\xff\xdd"
+ )
+ self.assertRaises(
+ ws._exceptions.WebSocketProtocolException,
+ a_bad_close_frame_2.validate,
+ skip_utf8_validation=False,
+ )
+ a_bad_close_frame_3 = ABNF(
+ 0, 0, 0, 0, opcode=ABNF.OPCODE_CLOSE, data=b"\x03\xe7"
+ )
+ self.assertRaises(
+ ws._exceptions.WebSocketProtocolException,
+ a_bad_close_frame_3.validate,
+ skip_utf8_validation=True,
+ )
def testMask(self):
- abnf_none_data = ABNF(0,0,0,0, opcode=ABNF.OPCODE_PING, mask=1, data=None)
+ abnf_none_data = ABNF(
+ 0, 0, 0, 0, opcode=ABNF.OPCODE_PING, mask_value=1, data=None
+ )
bytes_val = b"aaaa"
self.assertEqual(abnf_none_data._get_masked(bytes_val), bytes_val)
- abnf_str_data = ABNF(0,0,0,0, opcode=ABNF.OPCODE_PING, mask=1, data="a")
- self.assertEqual(abnf_str_data._get_masked(bytes_val), b'aaaa\x00')
+ abnf_str_data = ABNF(
+ 0, 0, 0, 0, opcode=ABNF.OPCODE_PING, mask_value=1, data="a"
+ )
+ self.assertEqual(abnf_str_data._get_masked(bytes_val), b"aaaa\x00")
def testFormat(self):
- abnf_bad_rsv_bits = ABNF(2,0,0,0, opcode=ABNF.OPCODE_TEXT)
+ abnf_bad_rsv_bits = ABNF(2, 0, 0, 0, opcode=ABNF.OPCODE_TEXT)
self.assertRaises(ValueError, abnf_bad_rsv_bits.format)
- abnf_bad_opcode = ABNF(0,0,0,0, opcode=5)
+ abnf_bad_opcode = ABNF(0, 0, 0, 0, opcode=5)
self.assertRaises(ValueError, abnf_bad_opcode.format)
- abnf_length_10 = ABNF(0,0,0,0, opcode=ABNF.OPCODE_TEXT, data="abcdefghij")
- self.assertEqual(b'\x01', abnf_length_10.format()[0].to_bytes(1, 'big'))
- self.assertEqual(b'\x8a', abnf_length_10.format()[1].to_bytes(1, 'big'))
+ abnf_length_10 = ABNF(0, 0, 0, 0, opcode=ABNF.OPCODE_TEXT, data="abcdefghij")
+ self.assertEqual(b"\x01", abnf_length_10.format()[0].to_bytes(1, "big"))
+ self.assertEqual(b"\x8a", abnf_length_10.format()[1].to_bytes(1, "big"))
self.assertEqual("fin=0 opcode=1 data=abcdefghij", abnf_length_10.__str__())
- abnf_length_20 = ABNF(0,0,0,0, opcode=ABNF.OPCODE_BINARY, data="abcdefghijabcdefghij")
- self.assertEqual(b'\x02', abnf_length_20.format()[0].to_bytes(1, 'big'))
- self.assertEqual(b'\x94', abnf_length_20.format()[1].to_bytes(1, 'big'))
- abnf_no_mask = ABNF(0,0,0,0, opcode=ABNF.OPCODE_TEXT, mask=0, data=b'\x01\x8a\xcc')
- self.assertEqual(b'\x01\x03\x01\x8a\xcc', abnf_no_mask.format())
+ abnf_length_20 = ABNF(
+ 0, 0, 0, 0, opcode=ABNF.OPCODE_BINARY, data="abcdefghijabcdefghij"
+ )
+ self.assertEqual(b"\x02", abnf_length_20.format()[0].to_bytes(1, "big"))
+ self.assertEqual(b"\x94", abnf_length_20.format()[1].to_bytes(1, "big"))
+ abnf_no_mask = ABNF(
+ 0, 0, 0, 0, opcode=ABNF.OPCODE_TEXT, mask_value=0, data=b"\x01\x8a\xcc"
+ )
+ self.assertEqual(b"\x01\x03\x01\x8a\xcc", abnf_no_mask.format())
def testFrameBuffer(self):
fb = frame_buffer(0, True)
@@ -81,7 +117,7 @@ class ABNFTest(unittest.TestCase):
fb.clear
self.assertEqual(fb.header, None)
self.assertEqual(fb.length, None)
- self.assertEqual(fb.mask, None)
+ self.assertEqual(fb.mask_value, None)
self.assertEqual(fb.has_mask(), False)