fix: throw when initial websocket connection times out

glzr-io · Sep 11, 2024 · 30ed519 · 30ed519
1 parent b4c265c
commit 30ed519
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/client.ts b/src/client.ts
@@ -372,7 +372,15 @@ export class WmClient {
    * @private
    */
   private async _waitForConnection(): Promise<WebSocket> {
-    if (this._socket && this._socket.readyState === this._socket.OPEN) {
+    if (
+      !this._socket ||
+      this._socket.readyState === this._socket.CLOSED ||
+      this._socket.readyState === this._socket.CLOSING
+    ) {
+      throw new Error('Websocket connection is closed.');
+    }
+
+    if (this._socket.readyState === this._socket.OPEN) {
       return this._socket;
     }