11477Szelenkov@nginx.comimport struct
21130Szelenkov@nginx.comimport time
31477Szelenkov@nginx.com
4*1635Szelenkov@nginx.comimport pytest
5*1635Szelenkov@nginx.com
6*1635Szelenkov@nginx.comfrom conftest import option
7*1635Szelenkov@nginx.comfrom conftest import skip_alert
81130Szelenkov@nginx.comfrom unit.applications.lang.node import TestApplicationNode
91130Szelenkov@nginx.comfrom unit.applications.websockets import TestApplicationWebsocket
101130Szelenkov@nginx.com
111150Szelenkov@nginx.com
121130Szelenkov@nginx.comclass TestNodeWebsockets(TestApplicationNode):
131467Szelenkov@nginx.com    prerequisites = {'modules': {'node': 'any'}}
141130Szelenkov@nginx.com
151130Szelenkov@nginx.com    ws = TestApplicationWebsocket()
161130Szelenkov@nginx.com
171596Szelenkov@nginx.com    def setup_method(self):
181596Szelenkov@nginx.com        super().setup_method()
191133Szelenkov@nginx.com
201596Szelenkov@nginx.com        assert 'success' in self.conf(
211596Szelenkov@nginx.com            {'http': {'websocket': {'keepalive_interval': 0}}}, 'settings'
221596Szelenkov@nginx.com        ), 'clear keepalive_interval'
231138Szelenkov@nginx.com
241596Szelenkov@nginx.com        skip_alert(r'socket close\(\d+\) failed')
251130Szelenkov@nginx.com
261130Szelenkov@nginx.com    def close_connection(self, sock):
271596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty soc'
281130Szelenkov@nginx.com
291130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, self.ws.serialize_close())
301130Szelenkov@nginx.com
311130Szelenkov@nginx.com        self.check_close(sock)
321130Szelenkov@nginx.com
331150Szelenkov@nginx.com    def check_close(self, sock, code=1000, no_close=False):
341130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
351130Szelenkov@nginx.com
361596Szelenkov@nginx.com        assert frame['fin'] == True, 'close fin'
371596Szelenkov@nginx.com        assert frame['opcode'] == self.ws.OP_CLOSE, 'close opcode'
381596Szelenkov@nginx.com        assert frame['code'] == code, 'close code'
391130Szelenkov@nginx.com
401130Szelenkov@nginx.com        if not no_close:
411130Szelenkov@nginx.com            sock.close()
421130Szelenkov@nginx.com
431130Szelenkov@nginx.com    def check_frame(self, frame, fin, opcode, payload, decode=True):
441130Szelenkov@nginx.com        if opcode == self.ws.OP_BINARY or not decode:
451130Szelenkov@nginx.com            data = frame['data']
461130Szelenkov@nginx.com        else:
471130Szelenkov@nginx.com            data = frame['data'].decode('utf-8')
481130Szelenkov@nginx.com
491596Szelenkov@nginx.com        assert frame['fin'] == fin, 'fin'
501596Szelenkov@nginx.com        assert frame['opcode'] == opcode, 'opcode'
511596Szelenkov@nginx.com        assert data == payload, 'payload'
521130Szelenkov@nginx.com
531130Szelenkov@nginx.com    def test_node_websockets_handshake(self):
541130Szelenkov@nginx.com        self.load('websockets/mirror')
551130Szelenkov@nginx.com
561130Szelenkov@nginx.com        resp, sock, key = self.ws.upgrade()
571130Szelenkov@nginx.com        sock.close()
581130Szelenkov@nginx.com
591596Szelenkov@nginx.com        assert resp['status'] == 101, 'status'
601596Szelenkov@nginx.com        assert resp['headers']['Upgrade'] == 'websocket', 'upgrade'
611596Szelenkov@nginx.com        assert resp['headers']['Connection'] == 'Upgrade', 'connection'
621596Szelenkov@nginx.com        assert resp['headers']['Sec-WebSocket-Accept'] == self.ws.accept(
631596Szelenkov@nginx.com            key
641596Szelenkov@nginx.com        ), 'key'
651130Szelenkov@nginx.com
661130Szelenkov@nginx.com    def test_node_websockets_mirror(self):
671130Szelenkov@nginx.com        self.load('websockets/mirror')
681130Szelenkov@nginx.com
691130Szelenkov@nginx.com        message = 'blah'
701130Szelenkov@nginx.com
711130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
721130Szelenkov@nginx.com
731130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, message)
741130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
751130Szelenkov@nginx.com
761596Szelenkov@nginx.com        assert message == frame['data'].decode('utf-8'), 'mirror'
771130Szelenkov@nginx.com
781130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, message)
791130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
801130Szelenkov@nginx.com
811596Szelenkov@nginx.com        assert message == frame['data'].decode('utf-8'), 'mirror 2'
821130Szelenkov@nginx.com
831130Szelenkov@nginx.com        sock.close()
841130Szelenkov@nginx.com
851130Szelenkov@nginx.com    def test_node_websockets_no_mask(self):
861130Szelenkov@nginx.com        self.load('websockets/mirror')
871130Szelenkov@nginx.com
881130Szelenkov@nginx.com        message = 'blah'
891130Szelenkov@nginx.com
901130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
911130Szelenkov@nginx.com
921130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, message, mask=False)
931130Szelenkov@nginx.com
941130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
951130Szelenkov@nginx.com
961596Szelenkov@nginx.com        assert frame['opcode'] == self.ws.OP_CLOSE, 'no mask opcode'
971596Szelenkov@nginx.com        assert frame['code'] == 1002, 'no mask close code'
981130Szelenkov@nginx.com
991130Szelenkov@nginx.com        sock.close()
1001130Szelenkov@nginx.com
1011130Szelenkov@nginx.com    def test_node_websockets_fragmentation(self):
1021130Szelenkov@nginx.com        self.load('websockets/mirror')
1031130Szelenkov@nginx.com
1041130Szelenkov@nginx.com        message = 'blah'
1051130Szelenkov@nginx.com
1061130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
1071130Szelenkov@nginx.com
1081130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, message, fin=False)
1091130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, ' ', fin=False)
1101130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, message)
1111130Szelenkov@nginx.com
1121130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
1131130Szelenkov@nginx.com
1141596Szelenkov@nginx.com        assert message + ' ' + message == frame['data'].decode(
1151596Szelenkov@nginx.com            'utf-8'
1161596Szelenkov@nginx.com        ), 'mirror framing'
1171130Szelenkov@nginx.com
1181130Szelenkov@nginx.com        sock.close()
1191130Szelenkov@nginx.com
1201130Szelenkov@nginx.com    def test_node_websockets_frame_fragmentation_invalid(self):
1211130Szelenkov@nginx.com        self.load('websockets/mirror')
1221130Szelenkov@nginx.com
1231130Szelenkov@nginx.com        message = 'blah'
1241130Szelenkov@nginx.com
1251130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
1261130Szelenkov@nginx.com
1271130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, message, fin=False)
1281130Szelenkov@nginx.com
1291130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
1301130Szelenkov@nginx.com
1311130Szelenkov@nginx.com        frame.pop('data')
1321596Szelenkov@nginx.com        assert frame == {
1331596Szelenkov@nginx.com            'fin': True,
1341596Szelenkov@nginx.com            'rsv1': False,
1351596Szelenkov@nginx.com            'rsv2': False,
1361596Szelenkov@nginx.com            'rsv3': False,
1371596Szelenkov@nginx.com            'opcode': self.ws.OP_CLOSE,
1381596Szelenkov@nginx.com            'mask': 0,
1391596Szelenkov@nginx.com            'code': 1002,
1401596Szelenkov@nginx.com            'reason': 'Fragmented control frame',
1411596Szelenkov@nginx.com        }, 'close frame'
1421130Szelenkov@nginx.com
1431130Szelenkov@nginx.com        sock.close()
1441130Szelenkov@nginx.com
1451130Szelenkov@nginx.com    def test_node_websockets_large(self):
1461130Szelenkov@nginx.com        self.load('websockets/mirror_fragmentation')
1471130Szelenkov@nginx.com
1481130Szelenkov@nginx.com        message = '0123456789' * 3000
1491130Szelenkov@nginx.com
1501130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
1511130Szelenkov@nginx.com
1521130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, message)
1531130Szelenkov@nginx.com
1541130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
1551130Szelenkov@nginx.com        data = frame['data'].decode('utf-8')
1561130Szelenkov@nginx.com
1571130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
1581130Szelenkov@nginx.com        data += frame['data'].decode('utf-8')
1591130Szelenkov@nginx.com
1601596Szelenkov@nginx.com        assert message == data, 'large'
1611130Szelenkov@nginx.com
1621130Szelenkov@nginx.com        sock.close()
1631130Szelenkov@nginx.com
1641130Szelenkov@nginx.com    def test_node_websockets_two_clients(self):
1651130Szelenkov@nginx.com        self.load('websockets/mirror')
1661130Szelenkov@nginx.com
1671130Szelenkov@nginx.com        message1 = 'blah1'
1681130Szelenkov@nginx.com        message2 = 'blah2'
1691130Szelenkov@nginx.com
1701130Szelenkov@nginx.com        _, sock1, _ = self.ws.upgrade()
1711130Szelenkov@nginx.com        _, sock2, _ = self.ws.upgrade()
1721130Szelenkov@nginx.com
1731130Szelenkov@nginx.com        self.ws.frame_write(sock1, self.ws.OP_TEXT, message1)
1741130Szelenkov@nginx.com        self.ws.frame_write(sock2, self.ws.OP_TEXT, message2)
1751130Szelenkov@nginx.com
1761130Szelenkov@nginx.com        frame1 = self.ws.frame_read(sock1)
1771130Szelenkov@nginx.com        frame2 = self.ws.frame_read(sock2)
1781130Szelenkov@nginx.com
1791596Szelenkov@nginx.com        assert message1 == frame1['data'].decode('utf-8'), 'client 1'
1801596Szelenkov@nginx.com        assert message2 == frame2['data'].decode('utf-8'), 'client 2'
1811130Szelenkov@nginx.com
1821130Szelenkov@nginx.com        sock1.close()
1831130Szelenkov@nginx.com        sock2.close()
1841130Szelenkov@nginx.com
1851596Szelenkov@nginx.com    @pytest.mark.skip('not yet')
1861150Szelenkov@nginx.com    def test_node_websockets_handshake_upgrade_absent(
1871150Szelenkov@nginx.com        self
1881150Szelenkov@nginx.com    ):  # FAIL https://tools.ietf.org/html/rfc6455#section-4.2.1
1891130Szelenkov@nginx.com        self.load('websockets/mirror')
1901130Szelenkov@nginx.com
1911150Szelenkov@nginx.com        resp = self.get(
1921150Szelenkov@nginx.com            headers={
1931150Szelenkov@nginx.com                'Host': 'localhost',
1941150Szelenkov@nginx.com                'Connection': 'Upgrade',
1951262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
1961150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
1971150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
1981150Szelenkov@nginx.com            },
1991150Szelenkov@nginx.com        )
2001130Szelenkov@nginx.com
2011596Szelenkov@nginx.com        assert resp['status'] == 400, 'upgrade absent'
2021130Szelenkov@nginx.com
2031130Szelenkov@nginx.com    def test_node_websockets_handshake_case_insensitive(self):
2041130Szelenkov@nginx.com        self.load('websockets/mirror')
2051130Szelenkov@nginx.com
2061262Szelenkov@nginx.com        resp, sock, _ = self.ws.upgrade(
2071150Szelenkov@nginx.com            headers={
2081150Szelenkov@nginx.com                'Host': 'localhost',
2091150Szelenkov@nginx.com                'Upgrade': 'WEBSOCKET',
2101150Szelenkov@nginx.com                'Connection': 'UPGRADE',
2111262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
2121150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2131150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
2141262Szelenkov@nginx.com            }
2151150Szelenkov@nginx.com        )
2161262Szelenkov@nginx.com        sock.close()
2171130Szelenkov@nginx.com
2181596Szelenkov@nginx.com        assert resp['status'] == 101, 'status'
2191130Szelenkov@nginx.com
2201596Szelenkov@nginx.com    @pytest.mark.skip('not yet')
2211150Szelenkov@nginx.com    def test_node_websockets_handshake_connection_absent(self):  # FAIL
2221130Szelenkov@nginx.com        self.load('websockets/mirror')
2231130Szelenkov@nginx.com
2241150Szelenkov@nginx.com        resp = self.get(
2251150Szelenkov@nginx.com            headers={
2261150Szelenkov@nginx.com                'Host': 'localhost',
2271150Szelenkov@nginx.com                'Upgrade': 'websocket',
2281262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
2291150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2301150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
2311150Szelenkov@nginx.com            },
2321150Szelenkov@nginx.com        )
2331130Szelenkov@nginx.com
2341596Szelenkov@nginx.com        assert resp['status'] == 400, 'status'
2351130Szelenkov@nginx.com
2361130Szelenkov@nginx.com    def test_node_websockets_handshake_version_absent(self):
2371130Szelenkov@nginx.com        self.load('websockets/mirror')
2381130Szelenkov@nginx.com
2391150Szelenkov@nginx.com        resp = self.get(
2401150Szelenkov@nginx.com            headers={
2411150Szelenkov@nginx.com                'Host': 'localhost',
2421150Szelenkov@nginx.com                'Upgrade': 'websocket',
2431150Szelenkov@nginx.com                'Connection': 'Upgrade',
2441262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
2451150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2461150Szelenkov@nginx.com            },
2471150Szelenkov@nginx.com        )
2481130Szelenkov@nginx.com
2491596Szelenkov@nginx.com        assert resp['status'] == 426, 'status'
2501130Szelenkov@nginx.com
2511596Szelenkov@nginx.com    @pytest.mark.skip('not yet')
2521130Szelenkov@nginx.com    def test_node_websockets_handshake_key_invalid(self):
2531130Szelenkov@nginx.com        self.load('websockets/mirror')
2541130Szelenkov@nginx.com
2551150Szelenkov@nginx.com        resp = self.get(
2561150Szelenkov@nginx.com            headers={
2571150Szelenkov@nginx.com                'Host': 'localhost',
2581150Szelenkov@nginx.com                'Upgrade': 'websocket',
2591150Szelenkov@nginx.com                'Connection': 'Upgrade',
2601150Szelenkov@nginx.com                'Sec-WebSocket-Key': '!',
2611150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2621150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
2631150Szelenkov@nginx.com            },
2641150Szelenkov@nginx.com        )
2651130Szelenkov@nginx.com
2661596Szelenkov@nginx.com        assert resp['status'] == 400, 'key length'
2671130Szelenkov@nginx.com
2681130Szelenkov@nginx.com        key = self.ws.key()
2691150Szelenkov@nginx.com        resp = self.get(
2701150Szelenkov@nginx.com            headers={
2711150Szelenkov@nginx.com                'Host': 'localhost',
2721150Szelenkov@nginx.com                'Upgrade': 'websocket',
2731150Szelenkov@nginx.com                'Connection': 'Upgrade',
2741150Szelenkov@nginx.com                'Sec-WebSocket-Key': [key, key],
2751150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2761150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
2771150Szelenkov@nginx.com            },
2781150Szelenkov@nginx.com        )
2791130Szelenkov@nginx.com
2801596Szelenkov@nginx.com        assert resp['status'] == 400, 'key double'  # FAIL https://tools.ietf.org/html/rfc6455#section-11.3.1
2811130Szelenkov@nginx.com
2821130Szelenkov@nginx.com    def test_node_websockets_handshake_method_invalid(self):
2831130Szelenkov@nginx.com        self.load('websockets/mirror')
2841130Szelenkov@nginx.com
2851150Szelenkov@nginx.com        resp = self.post(
2861150Szelenkov@nginx.com            headers={
2871150Szelenkov@nginx.com                'Host': 'localhost',
2881150Szelenkov@nginx.com                'Upgrade': 'websocket',
2891150Szelenkov@nginx.com                'Connection': 'Upgrade',
2901262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
2911150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2921150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
2931150Szelenkov@nginx.com            },
2941150Szelenkov@nginx.com        )
2951130Szelenkov@nginx.com
2961596Szelenkov@nginx.com        assert resp['status'] == 400, 'status'
2971130Szelenkov@nginx.com
2981130Szelenkov@nginx.com    def test_node_websockets_handshake_http_10(self):
2991130Szelenkov@nginx.com        self.load('websockets/mirror')
3001130Szelenkov@nginx.com
3011150Szelenkov@nginx.com        resp = self.get(
3021150Szelenkov@nginx.com            headers={
3031150Szelenkov@nginx.com                'Host': 'localhost',
3041150Szelenkov@nginx.com                'Upgrade': 'websocket',
3051150Szelenkov@nginx.com                'Connection': 'Upgrade',
3061262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
3071150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
3081150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
3091150Szelenkov@nginx.com            },
3101150Szelenkov@nginx.com            http_10=True,
3111150Szelenkov@nginx.com        )
3121130Szelenkov@nginx.com
3131596Szelenkov@nginx.com        assert resp['status'] == 400, 'status'
3141130Szelenkov@nginx.com
3151130Szelenkov@nginx.com    def test_node_websockets_handshake_uri_invalid(self):
3161130Szelenkov@nginx.com        self.load('websockets/mirror')
3171130Szelenkov@nginx.com
3181150Szelenkov@nginx.com        resp = self.get(
3191150Szelenkov@nginx.com            headers={
3201150Szelenkov@nginx.com                'Host': 'localhost',
3211150Szelenkov@nginx.com                'Upgrade': 'websocket',
3221150Szelenkov@nginx.com                'Connection': 'Upgrade',
3231262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
3241150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
3251150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
3261150Szelenkov@nginx.com            },
3271150Szelenkov@nginx.com            url='!',
3281150Szelenkov@nginx.com        )
3291130Szelenkov@nginx.com
3301596Szelenkov@nginx.com        assert resp['status'] == 400, 'status'
3311130Szelenkov@nginx.com
3321130Szelenkov@nginx.com    def test_node_websockets_protocol_absent(self):
3331130Szelenkov@nginx.com        self.load('websockets/mirror')
3341130Szelenkov@nginx.com
3351130Szelenkov@nginx.com        key = self.ws.key()
3361262Szelenkov@nginx.com        resp, sock, _ = self.ws.upgrade(
3371150Szelenkov@nginx.com            headers={
3381150Szelenkov@nginx.com                'Host': 'localhost',
3391150Szelenkov@nginx.com                'Upgrade': 'websocket',
3401150Szelenkov@nginx.com                'Connection': 'Upgrade',
3411150Szelenkov@nginx.com                'Sec-WebSocket-Key': key,
3421150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
3431262Szelenkov@nginx.com            }
3441150Szelenkov@nginx.com        )
3451262Szelenkov@nginx.com        sock.close()
3461130Szelenkov@nginx.com
3471596Szelenkov@nginx.com        assert resp['status'] == 101, 'status'
3481596Szelenkov@nginx.com        assert resp['headers']['Upgrade'] == 'websocket', 'upgrade'
3491596Szelenkov@nginx.com        assert resp['headers']['Connection'] == 'Upgrade', 'connection'
3501596Szelenkov@nginx.com        assert resp['headers']['Sec-WebSocket-Accept'] == self.ws.accept(
3511596Szelenkov@nginx.com            key
3521596Szelenkov@nginx.com        ), 'key'
3531130Szelenkov@nginx.com
3541130Szelenkov@nginx.com    # autobahn-testsuite
3551150Szelenkov@nginx.com    #
3561130Szelenkov@nginx.com    # Some following tests fail because of Unit does not support UTF-8
3571130Szelenkov@nginx.com    # validation for websocket frames.  It should be implemented
3581130Szelenkov@nginx.com    # by application, if necessary.
3591130Szelenkov@nginx.com
3601130Szelenkov@nginx.com    def test_node_websockets_1_1_1__1_1_8(self):
3611130Szelenkov@nginx.com        self.load('websockets/mirror')
3621130Szelenkov@nginx.com
3631130Szelenkov@nginx.com        opcode = self.ws.OP_TEXT
3641130Szelenkov@nginx.com
3651130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
3661130Szelenkov@nginx.com
3671130Szelenkov@nginx.com        def check_length(length, chopsize=None):
3681130Szelenkov@nginx.com            payload = '*' * length
3691130Szelenkov@nginx.com
3701130Szelenkov@nginx.com            self.ws.frame_write(sock, opcode, payload, chopsize=chopsize)
3711130Szelenkov@nginx.com
3721130Szelenkov@nginx.com            frame = self.ws.frame_read(sock)
3731130Szelenkov@nginx.com            self.check_frame(frame, True, opcode, payload)
3741130Szelenkov@nginx.com
3751130Szelenkov@nginx.com        check_length(0)                      # 1_1_1
3761130Szelenkov@nginx.com        check_length(125)                    # 1_1_2
3771130Szelenkov@nginx.com        check_length(126)                    # 1_1_3
3781130Szelenkov@nginx.com        check_length(127)                    # 1_1_4
3791130Szelenkov@nginx.com        check_length(128)                    # 1_1_5
3801130Szelenkov@nginx.com        check_length(65535)                  # 1_1_6
3811130Szelenkov@nginx.com        check_length(65536)                  # 1_1_7
3821130Szelenkov@nginx.com        check_length(65536, chopsize = 997)  # 1_1_8
3831130Szelenkov@nginx.com
3841130Szelenkov@nginx.com        self.close_connection(sock)
3851130Szelenkov@nginx.com
3861130Szelenkov@nginx.com    def test_node_websockets_1_2_1__1_2_8(self):
3871130Szelenkov@nginx.com        self.load('websockets/mirror')
3881130Szelenkov@nginx.com
3891130Szelenkov@nginx.com        opcode = self.ws.OP_BINARY
3901130Szelenkov@nginx.com
3911130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
3921130Szelenkov@nginx.com
3931130Szelenkov@nginx.com        def check_length(length, chopsize=None):
3941130Szelenkov@nginx.com            payload = b'\xfe' * length
3951130Szelenkov@nginx.com
3961130Szelenkov@nginx.com            self.ws.frame_write(sock, opcode, payload, chopsize=chopsize)
3971130Szelenkov@nginx.com            frame = self.ws.frame_read(sock)
3981130Szelenkov@nginx.com
3991130Szelenkov@nginx.com            self.check_frame(frame, True, opcode, payload)
4001130Szelenkov@nginx.com
4011130Szelenkov@nginx.com        check_length(0)                      # 1_2_1
4021130Szelenkov@nginx.com        check_length(125)                    # 1_2_2
4031130Szelenkov@nginx.com        check_length(126)                    # 1_2_3
4041130Szelenkov@nginx.com        check_length(127)                    # 1_2_4
4051130Szelenkov@nginx.com        check_length(128)                    # 1_2_5
4061130Szelenkov@nginx.com        check_length(65535)                  # 1_2_6
4071130Szelenkov@nginx.com        check_length(65536)                  # 1_2_7
4081130Szelenkov@nginx.com        check_length(65536, chopsize = 997)  # 1_2_8
4091130Szelenkov@nginx.com
4101130Szelenkov@nginx.com        self.close_connection(sock)
4111130Szelenkov@nginx.com
4121130Szelenkov@nginx.com    def test_node_websockets_2_1__2_6(self):
4131130Szelenkov@nginx.com        self.load('websockets/mirror')
4141130Szelenkov@nginx.com
4151130Szelenkov@nginx.com        op_ping = self.ws.OP_PING
4161130Szelenkov@nginx.com        op_pong = self.ws.OP_PONG
4171130Szelenkov@nginx.com
4181130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
4191130Szelenkov@nginx.com
4201130Szelenkov@nginx.com        def check_ping(payload, chopsize=None, decode=True):
4211130Szelenkov@nginx.com            self.ws.frame_write(sock, op_ping, payload, chopsize=chopsize)
4221130Szelenkov@nginx.com            frame = self.ws.frame_read(sock)
4231130Szelenkov@nginx.com
4241130Szelenkov@nginx.com            self.check_frame(frame, True, op_pong, payload, decode=decode)
4251130Szelenkov@nginx.com
4261130Szelenkov@nginx.com        check_ping('')                                                 # 2_1
4271130Szelenkov@nginx.com        check_ping('Hello, world!')                                    # 2_2
4281130Szelenkov@nginx.com        check_ping(b'\x00\xff\xfe\xfd\xfc\xfb\x00\xff', decode=False)  # 2_3
4291130Szelenkov@nginx.com        check_ping(b'\xfe' * 125, decode=False)                        # 2_4
4301130Szelenkov@nginx.com        check_ping(b'\xfe' * 125, chopsize=1, decode=False)            # 2_6
4311130Szelenkov@nginx.com
4321130Szelenkov@nginx.com        self.close_connection(sock)
4331130Szelenkov@nginx.com
4341130Szelenkov@nginx.com        # 2_5
4351130Szelenkov@nginx.com
4361130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
4371130Szelenkov@nginx.com
4381130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, b'\xfe' * 126)
4391130Szelenkov@nginx.com        self.check_close(sock, 1002)
4401130Szelenkov@nginx.com
4411130Szelenkov@nginx.com    def test_node_websockets_2_7__2_9(self):
4421130Szelenkov@nginx.com        self.load('websockets/mirror')
4431130Szelenkov@nginx.com
4441130Szelenkov@nginx.com        # 2_7
4451130Szelenkov@nginx.com
4461130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
4471130Szelenkov@nginx.com
4481130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PONG, '')
4491596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', '2_7'
4501130Szelenkov@nginx.com
4511130Szelenkov@nginx.com        # 2_8
4521130Szelenkov@nginx.com
4531130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PONG, 'unsolicited pong payload')
4541596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', '2_8'
4551130Szelenkov@nginx.com
4561130Szelenkov@nginx.com        # 2_9
4571130Szelenkov@nginx.com
4581130Szelenkov@nginx.com        payload = 'ping payload'
4591130Szelenkov@nginx.com
4601130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PONG, 'unsolicited pong payload')
4611130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, payload)
4621130Szelenkov@nginx.com
4631130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
4641130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, payload)
4651130Szelenkov@nginx.com
4661130Szelenkov@nginx.com        self.close_connection(sock)
4671130Szelenkov@nginx.com
4681130Szelenkov@nginx.com    def test_node_websockets_2_10__2_11(self):
4691130Szelenkov@nginx.com        self.load('websockets/mirror')
4701130Szelenkov@nginx.com
4711130Szelenkov@nginx.com        # 2_10
4721130Szelenkov@nginx.com
4731130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
4741130Szelenkov@nginx.com
4751130Szelenkov@nginx.com        for i in range(0, 10):
4761130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_PING, 'payload-%d' % i)
4771130Szelenkov@nginx.com
4781130Szelenkov@nginx.com        for i in range(0, 10):
4791130Szelenkov@nginx.com            frame = self.ws.frame_read(sock)
4801130Szelenkov@nginx.com            self.check_frame(frame, True, self.ws.OP_PONG, 'payload-%d' % i)
4811130Szelenkov@nginx.com
4821130Szelenkov@nginx.com        # 2_11
4831130Szelenkov@nginx.com
4841130Szelenkov@nginx.com        for i in range(0, 10):
4851130Szelenkov@nginx.com            opcode = self.ws.OP_PING
4861130Szelenkov@nginx.com            self.ws.frame_write(sock, opcode, 'payload-%d' % i, chopsize=1)
4871130Szelenkov@nginx.com
4881130Szelenkov@nginx.com        for i in range(0, 10):
4891130Szelenkov@nginx.com            frame = self.ws.frame_read(sock)
4901130Szelenkov@nginx.com            self.check_frame(frame, True, self.ws.OP_PONG, 'payload-%d' % i)
4911130Szelenkov@nginx.com
4921130Szelenkov@nginx.com        self.close_connection(sock)
4931130Szelenkov@nginx.com
4941596Szelenkov@nginx.com    @pytest.mark.skip('not yet')
4951130Szelenkov@nginx.com    def test_node_websockets_3_1__3_7(self):
4961130Szelenkov@nginx.com        self.load('websockets/mirror')
4971130Szelenkov@nginx.com
4981130Szelenkov@nginx.com        payload = 'Hello, world!'
4991130Szelenkov@nginx.com
5001130Szelenkov@nginx.com        # 3_1
5011130Szelenkov@nginx.com
5021130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5031130Szelenkov@nginx.com
5041130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload, rsv1=True)
5051130Szelenkov@nginx.com        self.check_close(sock, 1002)
5061130Szelenkov@nginx.com
5071130Szelenkov@nginx.com        # 3_2
5081130Szelenkov@nginx.com
5091130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5101130Szelenkov@nginx.com
5111130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
5121130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload, rsv2=True)
5131130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
5141130Szelenkov@nginx.com
5151130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
5161130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
5171130Szelenkov@nginx.com
5181150Szelenkov@nginx.com        self.check_close(sock, 1002, no_close=True)
5191130Szelenkov@nginx.com
5201596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty 3_2'
5211130Szelenkov@nginx.com        sock.close()
5221130Szelenkov@nginx.com
5231130Szelenkov@nginx.com        # 3_3
5241130Szelenkov@nginx.com
5251130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5261130Szelenkov@nginx.com
5271130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
5281130Szelenkov@nginx.com
5291130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
5301130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
5311130Szelenkov@nginx.com
5321130Szelenkov@nginx.com        self.ws.frame_write(
5331150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, payload, rsv1=True, rsv2=True
5341130Szelenkov@nginx.com        )
5351130Szelenkov@nginx.com
5361150Szelenkov@nginx.com        self.check_close(sock, 1002, no_close=True)
5371130Szelenkov@nginx.com
5381596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty 3_3'
5391130Szelenkov@nginx.com        sock.close()
5401130Szelenkov@nginx.com
5411130Szelenkov@nginx.com        # 3_4
5421130Szelenkov@nginx.com
5431130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5441130Szelenkov@nginx.com
5451130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload, chopsize=1)
5461130Szelenkov@nginx.com        self.ws.frame_write(
5471150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, payload, rsv3=True, chopsize=1
5481130Szelenkov@nginx.com        )
5491130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
5501130Szelenkov@nginx.com
5511130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
5521130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
5531130Szelenkov@nginx.com
5541150Szelenkov@nginx.com        self.check_close(sock, 1002, no_close=True)
5551130Szelenkov@nginx.com
5561596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty 3_4'
5571130Szelenkov@nginx.com        sock.close()
5581130Szelenkov@nginx.com
5591130Szelenkov@nginx.com        # 3_5
5601130Szelenkov@nginx.com
5611130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5621130Szelenkov@nginx.com
5631130Szelenkov@nginx.com        self.ws.frame_write(
5641130Szelenkov@nginx.com            sock,
5651130Szelenkov@nginx.com            self.ws.OP_BINARY,
5661130Szelenkov@nginx.com            b'\x00\xff\xfe\xfd\xfc\xfb\x00\xff',
5671130Szelenkov@nginx.com            rsv1=True,
5681130Szelenkov@nginx.com            rsv3=True,
5691130Szelenkov@nginx.com        )
5701130Szelenkov@nginx.com
5711130Szelenkov@nginx.com        self.check_close(sock, 1002)
5721130Szelenkov@nginx.com
5731130Szelenkov@nginx.com        # 3_6
5741130Szelenkov@nginx.com
5751130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5761130Szelenkov@nginx.com
5771130Szelenkov@nginx.com        self.ws.frame_write(
5781150Szelenkov@nginx.com            sock, self.ws.OP_PING, payload, rsv2=True, rsv3=True
5791130Szelenkov@nginx.com        )
5801130Szelenkov@nginx.com
5811130Szelenkov@nginx.com        self.check_close(sock, 1002)
5821130Szelenkov@nginx.com
5831130Szelenkov@nginx.com        # 3_7
5841130Szelenkov@nginx.com
5851130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5861130Szelenkov@nginx.com
5871130Szelenkov@nginx.com        self.ws.frame_write(
5881150Szelenkov@nginx.com            sock, self.ws.OP_CLOSE, payload, rsv1=True, rsv2=True, rsv3=True
5891130Szelenkov@nginx.com        )
5901130Szelenkov@nginx.com
5911130Szelenkov@nginx.com        self.check_close(sock, 1002)
5921130Szelenkov@nginx.com
5931130Szelenkov@nginx.com    def test_node_websockets_4_1_1__4_2_5(self):
5941130Szelenkov@nginx.com        self.load('websockets/mirror')
5951130Szelenkov@nginx.com
5961130Szelenkov@nginx.com        payload = 'Hello, world!'
5971130Szelenkov@nginx.com
5981130Szelenkov@nginx.com        # 4_1_1
5991130Szelenkov@nginx.com
6001130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6011130Szelenkov@nginx.com
6021130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x03, '')
6031130Szelenkov@nginx.com        self.check_close(sock, 1002)
6041130Szelenkov@nginx.com
6051130Szelenkov@nginx.com        # 4_1_2
6061130Szelenkov@nginx.com
6071130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6081130Szelenkov@nginx.com
6091130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x04, 'reserved opcode payload')
6101130Szelenkov@nginx.com        self.check_close(sock, 1002)
6111130Szelenkov@nginx.com
6121130Szelenkov@nginx.com        # 4_1_3
6131130Szelenkov@nginx.com
6141130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6151130Szelenkov@nginx.com
6161130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
6171130Szelenkov@nginx.com
6181130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
6191130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
6201130Szelenkov@nginx.com
6211130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x05, '')
6221130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
6231130Szelenkov@nginx.com
6241130Szelenkov@nginx.com        self.check_close(sock, 1002)
6251130Szelenkov@nginx.com
6261130Szelenkov@nginx.com        # 4_1_4
6271130Szelenkov@nginx.com
6281130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6291130Szelenkov@nginx.com
6301130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
6311130Szelenkov@nginx.com
6321130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
6331130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
6341130Szelenkov@nginx.com
6351130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x06, payload)
6361130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
6371130Szelenkov@nginx.com
6381130Szelenkov@nginx.com        self.check_close(sock, 1002)
6391130Szelenkov@nginx.com
6401130Szelenkov@nginx.com        # 4_1_5
6411130Szelenkov@nginx.com
6421130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6431130Szelenkov@nginx.com
6441130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload, chopsize=1)
6451130Szelenkov@nginx.com
6461130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
6471130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
6481130Szelenkov@nginx.com
6491130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x07, payload, chopsize=1)
6501130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
6511130Szelenkov@nginx.com
6521130Szelenkov@nginx.com        self.check_close(sock, 1002)
6531130Szelenkov@nginx.com
6541130Szelenkov@nginx.com        # 4_2_1
6551130Szelenkov@nginx.com
6561130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6571130Szelenkov@nginx.com
6581130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x0B, '')
6591130Szelenkov@nginx.com        self.check_close(sock, 1002)
6601130Szelenkov@nginx.com
6611130Szelenkov@nginx.com        # 4_2_2
6621130Szelenkov@nginx.com
6631130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6641130Szelenkov@nginx.com
6651130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x0C, 'reserved opcode payload')
6661130Szelenkov@nginx.com        self.check_close(sock, 1002)
6671130Szelenkov@nginx.com
6681130Szelenkov@nginx.com        # 4_2_3
6691130Szelenkov@nginx.com
6701130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6711130Szelenkov@nginx.com
6721130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
6731130Szelenkov@nginx.com
6741130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
6751130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
6761130Szelenkov@nginx.com
6771130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x0D, '')
6781130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
6791130Szelenkov@nginx.com
6801130Szelenkov@nginx.com        self.check_close(sock, 1002)
6811130Szelenkov@nginx.com
6821130Szelenkov@nginx.com        # 4_2_4
6831130Szelenkov@nginx.com
6841130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6851130Szelenkov@nginx.com
6861130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
6871130Szelenkov@nginx.com
6881130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
6891130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
6901130Szelenkov@nginx.com
6911130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x0E, payload)
6921130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
6931130Szelenkov@nginx.com
6941130Szelenkov@nginx.com        self.check_close(sock, 1002)
6951130Szelenkov@nginx.com
6961130Szelenkov@nginx.com        # 4_2_5
6971130Szelenkov@nginx.com
6981130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6991130Szelenkov@nginx.com
7001130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload, chopsize=1)
7011130Szelenkov@nginx.com
7021130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7031130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
7041130Szelenkov@nginx.com
7051130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x0F, payload, chopsize=1)
7061130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
7071130Szelenkov@nginx.com
7081130Szelenkov@nginx.com        self.check_close(sock, 1002)
7091130Szelenkov@nginx.com
7101130Szelenkov@nginx.com    def test_node_websockets_5_1__5_20(self):
7111130Szelenkov@nginx.com        self.load('websockets/mirror')
7121130Szelenkov@nginx.com
7131130Szelenkov@nginx.com        # 5_1
7141130Szelenkov@nginx.com
7151130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
7161130Szelenkov@nginx.com
7171130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, 'fragment1', fin=False)
7181130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7191130Szelenkov@nginx.com        self.check_close(sock, 1002)
7201130Szelenkov@nginx.com
7211130Szelenkov@nginx.com        # 5_2
7221130Szelenkov@nginx.com
7231130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
7241130Szelenkov@nginx.com
7251130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PONG, 'fragment1', fin=False)
7261130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7271130Szelenkov@nginx.com        self.check_close(sock, 1002)
7281130Szelenkov@nginx.com
7291130Szelenkov@nginx.com        # 5_3
7301130Szelenkov@nginx.com
7311130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
7321130Szelenkov@nginx.com
7331130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
7341130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7351130Szelenkov@nginx.com
7361130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7371130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
7381130Szelenkov@nginx.com
7391130Szelenkov@nginx.com        # 5_4
7401130Szelenkov@nginx.com
7411130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
7421596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', '5_4'
7431130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7441130Szelenkov@nginx.com
7451130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7461130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
7471130Szelenkov@nginx.com
7481130Szelenkov@nginx.com        # 5_5
7491130Szelenkov@nginx.com
7501130Szelenkov@nginx.com        self.ws.frame_write(
7511150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, 'fragment1', fin=False, chopsize=1
7521130Szelenkov@nginx.com        )
7531130Szelenkov@nginx.com        self.ws.frame_write(
7541150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'fragment2', fin=True, chopsize=1
7551130Szelenkov@nginx.com        )
7561130Szelenkov@nginx.com
7571130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7581130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
7591130Szelenkov@nginx.com
7601130Szelenkov@nginx.com        # 5_6
7611130Szelenkov@nginx.com
7621130Szelenkov@nginx.com        ping_payload = 'ping payload'
7631130Szelenkov@nginx.com
7641130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
7651130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, ping_payload)
7661130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7671130Szelenkov@nginx.com
7681130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7691130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, ping_payload)
7701130Szelenkov@nginx.com
7711130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7721130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
7731130Szelenkov@nginx.com
7741130Szelenkov@nginx.com        # 5_7
7751130Szelenkov@nginx.com
7761130Szelenkov@nginx.com        ping_payload = 'ping payload'
7771130Szelenkov@nginx.com
7781130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
7791596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', '5_7'
7801130Szelenkov@nginx.com
7811130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, ping_payload)
7821130Szelenkov@nginx.com
7831130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7841130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, ping_payload)
7851130Szelenkov@nginx.com
7861130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7871130Szelenkov@nginx.com
7881130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7891130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
7901130Szelenkov@nginx.com
7911130Szelenkov@nginx.com        # 5_8
7921130Szelenkov@nginx.com
7931130Szelenkov@nginx.com        ping_payload = 'ping payload'
7941130Szelenkov@nginx.com
7951130Szelenkov@nginx.com        self.ws.frame_write(
7961150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, 'fragment1', fin=False, chopsize=1
7971130Szelenkov@nginx.com        )
7981130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, ping_payload, chopsize=1)
7991130Szelenkov@nginx.com        self.ws.frame_write(
8001150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'fragment2', fin=True, chopsize=1
8011130Szelenkov@nginx.com        )
8021130Szelenkov@nginx.com
8031130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
8041130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, ping_payload)
8051130Szelenkov@nginx.com
8061130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
8071130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
8081130Szelenkov@nginx.com
8091130Szelenkov@nginx.com        # 5_9
8101130Szelenkov@nginx.com
8111130Szelenkov@nginx.com        self.ws.frame_write(
8121150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'non-continuation payload', fin=True
8131130Szelenkov@nginx.com        )
8141130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'Hello, world!', fin=True)
8151130Szelenkov@nginx.com        self.check_close(sock, 1002)
8161130Szelenkov@nginx.com
8171130Szelenkov@nginx.com        # 5_10
8181130Szelenkov@nginx.com
8191130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8201130Szelenkov@nginx.com
8211130Szelenkov@nginx.com        self.ws.frame_write(
8221150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'non-continuation payload', fin=True
8231130Szelenkov@nginx.com        )
8241130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'Hello, world!', fin=True)
8251130Szelenkov@nginx.com        self.check_close(sock, 1002)
8261130Szelenkov@nginx.com
8271130Szelenkov@nginx.com        # 5_11
8281130Szelenkov@nginx.com
8291130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8301130Szelenkov@nginx.com
8311130Szelenkov@nginx.com        self.ws.frame_write(
8321130Szelenkov@nginx.com            sock,
8331130Szelenkov@nginx.com            self.ws.OP_CONT,
8341130Szelenkov@nginx.com            'non-continuation payload',
8351130Szelenkov@nginx.com            fin=True,
8361130Szelenkov@nginx.com            chopsize=1,
8371130Szelenkov@nginx.com        )
8381130Szelenkov@nginx.com        self.ws.frame_write(
8391150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, 'Hello, world!', fin=True, chopsize=1
8401130Szelenkov@nginx.com        )
8411130Szelenkov@nginx.com        self.check_close(sock, 1002)
8421130Szelenkov@nginx.com
8431130Szelenkov@nginx.com        # 5_12
8441130Szelenkov@nginx.com
8451130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8461130Szelenkov@nginx.com
8471130Szelenkov@nginx.com        self.ws.frame_write(
8481150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'non-continuation payload', fin=False
8491130Szelenkov@nginx.com        )
8501130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'Hello, world!', fin=True)
8511130Szelenkov@nginx.com        self.check_close(sock, 1002)
8521130Szelenkov@nginx.com
8531130Szelenkov@nginx.com        # 5_13
8541130Szelenkov@nginx.com
8551130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8561130Szelenkov@nginx.com
8571130Szelenkov@nginx.com        self.ws.frame_write(
8581150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'non-continuation payload', fin=False
8591130Szelenkov@nginx.com        )
8601130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'Hello, world!', fin=True)
8611130Szelenkov@nginx.com        self.check_close(sock, 1002)
8621130Szelenkov@nginx.com
8631130Szelenkov@nginx.com        # 5_14
8641130Szelenkov@nginx.com
8651130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8661130Szelenkov@nginx.com
8671130Szelenkov@nginx.com        self.ws.frame_write(
8681130Szelenkov@nginx.com            sock,
8691130Szelenkov@nginx.com            self.ws.OP_CONT,
8701130Szelenkov@nginx.com            'non-continuation payload',
8711130Szelenkov@nginx.com            fin=False,
8721130Szelenkov@nginx.com            chopsize=1,
8731130Szelenkov@nginx.com        )
8741130Szelenkov@nginx.com        self.ws.frame_write(
8751150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, 'Hello, world!', fin=True, chopsize=1
8761130Szelenkov@nginx.com        )
8771130Szelenkov@nginx.com        self.check_close(sock, 1002)
8781130Szelenkov@nginx.com
8791130Szelenkov@nginx.com        # 5_15
8801130Szelenkov@nginx.com
8811130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8821130Szelenkov@nginx.com
8831130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
8841130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
8851130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment3', fin=False)
8861130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment4', fin=True)
8871130Szelenkov@nginx.com        self.check_close(sock, 1002)
8881130Szelenkov@nginx.com
8891130Szelenkov@nginx.com        # 5_16
8901130Szelenkov@nginx.com
8911130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8921130Szelenkov@nginx.com
8931130Szelenkov@nginx.com        for i in range(0, 2):
8941130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment1', fin=False)
8951130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment2', fin=False)
8961130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment3', fin=True)
8971130Szelenkov@nginx.com        self.check_close(sock, 1002)
8981130Szelenkov@nginx.com
8991130Szelenkov@nginx.com        # 5_17
9001130Szelenkov@nginx.com
9011130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
9021130Szelenkov@nginx.com
9031130Szelenkov@nginx.com        for i in range(0, 2):
9041130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment1', fin=True)
9051130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment2', fin=False)
9061130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment3', fin=True)
9071130Szelenkov@nginx.com        self.check_close(sock, 1002)
9081130Szelenkov@nginx.com
9091130Szelenkov@nginx.com        # 5_18
9101130Szelenkov@nginx.com
9111130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
9121130Szelenkov@nginx.com
9131130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
9141130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment2')
9151130Szelenkov@nginx.com        self.check_close(sock, 1002)
9161130Szelenkov@nginx.com
9171130Szelenkov@nginx.com        # 5_19
9181130Szelenkov@nginx.com
9191130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
9201130Szelenkov@nginx.com
9211130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
9221130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=False)
9231130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, 'pongme 1!')
9241130Szelenkov@nginx.com
9251130Szelenkov@nginx.com        time.sleep(1)
9261130Szelenkov@nginx.com
9271130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment3', fin=False)
9281130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment4', fin=False)
9291130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, 'pongme 2!')
9301130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment5')
9311130Szelenkov@nginx.com
9321130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9331130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, 'pongme 1!')
9341130Szelenkov@nginx.com
9351130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9361130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, 'pongme 2!')
9371130Szelenkov@nginx.com
9381130Szelenkov@nginx.com        self.check_frame(
9391130Szelenkov@nginx.com            self.ws.frame_read(sock),
9401130Szelenkov@nginx.com            True,
9411130Szelenkov@nginx.com            self.ws.OP_TEXT,
9421130Szelenkov@nginx.com            'fragment1fragment2fragment3fragment4fragment5',
9431130Szelenkov@nginx.com        )
9441130Szelenkov@nginx.com
9451130Szelenkov@nginx.com        # 5_20
9461130Szelenkov@nginx.com
9471130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
9481130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=False)
9491130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, 'pongme 1!')
9501130Szelenkov@nginx.com
9511130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9521130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, 'pongme 1!')
9531130Szelenkov@nginx.com
9541130Szelenkov@nginx.com        time.sleep(1)
9551130Szelenkov@nginx.com
9561130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment3', fin=False)
9571130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment4', fin=False)
9581130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, 'pongme 2!')
9591130Szelenkov@nginx.com
9601130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9611130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, 'pongme 2!')
9621130Szelenkov@nginx.com
9631596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', '5_20'
9641130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment5')
9651130Szelenkov@nginx.com
9661130Szelenkov@nginx.com        self.check_frame(
9671130Szelenkov@nginx.com            self.ws.frame_read(sock),
9681130Szelenkov@nginx.com            True,
9691130Szelenkov@nginx.com            self.ws.OP_TEXT,
9701130Szelenkov@nginx.com            'fragment1fragment2fragment3fragment4fragment5',
9711130Szelenkov@nginx.com        )
9721130Szelenkov@nginx.com
9731130Szelenkov@nginx.com        self.close_connection(sock)
9741130Szelenkov@nginx.com
9751130Szelenkov@nginx.com    def test_node_websockets_6_1_1__6_4_4(self):
9761130Szelenkov@nginx.com        self.load('websockets/mirror')
9771130Szelenkov@nginx.com
9781130Szelenkov@nginx.com        # 6_1_1
9791130Szelenkov@nginx.com
9801130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
9811130Szelenkov@nginx.com
9821130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, '')
9831130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9841130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, '')
9851130Szelenkov@nginx.com
9861130Szelenkov@nginx.com        # 6_1_2
9871130Szelenkov@nginx.com
9881130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, '', fin=False)
9891130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, '', fin=False)
9901130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, '')
9911130Szelenkov@nginx.com
9921130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9931130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, '')
9941130Szelenkov@nginx.com
9951130Szelenkov@nginx.com        # 6_1_3
9961130Szelenkov@nginx.com
9971130Szelenkov@nginx.com        payload = 'middle frame payload'
9981130Szelenkov@nginx.com
9991130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, '', fin=False)
10001130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, payload, fin=False)
10011130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, '')
10021130Szelenkov@nginx.com
10031130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10041130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10051130Szelenkov@nginx.com
10061130Szelenkov@nginx.com        # 6_2_1
10071130Szelenkov@nginx.com
10081130Szelenkov@nginx.com        payload = 'Hello-µ@ßöäüàá-UTF-8!!'
10091130Szelenkov@nginx.com
10101130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
10111130Szelenkov@nginx.com
10121130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10131130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10141130Szelenkov@nginx.com
10151130Szelenkov@nginx.com        # 6_2_2
10161130Szelenkov@nginx.com
10171130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload[:12], fin=False)
10181130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, payload[12:])
10191130Szelenkov@nginx.com
10201130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10211130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10221130Szelenkov@nginx.com
10231130Szelenkov@nginx.com        # 6_2_3
10241130Szelenkov@nginx.com
10251130Szelenkov@nginx.com        self.ws.message(sock, self.ws.OP_TEXT, payload, fragmention_size=1)
10261130Szelenkov@nginx.com
10271130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10281130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10291130Szelenkov@nginx.com
10301130Szelenkov@nginx.com        # 6_2_4
10311130Szelenkov@nginx.com
10321130Szelenkov@nginx.com        payload = '\xce\xba\xe1\xbd\xb9\xcf\x83\xce\xbc\xce\xb5'
10331130Szelenkov@nginx.com
10341130Szelenkov@nginx.com        self.ws.message(sock, self.ws.OP_TEXT, payload, fragmention_size=1)
10351130Szelenkov@nginx.com
10361130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10371130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10381130Szelenkov@nginx.com
10391130Szelenkov@nginx.com        self.close_connection(sock)
1040