hjjc-workbench-web/tests/unit/components/BudgetSubmissionForm.spec.js

jest.mock('@/api/budget/departmentSubmission.js', () => ({}))

import BudgetSubmissionForm from '@/views/budget/collection/components/BudgetSubmissionForm.vue'

describe('BudgetSubmissionForm attachment removal', () => {
  const removedFile = { uid: 1, fileId: 101 }
  const remainingFile = { uid: 2, fileId: 102 }

  it('uses Element UI’s post-removal list for economic-item attachments', () => {
    const row = { fileList: [removedFile, remainingFile] }
    const remainingFiles = [remainingFile]

    BudgetSubmissionForm.methods.handleUploadRemove(removedFile, remainingFiles, row)

    expect(row.fileList).toEqual([remainingFile])
    expect(row.fileList).not.toBe(remainingFiles)
  })

  it('uses Element UI’s post-removal list for general attachments', () => {
    const context = {
      formData: {
        generalFiles: [removedFile, remainingFile]
      }
    }
    const remainingFiles = [remainingFile]

    BudgetSubmissionForm.methods.handleGeneralUploadRemove.call(context, removedFile, remainingFiles)

    expect(context.formData.generalFiles).toEqual([remainingFile])
    expect(context.formData.generalFiles).not.toBe(remainingFiles)
  })
})