import { test } from 'node:test'
import assert from 'node:assert/strict'

import { canDownloadItem } from '../src/shared/merge-live-progress.ts'

interface Item {
  taskId?: number
  resultId?: number
  fileReady?: boolean
  fileStatus?: string
  downloadUrl?: string
}

test('downloadable when fileReady is true', () => {
  assert.equal(canDownloadItem({ resultId: 1, fileReady: true }), true)
})

test('downloadable when fileStatus is SUCCESS', () => {
  assert.equal(canDownloadItem({ resultId: 1, fileStatus: 'SUCCESS' }), true)
})

test('downloadable when fileStatus is FAILED (error file or display)', () => {
  assert.equal(canDownloadItem({ resultId: 1, fileStatus: 'FAILED' }), true)
})

test('not downloadable when not ready', () => {
  assert.equal(canDownloadItem({ resultId: 1, fileReady: false, fileStatus: 'RUNNING' }), false)
  assert.equal(canDownloadItem({ resultId: 1 }), false)
})

test('downloadable when downloadUrl present', () => {
  assert.equal(canDownloadItem({ resultId: 1, downloadUrl: 'https://dl.example/a.xlsx' }), true)
})

test('not downloadable without resultId', () => {
  assert.equal(canDownloadItem({ fileReady: true }), false)
  assert.equal(canDownloadItem({ fileStatus: 'SUCCESS' }), false)
})

test('not downloadable while pending or running', () => {
  assert.equal(canDownloadItem({ resultId: 1, fileStatus: 'PENDING' }), false)
  assert.equal(canDownloadItem({ resultId: 1, fileStatus: 'RUNNING' }), false)
  assert.equal(canDownloadItem({ resultId: 1, fileStatus: null }), false)
})

test('downloadability contract frozen', () => {
  assert.equal(canDownloadItem(undefined), false)
  assert.equal(canDownloadItem(null), false)
  assert.equal(canDownloadItem({}), false)
  // 终态与就绪任一满足即可下载；RUNNING/PENDING/空态均不可
  const terminalCases = ['SUCCESS', 'FAILED']
  const runningCases = ['PENDING', 'RUNNING', undefined, null]
  for (const status of terminalCases) {
    assert.equal(canDownloadItem({ resultId: 1, fileStatus: status as string }), true, `终态 ${status} 可下载`)
  }
  for (const status of runningCases) {
    assert.equal(canDownloadItem({ resultId: 1, fileStatus: status as string | undefined }), false,
      `非终态 ${String(status)} 不可下载`)
  }
})