✨ add exception condition

Ackee-Blockchain · Sep 10, 2024 · 5c2fd55 · 5c2fd55
1 parent 027a408
commit 5c2fd55
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/wake/testing/pytest_plugin_multiprocess.py b/wake/testing/pytest_plugin_multiprocess.py
@@ -27,6 +27,7 @@
 )
 from wake.testing.coverage import CoverageHandler
 from wake.utils.tee import StderrTee, StdoutTee
+import websocket
 
 
 class PytestWakePluginMultiprocess:
@@ -86,6 +87,15 @@ def _exception_handler(
         e: Optional[BaseException],
         tb: Optional[TracebackType],
     ) -> None:
+        ignored_exceptions = (
+            websocket._exceptions.WebSocketConnectionClosedException,
+            ConnectionResetError,
+            BrokenPipeError,
+            KeyboardInterrupt
+        )
+        if e_type in ignored_exceptions:
+            return
+
         self._cleanup_stdio()
         self._exception_handled = True