fix: make mock publisher more robust

Parkreiner · Parkreiner · commit 8e34e915bef8 · 2025-03-25T19:19:43.000Z
diff --git a/site/src/utils/OneWayWebSocket.test.ts b/site/src/utils/OneWayWebSocket.test.ts
@@ -108,24 +108,36 @@ function createMockWebSocket(
 
 	const publisher: MockPublisher = {
 		publishOpen: (event) => {
+			if (closed) {
+				return;
+			}
 			for (const sub of store.open) {
 				sub(event);
 			}
 		},
 
 		publishError: (event) => {
+			if (closed) {
+				return;
+			}
 			for (const sub of store.error) {
 				sub(event);
 			}
 		},
 
 		publishMessage: (event) => {
+			if (closed) {
+				return;
+			}
 			for (const sub of store.message) {
 				sub(event);
 			}
 		},
 
 		publishClose: (event) => {
+			if (closed) {
+				return;
+			}
 			for (const sub of store.close) {
 				sub(event);
 			}
