src/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 def web_socket_do_extra_handshake(request):
   6     pass
   7
   8 def web_socket_transfer_data(request):
   9     line = request.ws_stream.receive_message()
  10     if line is None:
  11         return
  12     if line == '-':
  13         data = ''
  14     elif line == '--':
  15         data = 'X'
  16     else:
  17         code, reason = line.split(' ', 1)
  18         data = struct.pack('!H', int(code)) + reason.encode('utf-8')
  19     request.connection.write(stream.create_close_frame(data))
  20     request.server_terminated = True