orca/mobile/src/session/mobile-diff-comment-edit.test.ts

import { describe, expect, it } from 'vitest'
import type { DiffComment } from '../../../src/shared/types'
import {
  clearSentMobileDiffComments,
  countUnsentMobileDiffComments,
  getUnsentMobileDiffComments,
  markMobileDiffCommentsSent,
  updateMobileDiffComment
} from './mobile-diff-comment-edit'

function comment(overrides: Partial<DiffComment> & Pick<DiffComment, 'id'>): DiffComment {
  const { id, ...rest } = overrides
  return {
    id,
    worktreeId: 'wt-1',
    filePath: 'src/app.ts',
    source: 'diff',
    lineNumber: 4,
    body: 'check this',
    createdAt: 100,
    side: 'modified',
    ...rest
  }
}

describe('mobile diff comment editing', () => {
  it('edits notes and clears sent state', () => {
    const result = updateMobileDiffComment([comment({ id: 'a', sentAt: 150 })], {
      id: 'a',
      body: '  updated  ',
      updatedAt: 200
    })

    expect(result.comment).toMatchObject({ id: 'a', body: 'updated', updatedAt: 200 })
    expect(result.comment?.sentAt).toBeUndefined()
  })

  it('marks notes sent and excludes them from unsent counts', () => {
    const comments = markMobileDiffCommentsSent(
      [comment({ id: 'a' }), comment({ id: 'b' })],
      new Set(['a']),
      250
    )

    expect(comments[0]?.sentAt).toBe(250)
    expect(countUnsentMobileDiffComments(comments)).toBe(1)
    expect(getUnsentMobileDiffComments(comments)).toEqual([comment({ id: 'b' })])
  })

  it('clears sent notes without removing unsent edits', () => {
    expect(
      clearSentMobileDiffComments([comment({ id: 'a', sentAt: 1 }), comment({ id: 'b' })])
    ).toEqual([comment({ id: 'b' })])
  })
})