Skip to content

Commit 35a40df

Browse files
committed
fix: update wonky types
1 parent 3a5f7bb commit 35a40df

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

site/src/testHelpers/websockets.ts

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -60,32 +60,30 @@ export function createMockWebSocket(
6060

6161
addEventListener: <E extends WebSocketEventType>(
6262
eventType: E,
63-
callback: WebSocketEventMap[E],
63+
callback: (event: WebSocketEventMap[E]) => void,
6464
) => {
6565
if (!isOpen) {
6666
return;
6767
}
6868

6969
const subscribers = store[eventType];
70-
const cb = callback as unknown as CallbackStore[E][0];
71-
if (!subscribers.includes(cb)) {
72-
subscribers.push(cb);
70+
if (!subscribers.includes(callback)) {
71+
subscribers.push(callback);
7372
}
7473
},
7574

7675
removeEventListener: <E extends WebSocketEventType>(
7776
eventType: E,
78-
callback: WebSocketEventMap[E],
77+
callback: (event: WebSocketEventMap[E]) => void,
7978
) => {
8079
if (!isOpen) {
8180
return;
8281
}
8382

8483
const subscribers = store[eventType];
85-
const cb = callback as unknown as CallbackStore[E][0];
86-
if (subscribers.includes(cb)) {
87-
const updated = store[eventType].filter((c) => c !== cb);
88-
store[eventType] = updated as unknown as CallbackStore[E];
84+
if (subscribers.includes(callback)) {
85+
const updated = store[eventType].filter((c) => c !== callback);
86+
store[eventType] = updated as CallbackStore[E];
8987
}
9088
},
9189

0 commit comments

Comments
 (0)