fix: Add changeUrl method to AutoReconnectingWebsocket

package.json

@@ -40,7 +40,8 @@
     "semantic-release": "^24.0.0",
     "tsup": "^8.1.0",
     "typescript": "^5.5.3",
-    "vitest": "^2.0.2"
+    "vitest": "^2.0.2",
+    "ws": "^8.18.0"
   },
   "dependencies": {
     "@wandelbots/wandelbots-api-client": "^0.97.0",

src/lib/AutoReconnectingWebsocket.ts

@@ -13,10 +13,6 @@ export class AutoReconnectingWebsocket extends ReconnectingWebSocket {
       startClosed: true,
     })
 
-    // reconnecting-websocket doesn't set this properly with startClosed
-    // so we do it ourselves
-    Object.defineProperty(this, "url", { value: url })
-
     this.addEventListener("open", () => {
       console.log(`Websocket to ${url} opened`)
     })
@@ -25,11 +21,24 @@ export class AutoReconnectingWebsocket extends ReconnectingWebSocket {
       console.log(`Websocket to ${url} closed`)
     })
 
-    if (this.opts.mock) {
-      this.opts.mock.handleWebsocketConnection(this)
-    } else {
-      this.reconnect()
+    const origReconnect = this.reconnect
+    this.reconnect = () => {
+      if (this.opts.mock) {
+        this.opts.mock.handleWebsocketConnection(this)
+      } else {
+        this.reconnect()
+      }
+      origReconnect.apply(this)
     }
+
+    this.changeUrl(url)
+  }
+
+  changeUrl(url: string) {
+    // reconnecting-websocket doesn't set this properly with startClosed
+    // so we do it ourselves
+    Object.defineProperty(this, "url", { value: url, configurable: true })
+    this.reconnect()
   }
 
   sendJson(data: unknown) {

test/setup.ts

@@ -0,0 +1,2 @@
+import ws from "ws"
+globalThis.WebSocket = ws

vitest.config.ts

@@ -1,4 +1,8 @@
 /// <reference types="vitest" />
 import { defineConfig } from "vite"
 
-export default defineConfig({})
+export default defineConfig({
+  test: {
+    setupFiles: ["./test/setup.ts"],
+  },
+})