third_party/WebKit/LayoutTests/http/tests/websocket/handshake-fail-by-extensions-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     message = 'HTTP/1.1 101 Switching Protocols\r\n'
   7     message += 'Upgrade: websocket\r\n'
   8     message += 'Connection: Upgrade\r\n'
   9     message += 'Sec-WebSocket-Accept: %s\r\n' % compute_accept(request.headers_in['Sec-WebSocket-Key'])[0]
  10     message += 'Sec-WebSocket-Extensions: x-foo\r\n'
  11     message += '\r\n'
  12     request.connection.write(message)
  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