All files / msw / handlers test-sse.ts

11.11% Statements 4/36
0% Branches 0/10
0% Functions 0/9
12.12% Lines 4/33

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77  73x 73x 73x                                   73x                                                                                                              
import { http, HttpResponse } from 'msw'
 
export type SSEController = {
  send: (event: string, data?: string) => void
  close: () => void
  isOpen: () => boolean
}
 
export let activeController: SSEController | null = null

let autoFailCount = 0
let autoFailRemaining = 0

export function setAutoFailConnections(n: number) {
  autoFailCount = 0
  autoFailRemaining = n
}

export function resetAutoFail() {
  autoFailCount = 0
  autoFailRemaining = 0
}

export function resetSSEController() {
  if (activeController) {
    try { activeController.close() } catch {}
    activeController = null
  }
  resetAutoFail()
}

export const testSSEHandlers = [
  http.get('/test-events', () => {
    if (autoFailCount < autoFailRemaining) {
      autoFailCount++
      return new HttpResponse(null, { status: 500 })
    }
 
    let closed = false
    const encoder = new TextEncoder()

    const stream = new ReadableStream({
      start(controller) {
        activeController = {
          send(event: string, data?: string) {
            if (closed) return
            let msg = `event: ${event}\n`
            if (data !== undefined) msg += `data: ${data}\n`
            msg += '\n'
            controller.enqueue(encoder.encode(msg))
          },
          close() {
            if (closed) return
            closed = true
            activeController = null
            controller.close()
          },
          isOpen() {
            return !closed
          },
        }
      },
      cancel() {
        closed = true
        activeController = null
      },
    })
 
    return new HttpResponse(stream, {
      headers: {
        'Content-Type': 'text/event-stream',
        'Cache-Control': 'no-cache',
      },
    })
  }),
]