diff --git a/src/calibre/srv/tests/web_sockets.py b/src/calibre/srv/tests/web_sockets.py
index 1620fb7b5f..b0a95819bb 100644
--- a/src/calibre/srv/tests/web_sockets.py
+++ b/src/calibre/srv/tests/web_sockets.py
@@ -195,7 +195,12 @@ class WebSocketTest(BaseTest):
                 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 @@ class WebSocketTest(BaseTest):
                 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)