michael@0: from mod_pywebsocket import common
michael@0: from mod_pywebsocket import msgutil
michael@0: 
michael@0: 
michael@0: def web_socket_do_extra_handshake(request):
michael@0:     pass # Always accept.
michael@0: 
michael@0: 
michael@0: def web_socket_transfer_data(request):
michael@0:     expected_messages = ['Hello, world!', '', all_distinct_bytes()]
michael@0: 
michael@0:     for test_number, expected_message in enumerate(expected_messages):
michael@0:         message = msgutil.receive_message(request)
michael@0:         if type(message) == str and message == expected_message:
michael@0:             msgutil.send_message(request, 'PASS: Message #%d.' % test_number)
michael@0:         else:
michael@0:             msgutil.send_message(request, 'FAIL: Message #%d: Received unexpected message: %r' % (test_number, message))
michael@0: 
michael@0: 
michael@0: def all_distinct_bytes():
michael@0:     return ''.join([chr(i) for i in xrange(256)])