Bug 1840701 - [marionette] TcpTransport:receive needs to return immed…

…iately when socket was closed. r=webdriver-reviewers,jgraham, a=testonly Differential Revision: https://phabricator.services.mozilla.com/D182622
dothq · Jul 7, 2023 · c43ed2f · c43ed2f
1 parent 0bdfc4a
commit c43ed2f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/testing/marionette/client/marionette_driver/transport.py b/testing/marionette/client/marionette_driver/transport.py
@@ -221,7 +221,8 @@ def receive(self, unmarshal=True):
 
                 try:
                     chunk = sock.recv(recv_bytes)
-                except OSError:
+                except socket.timeout:
+                    # Lets handle it with our own timeout check
                     continue
 
                 if not chunk: