packages/docs-core/lib/Controller/Document/DocParticipantTracker.spec.ts

   1 import type { InternalEventBusInterface } from '@proton/docs-shared'
   2 import type { UserState } from '@lexical/yjs'
   3 import { DocParticipantTracker, ParticipantTrackerEvent } from './DocParticipantTracker'
   4
   5 describe('DocParticipantTracker', () => {
   6   let tracker: DocParticipantTracker
   7
   8   beforeEach(() => {
   9     tracker = new DocParticipantTracker({
  10       publish: jest.fn(),
  11     } as unknown as InternalEventBusInterface)
  12   })
  13
  14   describe('updateParticipantsFromUserStates', () => {
  15     it('should update the totalParticipants and publish an event if the limit is reached', () => {
  16       const states = new Array(10).fill({}) as UserState[]
  17
  18       tracker.updateParticipantsFromUserStates(states)
  19
  20       expect(tracker.isParticipantLimitReached()).toBe(true)
  21       expect(tracker.eventBus.publish).toHaveBeenCalledWith({
  22         type: ParticipantTrackerEvent.DocumentLimitBreached,
  23         payload: undefined,
  24       })
  25     })
  26
  27     it('should update the totalParticipants and publish an event if the limit is unbreached', () => {
  28       tracker.updateParticipantsFromUserStates(new Array(10).fill({}) as UserState[])
  29
  30       tracker.eventBus.publish = jest.fn()
  31
  32       tracker.updateParticipantsFromUserStates(new Array(9).fill({}) as UserState[])
  33
  34       expect(tracker.isParticipantLimitReached()).toBe(false)
  35       expect(tracker.eventBus.publish).toHaveBeenCalledWith({
  36         type: ParticipantTrackerEvent.DocumentLimitUnbreached,
  37         payload: undefined,
  38       })
  39     })
  40   })
  41 })