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 |