third_party/WebKit/LayoutTests/http/tests/websocket/handshake-fail-by-no-connection-header_wsh.py

   1 from mod_pywebsocket import handshake
   2 from mod_pywebsocket.handshake.hybi import compute_accept
   3
   4
   5 def web_socket_do_extra_handshake(request):
   6     msg = 'HTTP/1.1 101 Switching Protocols\r\n'
   7     msg += 'Upgrade: websocket\r\n'
   8 #   Missing 'Connection: Upgrade\r\n'
   9     msg += 'Sec-WebSocket-Accept: %s\r\n' % compute_accept(request.headers_in['Sec-WebSocket-Key'])[0]
  10     msg += '\r\n'
  11     request.connection.write(msg)
  12     print msg
  13     # Prevents pywebsocket from sending its own handshake message.
  14     raise handshake.AbortedByUserException('Abort the connection')
  15
  16
  17 def web_socket_transfer_data(request):
  18     pass