shudao-main/shudao-vue-frontend/src/utils/attachmentFile.test.js

import { describe, expect, it } from 'vitest'

import { getAttachmentCardIcon } from './attachmentFile'

describe('attachmentFile', () => {
  it('uses the document card icon for supported upload formats', () => {
    const documentIcon = '/assets/doc-card.png'

    expect(getAttachmentCardIcon('.docx', documentIcon)).toBe(documentIcon)
    expect(getAttachmentCardIcon('pdf', documentIcon)).toBe(documentIcon)
    expect(getAttachmentCardIcon('.pptx', documentIcon)).toBe(documentIcon)
    expect(getAttachmentCardIcon('.txt', documentIcon)).toBe(documentIcon)
  })

  it('uses fallback icon for unknown formats', () => {
    expect(getAttachmentCardIcon('.zip', '/assets/doc-card.png', 'fallback')).toBe('fallback')
  })
})