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

   1 import type { CommentInterface, CommentThreadInterface, InternalEventBusInterface } from '@proton/docs-shared'
   2 import { CommentThreadState } from '@proton/docs-shared'
   3 import { LocalCommentsState } from './LocalCommentsState'
   4
   5 describe('LocalCommentsState', () => {
   6   let state: LocalCommentsState
   7   let eventBus: InternalEventBusInterface
   8
   9   beforeEach(() => {
  10     eventBus = {
  11       publish: jest.fn(),
  12     } as unknown as InternalEventBusInterface
  13     state = new LocalCommentsState(eventBus)
  14   })
  15
  16   describe('addThread', () => {
  17     it('should add a thread', () => {
  18       state.addThread({ id: '1' } as CommentThreadInterface)
  19
  20       expect(state.getAllThreads()).toEqual([{ id: '1' }])
  21     })
  22
  23     it('should mark a thread as unread', () => {
  24       state.addThread({ id: '1' } as CommentThreadInterface, true)
  25
  26       expect(state.hasUnreadThreads()).toBe(true)
  27     })
  28
  29     it('should prevent duplicates', () => {
  30       state.addThread({ id: '1' } as CommentThreadInterface)
  31       state.addThread({ id: '1' } as CommentThreadInterface)
  32
  33       expect(state.getAllThreads()).toEqual([{ id: '1' }])
  34     })
  35   })
  36
  37   describe('deleteThread', () => {
  38     it('should delete a thread', () => {
  39       state.addThread({ id: '1' } as CommentThreadInterface)
  40       state.deleteThread('1')
  41
  42       expect(state.getAllThreads()).toEqual([])
  43     })
  44
  45     it('should remove thread from unread list', () => {
  46       state.addThread({ id: '1' } as CommentThreadInterface, true)
  47       state.deleteThread('1')
  48
  49       expect(state.hasUnreadThreads()).toBe(false)
  50     })
  51   })
  52
  53   describe('resolveThread', () => {
  54     it('should resolve a thread', () => {
  55       state.addThread({ id: '1', state: CommentThreadState.Active } as CommentThreadInterface)
  56       state.resolveThread('1')
  57
  58       expect(state.findThreadById('1')?.state).toBe(CommentThreadState.Resolved)
  59     })
  60   })
  61
  62   describe('unresolveThread', () => {
  63     it('should unresolve a thread', () => {
  64       state.addThread({ id: '1', state: CommentThreadState.Resolved } as CommentThreadInterface)
  65       state.unresolveThread('1')
  66
  67       expect(state.findThreadById('1')?.state).toBe(CommentThreadState.Active)
  68     })
  69   })
  70
  71   describe('changeThreadState', () => {
  72     it('should change the state of a thread', () => {
  73       state.addThread({ id: '1', state: CommentThreadState.Active } as CommentThreadInterface)
  74       state.changeThreadState('1', CommentThreadState.Resolved)
  75
  76       expect(state.findThreadById('1')?.state).toBe(CommentThreadState.Resolved)
  77     })
  78   })
  79
  80   describe('addComment', () => {
  81     it('should add a comment to a thread', () => {
  82       state.addThread({ id: '1', comments: [] as CommentInterface[] } as CommentThreadInterface)
  83       state.addComment({ id: '1', content: 'test' } as CommentInterface, '1')
  84
  85       expect(state.findThreadById('1')?.comments).toEqual([{ id: '1', content: 'test' }])
  86     })
  87
  88     it('should mark thread as unread when adding a comment', () => {
  89       state.addThread({ id: '1', comments: [] as CommentInterface[] } as CommentThreadInterface)
  90       state.addComment({ id: '1', content: 'test' } as CommentInterface, '1', true)
  91
  92       expect(state.hasUnreadThreads()).toBe(true)
  93     })
  94   })
  95
  96   describe('editComment', () => {
  97     it('should edit a comment in a thread', () => {
  98       state.addThread({ id: '1', comments: [{ id: '1', content: 'test' }] } as CommentThreadInterface)
  99       state.editComment({ commentID: '1', threadID: '1', content: 'edited' })
 100
 101       expect(state.findThreadById('1')?.comments).toEqual([{ id: '1', content: 'edited' }])
 102     })
 103
 104     it('should mark thread as unread when editing a comment', () => {
 105       state.addThread({ id: '1', comments: [{ id: '1', content: 'test' }] } as CommentThreadInterface)
 106       state.editComment({ commentID: '1', threadID: '1', content: 'edited', markThreadUnread: true })
 107
 108       expect(state.hasUnreadThreads()).toBe(true)
 109     })
 110   })
 111
 112   describe('deleteComment', () => {
 113     it('should delete a comment from a thread', () => {
 114       state.addThread({ id: '1', comments: [{ id: '1', content: 'test' }] } as CommentThreadInterface)
 115       state.deleteComment({ commentID: '1', threadID: '1' })
 116
 117       expect(state.findThreadById('1')?.comments).toEqual([])
 118     })
 119   })
 120
 121   describe('replacePlaceholderThread', () => {
 122     it('should replace a placeholder thread', () => {
 123       state.addThread({ id: 'placeholder', isPlaceholder: true } as CommentThreadInterface)
 124       state.replacePlaceholderThread('placeholder', { id: '1', isPlaceholder: false } as CommentThreadInterface)
 125
 126       expect(state.findThreadById('1')?.isPlaceholder).toBe(false)
 127     })
 128   })
 129
 130   describe('replacePlaceholderComment', () => {
 131     it('should replace a placeholder comment', () => {
 132       state.addThread({ id: '1', comments: [{ id: 'placeholder', isPlaceholder: true }] } as CommentThreadInterface)
 133       state.replacePlaceholderComment('placeholder', { id: '1', isPlaceholder: false } as CommentInterface)
 134
 135       expect(state.findThreadById('1')?.comments).toEqual([{ id: '1', isPlaceholder: false }])
 136     })
 137   })
 138 })