content/base/test/websocket_hybi/file_check-binary-messages_wsh.py

Thu, 22 Jan 2015 13:21:57 +0100

author
Michael Schloh von Bennewitz <michael@schloh.com>
date
Thu, 22 Jan 2015 13:21:57 +0100
branch
TOR_BUG_9701
changeset 15
b8a032363ba2
permissions
-rw-r--r--

Incorporate requested changes from Mozilla in review:
https://bugzilla.mozilla.org/show_bug.cgi?id=1123480#c6

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

mercurial