fix: OBS idle timeout, payload validation, workspace list

1 files changed, 22 insertions, 0 deletions

diff --git a/tests/test_obsclient.py b/tests/test_obsclient.py
index 7d1a5ca..54eadd1 100644
--- a/tests/test_obsclient.py
+++ b/tests/test_obsclient.py
@@ -25,5 +25,27 @@ class Disconnected(unittest.TestCase):
             oc.ObsClient(host="127.0.0.1", port=4455, password="").request("GetVersion")
 
 
+class RecvLoop(unittest.TestCase):
+    def test_a_recv_timeout_does_not_end_the_loop(self):
+        import websocket
+
+        class FakeWS:
+            calls = 0
+
+            def recv(self):
+                FakeWS.calls += 1
+                if FakeWS.calls <= 3:
+                    raise websocket.WebSocketTimeoutException("idle")
+                raise RuntimeError("closed")
+
+        c = oc.ObsClient()
+        c.ws = FakeWS()
+        c.connected = True
+        c._recv_loop()
+        # It kept calling recv past three idle timeouts; a timeout must not
+        # look like a closed socket.
+        self.assertEqual(FakeWS.calls, 4)
+
+
 if __name__ == "__main__":
     unittest.main()