summaryrefslogtreecommitdiffstats
path: root/dom/websocket/tests/file_websocket_basic_wsh.py
blob: 050e7ccc106ccbe85739674de1e789c48f0783d3 (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
28
29
30
31
from mod_pywebsocket import msgutil


def web_socket_do_extra_handshake(request):
    # must set request.ws_protocol to the selected version from ws_requested_protocols
    request.ws_protocol = request.ws_requested_protocols[0]

    if request.ws_protocol == "error":
        raise ValueError("Error")
    pass


def web_socket_transfer_data(request):
    while True:
        line = msgutil.receive_message(request)
        if line == "protocol":
            msgutil.send_message(request, request.ws_protocol)
            continue

        if line == "resource":
            msgutil.send_message(request, request.ws_resource)
            continue

        if line == "origin":
            msgutil.send_message(request, request.ws_origin)
            continue

        msgutil.send_message(request, line)

        if line == "end":
            return