packages/docs-core/lib/ParticipantTracker/DocParticipantTracker.spec.ts

   1 import type { UserState } from '@lexical/yjs'
   2 import { DocParticipantTracker } from './DocParticipantTracker'
   3 import type { DocumentState } from '../State/DocumentState'
   4
   5 describe('DocParticipantTracker', () => {
   6   let tracker: DocParticipantTracker
   7   let mockSharedState: jest.Mocked<DocumentState>
   8
   9   beforeEach(() => {
  10     mockSharedState = {
  11       getProperty: jest.fn(),
  12       setProperty: jest.fn(),
  13     } as unknown as jest.Mocked<DocumentState>
  14
  15     tracker = new DocParticipantTracker(mockSharedState)
  16   })
  17
  18   describe('updateParticipantsFromUserStates', () => {
  19     it('should update shared state if the limit is reached', () => {
  20       mockSharedState.getProperty.mockReturnValue(false)
  21       const states = new Array(10).fill({}) as UserState[]
  22
  23       tracker.updateParticipantsFromUserStates(states)
  24
  25       expect(mockSharedState.setProperty).toHaveBeenCalledWith('realtimeIsParticipantLimitReached', true)
  26     })
  27
  28     it('should update shared state if the limit is unbreached', () => {
  29       mockSharedState.getProperty.mockReturnValue(true)
  30       const states = new Array(9).fill({}) as UserState[]
  31
  32       tracker.updateParticipantsFromUserStates(states)
  33
  34       expect(mockSharedState.setProperty).toHaveBeenCalledWith('realtimeIsParticipantLimitReached', false)
  35     })
  36
  37     it('should not update shared state if the limit status has not changed', () => {
  38       mockSharedState.getProperty.mockReturnValue(true)
  39       const states = new Array(10).fill({}) as UserState[]
  40
  41       tracker.updateParticipantsFromUserStates(states)
  42
  43       expect(mockSharedState.setProperty).not.toHaveBeenCalled()
  44     })
  45   })
  46 })