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