test: update test case.

jaywcjlove · Dec 7, 2024 · 2fa546a · 2fa546a
1 parent 5d39bf3
commit 2fa546a
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/test/run.test.js b/test/run.test.js
@@ -70,6 +70,21 @@ function __triggerKeyboardFocus(el, keyCode, opt) {
   el.dispatchEvent ? el.dispatchEvent(eventObj) : el.fireEvent('onfocus', eventObj);
 }
 
+jest.mock('ws', () => {
+  return {
+    WebSocket: class {
+      constructor(url) {
+        this.url = url;
+        this.readyState = 1;
+        this.send = jest.fn();
+        this.close = jest.fn();
+        this.addEventListener = jest.fn();
+        this.removeEventListener = jest.fn();
+      }
+    }
+  };
+});
+
 beforeAll(async () => {
   browser = await puppeteer.launch({ args: ['--no-sandbox'] });
   page = await browser.newPage();