Backup_Repos/calibre
1
0
Fork 0
mirror of git://github.com/kovidgoyal/calibre.git synced 2026-05-08 23:03:26 +02:00
Code Issues Projects Releases Wiki Activity

Dont fail the websocket test when sending garbage and the connection is unexpectedly closed by the OS

Browse source
This commit is contained in:
Kovid Goyal 2019-12-05 20:28:49 +05:30
parent 0f852ee6f3
commit 881d2a4ae9
No known key found for this signature in database
GPG key ID: 06BC317B515ACE7C
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/calibre/srv/tests/web_sockets.py
View file

@ -195,7 +195,12 @@ def simple_test(self, server, msgs, expected=(), close_code=NORMAL_CLOSE, send_c
expected_messages.append(ex)
if send_close:
client.write_close(close_code, close_reason)
messages, control_frames = client.read_messages()
try:
messages, control_frames = client.read_messages()
except ConnectionAbortedError:
if expected_messages or expected_controls or send_close:
raise
return
self.ae(expected_messages, messages)
self.assertGreaterEqual(len(control_frames), 1)
self.ae(expected_controls, control_frames[:-1])
@ -317,3 +322,8 @@ def test_websocket_perf(self):
sz *= 1024
t, b = 'a'*sz, b'a'*sz
simple_test([t, b], [t, b])
def find_tests():
import unittest
return unittest.defaultTestLoader.loadTestsFromTestCase(WebSocketTest)