summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/websockets/handlers/receive-many-with-backpressure_wsh.py
blob: 8e35beebdf5027fdcb034eda27ebfb2a4c2dbd4c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# Sleep to build backpressure, receive messages, and send back their length.
# Used by send-many-64K-messages-with-backpressure.any.js.


import time


def web_socket_do_extra_handshake(request):
    # Compression will interfere with backpressure, so disable the
    # permessage-delate extension.
    request.ws_extension_processors = []


def web_socket_transfer_data(request):
    while True:
        # Don't read the message immediately, so backpressure can build.
        time.sleep(0.1)
        line = request.ws_stream.receive_message()
        if line is None:
            return
        # Send back the size of the message as acknowledgement that it was
        # received.
        request.ws_stream.send_message(str(len(line)), binary=False)