Test timeout enhancement

src/Client.php

@@ -181,6 +181,9 @@ protected function handleDisconnection(): void
         );
     }
 
+    /**
+     * @throws TimeoutException|Exception
+     */
     protected function handleClientConnection(): void
     {
         $this->client = new WsClient(
@@ -196,11 +199,19 @@ protected function handleClientConnection(): void
         $this->handleListeners();
         $this->connectionReady();
 
-        while($message = $this->client->receive()) {
-            if (null === $this->onMessageCallback) {
-                continue;
+        try {
+            while ($message = $this->client->receive()) {
+                if (null === $this->onMessageCallback) {
+                    continue;
+                }
+                call_user_func($this->onMessageCallback, $this, $message);
             }
-            call_user_func($this->onMessageCallback, $this, $message);
+        } catch (TimeoutException $e) {
+            if ($this->timeout !== -1) {
+                throw $e;
+            }
+        } catch (Exception $e) {
+            throw $e;
         }
     }