Mock websocket on storybook to support OneWayWebsocket

coder · BrunoQuaresma · May 15, 2025 · May 13, 2025 · May 13, 2025 · May 14, 2025
commit a7e13a0362fe3853f4a342d955f5dec680ff9ada
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;