summaryrefslogtreecommitdiffstats
path: root/dom/websocket/tests/websocket_hybi/file_check-binary-messages_wsh.py
blob: d72ec1f4627e1c6384cd76751d1f4c6824f6690c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import six
from mod_pywebsocket import msgutil


def web_socket_do_extra_handshake(request):
    pass  # Always accept.


def web_socket_transfer_data(request):
    expected_messages = ["Hello, world!", "", all_distinct_bytes()]

    for test_number, expected_message in enumerate(expected_messages):
        expected_message = six.b(expected_message)
        message = msgutil.receive_message(request)
        if message == expected_message:
            msgutil.send_message(request, "PASS: Message #{:d}.".format(test_number))
        else:
            msgutil.send_message(
                request,
                "FAIL: Message #{:d}: Received unexpected message: {!r}".format(
                    test_number, message
                ),
            )


def all_distinct_bytes():
    return "".join([chr(i) for i in range(256)])