Mock websocket on storybook to support OneWayWebsocket

BrunoQuaresma · BrunoQuaresma · commit a7e13a0362fe · 2025-05-14T16:35:43.000Z
diff --git a/site/src/testHelpers/storybook.tsx b/site/src/testHelpers/storybook.tsx
@@ -75,6 +75,8 @@ export const withWebSocket = (Story: FC, { parameters }: StoryContext) => {
 	let callEventsDelay: number;
 
 	window.WebSocket = class WebSocket {
+		public readyState = 1;
+
 		addEventListener(type: string, callback: CallbackFn) {
 			listeners.set(type, callback);
 
@@ -93,6 +95,8 @@ export const withWebSocket = (Story: FC, { parameters }: StoryContext) => {
 			}, 0);
 		}
 
+		removeEventListener(type: string, callback: CallbackFn) {}
+
 		close() {}
 	} as unknown as typeof window.WebSocket;
 
