优化

2026-08-03 18:15:41 +08:00
parent 370a3926d7
commit 1241d11c01
12 changed files with 637 additions and 44 deletions
@@ -0,0 +1,88 @@
+import ElementPlus from 'element-plus'
+import { createPinia, setActivePinia } from 'pinia'
+import { enableAutoUnmount, flushPromises, mount } from '@vue/test-utils'
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
+import {
+  downloadPlatformExportTask,
+  listPlatformExportTasks,
+  retryPlatformExportTask,
+  type PlatformExportTask,
+} from '@/api/platformExportTasks'
+import { useAccessPreviewStore } from '@/stores/accessPreview'
+import { useAuthStore } from '@/stores/auth'
+import DownloadCenterView from './DownloadCenterView.vue'
+
+vi.mock('@/api/platformExportTasks', () => ({
+  downloadPlatformExportTask: vi.fn(),
+  listPlatformExportTasks: vi.fn(),
+  retryPlatformExportTask: vi.fn(),
+}))
+
+class ResizeObserverStub implements ResizeObserver {
+  observe() {}
+  unobserve() {}
+  disconnect() {}
+}
+
+const task: PlatformExportTask = {
+  id: '9', taskNo: 'EXP9', exportType: 'CUSTOMER_LIST', format: 'XLSX', taskName: '客户列表',
+  status: 'SUCCEEDED', totalRows: 12, processedRows: 12, fileName: '客户列表.xlsx',
+  contentType: 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet', sizeBytes: 2048,
+  failureMessage: null, requestedBy: '1', requestedUsername: 'admin', requestedAt: '2026-08-03T10:00:00',
+  startedAt: '2026-08-03T10:00:01', completedAt: '2026-08-03T10:00:02', expiresAt: '2026-08-10T10:00:02',
+}
+
+enableAutoUnmount(afterEach)
+
+describe('DownloadCenterView', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    vi.stubGlobal('ResizeObserver', ResizeObserverStub)
+    Object.defineProperty(URL, 'createObjectURL', { configurable: true, value: vi.fn(() => 'blob:export') })
+    Object.defineProperty(URL, 'revokeObjectURL', { configurable: true, value: vi.fn() })
+    vi.spyOn(HTMLAnchorElement.prototype, 'click').mockImplementation(() => {})
+    const pinia = createPinia()
+    setActivePinia(pinia)
+    const auth = useAuthStore()
+    auth.token = 'token'
+    auth.permissions = ['platform:download:list', 'platform:customer:export']
+    useAccessPreviewStore().initializeAfterLogin()
+    vi.mocked(listPlatformExportTasks).mockResolvedValue({ records: [task], page: 1, size: 20, total: 1, pages: 1 })
+    vi.mocked(downloadPlatformExportTask).mockResolvedValue(new Blob(['xlsx']))
+    vi.mocked(retryPlatformExportTask).mockResolvedValue({ ...task, status: 'PENDING' })
+  })
+
+  it('lists completed tasks and downloads the backend-generated artifact', async () => {
+    const pinia = setActivePinia(createPinia())
+    const auth = useAuthStore()
+    auth.token = 'token'
+    auth.permissions = ['platform:download:list', 'platform:customer:export']
+    useAccessPreviewStore().initializeAfterLogin()
+    mount(DownloadCenterView, { attachTo: document.body, global: { plugins: [pinia, ElementPlus] } })
+    await flushPromises()
+
+    expect(document.body.textContent).toContain('客户列表')
+    const button = [...document.querySelectorAll<HTMLButtonElement>('button')].find((item) => item.textContent?.trim() === '下载')
+    button?.click()
+    await flushPromises()
+
+    expect(downloadPlatformExportTask).toHaveBeenCalledWith('9')
+    expect(URL.createObjectURL).toHaveBeenCalled()
+  })
+
+  it('shows retry only for failed tasks with the source export permission', async () => {
+    vi.mocked(listPlatformExportTasks).mockResolvedValue({ records: [{ ...task, status: 'FAILED', failureMessage: 'IAM timeout' }], page: 1, size: 20, total: 1, pages: 1 })
+    const pinia = setActivePinia(createPinia())
+    const auth = useAuthStore()
+    auth.token = 'token'
+    auth.permissions = ['platform:download:list', 'platform:customer:export']
+    useAccessPreviewStore().initializeAfterLogin()
+    mount(DownloadCenterView, { attachTo: document.body, global: { plugins: [pinia, ElementPlus] } })
+    await flushPromises()
+
+    const button = [...document.querySelectorAll<HTMLButtonElement>('button')].find((item) => item.textContent?.trim() === '重试')
+    button?.click()
+    await flushPromises()
+    expect(retryPlatformExportTask).toHaveBeenCalledWith('9')
+  })
+})