Ver Fonte

fix tests for sse service

Olivier Massot há 3 anos atrás
pai
commit
042919f5a6

+ 1 - 1
services/sse/sseSource.ts

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

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

@@ -122,8 +122,14 @@ describe('subscribe', () => {
         expect(onMessage).toHaveBeenCalledWith(1)
         expect(onMessage).toHaveBeenCalledWith(1)
 
 
         // @ts-ignore
         // @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()
     })
     })
 })
 })