third_party/WebKit/LayoutTests/http/tests/websocket/check-binary-messages_wsh.py

   1 from mod_pywebsocket import common
   2 from mod_pywebsocket import msgutil
   3
   4
   5 def web_socket_do_extra_handshake(request):
   6     pass  # Always accept.
   7
   8
   9 def web_socket_transfer_data(request):
  10     expected_messages = ['Hello, world!', '', all_distinct_bytes()]
  11
  12     for test_number, expected_message in enumerate(expected_messages):
  13         message = msgutil.receive_message(request)
  14         if type(message) == str and message == expected_message:
  15             msgutil.send_message(request, 'PASS: Message #%d.' % test_number)
  16         else:
  17             msgutil.send_message(request, 'FAIL: Message #%d: Received unexpected message: %r' % (test_number, message))
  18
  19
  20 def all_distinct_bytes():
  21     return ''.join([chr(i) for i in xrange(256)])