packages/docs-core/lib/Realtime/WebsocketState.spec.ts

   1 import { WebsocketState } from './WebsocketState'
   2
   3 describe('WebsocketState', () => {
   4   let state: WebsocketState
   5
   6   beforeEach(() => {
   7     state = new WebsocketState()
   8   })
   9
  10   afterEach(() => {
  11     state.destroy()
  12   })
  13
  14   describe('attempt count', () => {
  15     it('should increment attempts on close', () => {
  16       state.didClose()
  17
  18       expect(state.attemptCount).toBe(1)
  19
  20       state.didClose()
  21
  22       expect(state.attemptCount).toBe(2)
  23     })
  24
  25     it('should increment attempts on fail to fetch token', () => {
  26       state.didFailToFetchToken()
  27
  28       expect(state.attemptCount).toBe(1)
  29
  30       state.didFailToFetchToken()
  31
  32       expect(state.attemptCount).toBe(2)
  33     })
  34   })
  35
  36   describe('connection', () => {
  37     it('should not be connected initially', () => {
  38       expect(state.isConnected).toBe(false)
  39     })
  40
  41     it('should set connected on open', () => {
  42       state.didOpen()
  43
  44       expect(state.isConnected).toBe(true)
  45     })
  46   })
  47
  48   describe('exponential backoff', () => {
  49     it('should have minimum 2 second backoff if no attempts', () => {
  50       expect(state.getBackoffWithoutJitter()).toBe(2000)
  51     })
  52
  53     it('should return backoff time and limit to max backoff time', () => {
  54       state.didClose()
  55       expect(state.getBackoffWithoutJitter()).toBe(2000)
  56
  57       state.didClose()
  58       expect(state.getBackoffWithoutJitter()).toBe(4000)
  59
  60       state.didClose()
  61       expect(state.getBackoffWithoutJitter()).toBe(8000)
  62
  63       state.didClose()
  64       expect(state.getBackoffWithoutJitter()).toBe(16000)
  65
  66       state.didClose()
  67       expect(state.getBackoffWithoutJitter()).toBe(32000)
  68
  69       state.didClose()
  70       expect(state.getBackoffWithoutJitter()).toBe(32000)
  71     })
  72
  73     it('should reset attempts', () => {
  74       state.didClose()
  75       expect(state.getBackoffWithoutJitter()).toBe(2000)
  76
  77       state.didClose()
  78       expect(state.getBackoffWithoutJitter()).toBe(4000)
  79
  80       state.resetAttempts()
  81       expect(state.getBackoffWithoutJitter()).toBe(2000)
  82     })
  83
  84     it('should get backoff with jitter', () => {
  85       state.getJitterFactor = jest.fn().mockReturnValue(1)
  86
  87       state.didClose()
  88       expect(state.getBackoff()).toBe(2000)
  89
  90       state.getJitterFactor = jest.fn().mockReturnValue(1.5)
  91
  92       state.didClose()
  93       expect(state.getBackoff()).toBe(6000)
  94     })
  95   })
  96 })