Browse Source

fix tests for sse service

Olivier Massot 3 năm trước cách đây
mục cha
commit
042919f5a6
2 tập tin đã thay đổi với 9 bổ sung3 xóa
  1. 1 1
      services/sse/sseSource.ts
  2. 8 2
      tests/units/services/sse/sseSource.test.ts

+ 1 - 1
services/sse/sseSource.ts

@@ -48,7 +48,7 @@ class SseSource {
     this.eventSource = this.createEventSource(this.url.toString(), this.withCredentials)
 
     this.eventSource.onerror = (event) => {
-      if (event.target.readyState === EventSource.CLOSED) {
+      if (event.target.readyState === EventSourcePolyfill.CLOSED) {
         console.log("SSE closed");
       }
       // @ts-ignore

+ 8 - 2
tests/units/services/sse/sseSource.test.ts

@@ -122,8 +122,14 @@ describe('subscribe', () => {
         expect(onMessage).toHaveBeenCalledWith(1)
 
         // @ts-ignore
-        dummyEventSource.onerror()
-        expect(console.error).toHaveBeenCalledWith('SSE - An error happened')
+        const event = { target: { readyState: EventSourcePolyfill.CLOSED } }
+        dummyEventSource.close = vi.fn()
+
+        // @ts-ignore
+        dummyEventSource.onerror(event)
+        expect(console.log).toHaveBeenCalledWith('SSE closed')
+
+        expect(dummyEventSource.close).toHaveBeenCalled()
     })
 })