content/base/test/file_websocket_basic_wsh.py
author Gregory Szorc <gps@mozilla.com>
Wed, 28 Jan 2015 13:37:00 -0800
branchMOBILE130_2012050817_RELBRANCH
changeset 120184 c9527b7c0b9a28a3f1209c9488fb16965ed051cd
parent 72823 58c721a6bb1a0f6b32532e5bf32351d8b0d3e77c
permissions -rw-r--r--
Close old release branch MOBILE130_2012050817_RELBRANCH

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