✨ Adapt mock and add workspace test with websocket mock

frontend/playwright/helpers/MockAPI.js

@@ -0,0 +1,17 @@
+export const interceptRPC = (page, path, jsonFilename) =>
+  page.route(`**/api/rpc/command/${path}`, (route) =>
+    route.fulfill({
+      status: 200,
+      contentType: "application/transit+json",
+      path: `playwright/fixtures/${jsonFilename}`,
+    }),
+  );
+
+export const interceptRPCByRegex = (page, regex, jsonFilename) =>
+  page.route(regex, (route) =>
+    route.fulfill({
+      status: 200,
+      contentType: "application/transit+json",
+      path: `playwright/fixtures/${jsonFilename}`,
+    }),
+  );

frontend/playwright/helpers/MockRPC.js

@@ -1,8 +0,0 @@
-export const interceptRPC = (page, path, jsonFilename) =>
-  page.route(`**/api/rpc/command/${path}`, (route) =>
-    route.fulfill({
-      status: 200,
-      contentType: "application/transit+json",
-      path: `playwright/fixtures/${jsonFilename}`,
-    })
-  );

frontend/playwright/helpers/MockWebSocket.js

@@ -1,81 +1,5 @@
-export class MockWebSocket extends EventTarget {
-  static #mocks = new Map();
-
+export class WebSocketManager {
   static async init(page) {
-    await page.exposeFunction('MockWebSocket$$constructor', (url, protocols) => {
-      console.log('MockWebSocket$$constructor', MockWebSocket, url, protocols)
-      const webSocket = new MockWebSocket(page, url, protocols);
-      this.#mocks.set(url, webSocket);
-    });
-    await page.exposeFunction('MockWebSocket$$spyMessage', (url, data) => {
-      console.log('MockWebSocket$$spyMessage', url, data)
-      this.#mocks.get(url).dispatchEvent(new MessageEvent('message', { data }))
-    });
-    await page.exposeFunction('MockWebSocket$$spyClose', (url, code, reason) => {
-      console.log('MockWebSocket$$spyClose', url, code, reason)
-      this.#mocks.get(url).dispatchEvent(new CloseEvent('close', { code, reason }))
-    });
     await page.addInitScript({ path: "playwright/scripts/MockWebSocket.js" });
   }
-
-  static waitForURL(url) {
-    return new Promise((resolve) => {
-      let intervalID = setInterval(() => {
-        for (const [wsURL, ws] of this.#mocks) {
-          console.log(wsURL)
-          if (wsURL.includes(url)) {
-            clearInterval(intervalID);
-            return resolve(ws);
-          }
-        }
-      }, 30)
-    })
-  }
-
-  #page = null
-  #url
-  #protocols
-
-  // spies.
-  #spyClose = null
-  #spyMessage = null
-
-  constructor(page, url, protocols) {
-    super()
-    this.#page = page
-    this.#url = url
-    this.#protocols = protocols
-  }
-
-  mockOpen(options) {
-    return this.#page.evaluate((options) => {
-      WebSocket.getByURL(url).mockOpen(options)
-    }, options)
-  }
-
-  mockMessage(data) {
-    return this.#page.evaluate((data) => {
-      WebSocket.getByURL(url).mockMessage(data)
-    }, data)
-  }
-
-  mockClose() {
-    return this.#page.evaluate(() => {
-      WebSocket.getByURL(url).mockClose()
-    })
-  }
-
-  spyClose(fn) {
-    if (typeof fn !== 'function') {
-      throw new TypeError('Invalid callback')
-    }
-    this.#spyClose = fn
-  }
-
-  spyMessage(fn) {
-    if (typeof fn !== 'function') {
-      throw new TypeError('Invalid callback')
-    }
-    this.#spyMessage = fn
-  }
 }