packages/docs-core/lib/Services/Comments/CommentController.spec.ts

   1 import { CommentController } from './CommentController'
   2 import type { DocumentKeys, NodeMeta } from '@proton/drive-store'
   3 import type { LoggerInterface } from '@proton/utils/logs'
   4 import type { InternalEventBusInterface } from '@proton/docs-shared'
   5
   6 import type { PrivateKeyReference, SessionKey } from '@proton/crypto'
   7 import type { WebsocketServiceInterface } from '../Websockets/WebsocketServiceInterface'
   8 import type { DocsApi } from '../../Api/DocsApi'
   9 import type { EncryptComment } from '../../UseCase/EncryptComment'
  10 import type { CreateThread } from '../../UseCase/CreateThread'
  11 import type { CreateComment } from '../../UseCase/CreateComment'
  12 import type { LoadThreads } from '../../UseCase/LoadThreads'
  13 import type { HandleRealtimeCommentsEvent } from '../../UseCase/HandleRealtimeCommentsEvent'
  14 import { WebsocketConnectionEvent } from '../../Realtime/WebsocketEvent/WebsocketConnectionEvent'
  15 import type { MetricService } from '../Metrics/MetricService'
  16 import type { UnleashClient } from '@proton/unleash'
  17
  18 describe('CommentController', () => {
  19   let controller: CommentController
  20
  21   let encryptComment: EncryptComment
  22   let createThread: CreateThread
  23   let createComment: CreateComment
  24   let loadThreads: LoadThreads
  25
  26   let websocketService: WebsocketServiceInterface
  27   let eventBus: InternalEventBusInterface
  28   let api: DocsApi
  29   let logger: LoggerInterface
  30   let document: NodeMeta
  31   let keys: DocumentKeys
  32   let handleRealtimeCommentsEvent: HandleRealtimeCommentsEvent
  33
  34   beforeEach(() => {
  35     document = { linkId: 'link-id-123', volumeId: 'volume-id-456' } as NodeMeta
  36
  37     keys = {
  38       documentContentKey: 'key-123' as unknown as SessionKey,
  39       userAddressPrivateKey: 'private-key-123' as unknown as PrivateKeyReference,
  40       userOwnAddress: 'foo',
  41     }
  42
  43     websocketService = {} as unknown as jest.Mocked<WebsocketServiceInterface>
  44
  45     api = {} as unknown as jest.Mocked<DocsApi>
  46
  47     eventBus = {
  48       addEventHandler: jest.fn(),
  49     } as unknown as jest.Mocked<InternalEventBusInterface>
  50
  51     logger = {
  52       info: jest.fn(),
  53       debug: jest.fn(),
  54       error: jest.fn(),
  55     } as unknown as jest.Mocked<LoggerInterface>
  56
  57     encryptComment = {
  58       execute: jest.fn(),
  59     } as unknown as jest.Mocked<EncryptComment>
  60
  61     createThread = {
  62       execute: jest.fn(),
  63     } as unknown as jest.Mocked<CreateThread>
  64
  65     createComment = {
  66       execute: jest.fn(),
  67     } as unknown as jest.Mocked<CreateComment>
  68
  69     loadThreads = {
  70       execute: jest.fn(),
  71     } as unknown as jest.Mocked<LoadThreads>
  72
  73     handleRealtimeCommentsEvent = {
  74       execute: jest.fn(),
  75     } as unknown as jest.Mocked<HandleRealtimeCommentsEvent>
  76
  77     const metricService = {} as unknown as jest.Mocked<MetricService>
  78
  79     const getLatestDocumentName = jest.fn()
  80
  81     const unleashClient = {
  82       isReady: () => false,
  83       on: jest.fn(),
  84     } as unknown as jest.Mocked<UnleashClient>
  85
  86     controller = new CommentController(
  87       document,
  88       keys,
  89       websocketService,
  90       metricService,
  91       api,
  92       encryptComment,
  93       createThread,
  94       createComment,
  95       loadThreads,
  96       handleRealtimeCommentsEvent,
  97       getLatestDocumentName,
  98       unleashClient,
  99       eventBus,
 100       logger,
 101     )
 102   })
 103
 104   it('should refetch all comments upon websocket reconnection', async () => {
 105     controller.fetchAllComments = jest.fn()
 106
 107     await controller.handleEvent({
 108       type: WebsocketConnectionEvent.ConnectionEstablishedButNotYetReady,
 109       payload: undefined,
 110     })
 111
 112     expect(controller.fetchAllComments).toHaveBeenCalled()
 113   })
 114 })