third_party/WebKit/LayoutTests/http/tests/websocket/count-received-bytes_wsh.py

   1 # Copyright 2014 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5
   6 import codecs
   7
   8
   9 def web_socket_do_extra_handshake(request):
  10     request.ws_extension_processors = []
  11     request.received_bytes = 0
  12
  13
  14 def web_socket_transfer_data(request):
  15     while True:
  16         line = request.ws_stream.receive_message()
  17         if line is None:
  18             return
  19         if isinstance(line, unicode):
  20             request.received_bytes += len(codecs.encode(line, 'utf-8'))
  21         else:
  22             request.received_bytes += len(line)
  23
  24
  25 def web_socket_passive_closing_handshake(request):
  26     return 1000, 'received %d bytes' % request.received_bytes