fix: add testcase of unsub

alexzhang1030 · May 13, 2024 · 5b75e7a · 5b75e7a
1 parent ff68cdd
commit 5b75e7a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/src/__test__/send.test.ts b/src/__test__/send.test.ts
@@ -117,7 +117,7 @@ describe('send', () => {
         throw new Error('a')
     })
 
-    wshe.subscribeRaw(ev => (event = ev))
+    const unsubscribe = wshe.subscribeRaw(ev => (event = ev))
     wshe.sendRaw(eventData)
     await vi.waitFor(() => {
       vi.setSystemTime(date)
@@ -127,5 +127,16 @@ describe('send', () => {
 
     expect(event).instanceof(Blob)
     expect(event).toStrictEqual(new Blob([eventData]))
+
+    event = undefined
+    unsubscribe()
+    wshe.sendRaw(eventData)
+    await expect(new Promise<void>((_, reject) => {
+      vi.waitFor(() => {
+        vi.setSystemTime(date)
+        if (event === undefined)
+          reject(new Error('Binary Data received back'))
+      })
+    })).rejects.toThrow()
   })
 })
diff --git a/src/index.ts b/src/index.ts
@@ -56,8 +56,6 @@ export function createWSHE<
      * @returns unsubscribe function
      */
     subscribeRaw<T extends DataTypes>(callback: (data: T) => void) {
-      if (!ws)
-        return
       emitter.on(RAW_EVENT, callback)
       return () => {
         emitter.off(RAW_EVENT, callback)