xref: /aosp_15_r20/external/cronet/net/data/websocket/protocol-test_wsh.py (revision 6777b5387eb2ff775bb5750e3f5d96f37fb7352b)
1# Copyright 2012 The Chromium Authors
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5from mod_pywebsocket import msgutil
6from six.moves.urllib import parse
7
8
9def web_socket_do_extra_handshake(request):
10  r = request.ws_resource.split('?', 1)
11  if len(r) == 1:
12    return
13  param = parse.parse_qs(r[1])
14  if 'protocol' in param:
15    request.ws_protocol = param['protocol'][0]
16
17
18def web_socket_transfer_data(request):
19  msgutil.send_message(request, request.ws_protocol)
20  # Wait for a close message.
21  unused = request.ws_stream.receive_message()
22