third_party/WebKit/LayoutTests/http/tests/websocket/close-code-and-reason_wsh.py

   1 import struct
   2
   3 from mod_pywebsocket import stream
   4
   5
   6 def web_socket_do_extra_handshake(request):
   7     pass
   8
   9
  10 def web_socket_transfer_data(request):
  11     line = request.ws_stream.receive_message()
  12     if line is None:
  13         return
  14     if line == '-':
  15         data = ''
  16     elif line == '--':
  17         data = 'X'
  18     else:
  19         code, reason = line.split(' ', 1)
  20         data = struct.pack('!H', int(code)) + reason.encode('utf-8')
  21     request.connection.write(stream.create_close_frame(data))
  22     request.server_terminated = True
  23     # Wait for Close frame from client.
  24     request.ws_stream.receive_message()