Class WebSocket

site/src/testHelpers/storybook.tsx:100–148

	const routedEvents = isRouted ? param : {};

	window.WebSocket = class WebSocket {
		public readyState = 1;
		public binaryType = "blob";

		#listeners = new Map<string, CallbackFn>();
		#callEventsDelay: number | undefined;
		#url: string;

		constructor(url?: string) {
			this.#url = url ?? "";
		}

		send() {}

		addEventListener(type: string, callback: CallbackFn) {
			this.#listeners.set(type, callback);

			// Determine which events this socket should receive.
			let events = broadcastEvents;
			if (isRouted) {
				const matchingKey = Object.keys(routedEvents).find((key) =>
					this.#url.includes(key),
				);
				events = matchingKey ? routedEvents[matchingKey] : [];
			}

			if (events.length === 0) {
				return;
			}

			// Runs when the last event listener is added
			clearTimeout(this.#callEventsDelay);
			this.#callEventsDelay = window.setTimeout(() => {
				for (const entry of events) {
					const callback = this.#listeners.get(entry.event);

					if (callback) {
						entry.event === "message"
							? callback({ data: entry.data })
							: callback();
					}
				}
			}, 0);
		}

		removeEventListener(_type: string, _callback: CallbackFn) {}

		close() {}
	} as unknown as typeof WebSocket;

	return <Story />;
};