third_party/WebKit/LayoutTests/http/tests/websocket/send-file-blob_wsh.py

   1 from mod_pywebsocket import common
   2 from mod_pywebsocket import msgutil
   3
   4
   5 def _retrieve_frame(stream):
   6     # FIXME: Use better API.
   7     frame = stream._receive_frame_as_frame_object()
   8     for frame_filter in stream._options.incoming_frame_filters:
   9         frame_filter.filter(frame)
  10     return frame
  11
  12
  13 def web_socket_do_extra_handshake(request):
  14     # Disable compression extensions because we handle frame directly
  15     # in this test.
  16     request.ws_extension_processors = []
  17
  18
  19 def web_socket_transfer_data(request):
  20     expected_messages = ['Hello, world!']
  21
  22     for test_number, expected_message in enumerate(expected_messages):
  23         frame = _retrieve_frame(request.ws_stream)
  24         if frame.opcode == common.OPCODE_BINARY and frame.payload == expected_message and frame.fin:
  25             msgutil.send_message(request, 'PASS: Message #%d.' % test_number)
  26         else:
  27             msgutil.send_message(request, 'FAIL: Message #%d: Received unexpected frame: opcode = %r, payload = %r, final = %r' % (test_number, frame.opcode, frame.payload, frame.fin))
  28
  29
  30 def all_distinct_bytes():
  31     return ''.join([chr(i) for i in xrange(256)])