feat: 语音

2026-08-17 11:37:06 +08:00 · 2026-08-05 09:45:59 +08:00
parent 69bc6f57e7
commit 7529a67f09
49 changed files with 3013 additions and 154 deletions
@@ -0,0 +1,72 @@
+import { render, screen, waitFor } from '@testing-library/react'
+import userEvent from '@testing-library/user-event'
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+import { ExportWorkspace } from '../../src/renderer/src/components/export/ExportWorkspace'
+import type { VoiceModelStatus } from '../../src/shared/voice-recognition'
+
+const readyStatus: VoiceModelStatus = {
+  modelId: 'sensevoice-small-int8',
+  version: '2024-07-17',
+  state: 'ready',
+  downloadedBytes: 239_549_735,
+  totalBytes: 239_549_735,
+  progress: 1,
+  platform: 'win32',
+  architecture: 'x64',
+  supported: true
+}
+
+describe('export voice transcripts', () => {
+  beforeEach(() => {
+    window.api = {
+      getVoiceModelStatus: vi.fn().mockResolvedValue(readyStatus),
+      onExportProgress: vi.fn(() => vi.fn())
+    } as typeof window.api
+  })
+
+  it('enables voice transcription by default for a ready HTML voice export', async () => {
+    const onStartExport = vi.fn().mockResolvedValue({
+      success: true,
+      messageCount: 1,
+      outputPath: 'C:\\fixture\\index.html'
+    })
+    render(
+      <ExportWorkspace
+        contacts={[
+          {
+            m_nsUsrName: 'filehelper',
+            m_nsNickName: '文件传输助手',
+            md5: 'fixture-contact',
+            type: 'user'
+          }
+        ]}
+        selectedContact={null}
+        previewMessages={[]}
+        selfInfo={null}
+        dbReady
+        onSelectContact={vi.fn()}
+        onOpenSettings={vi.fn()}
+        exportTasks={[]}
+        onStartExport={onStartExport}
+        onCancelExport={vi.fn()}
+      />
+    )
+
+    await userEvent.click(screen.getAllByRole('button', { name: /HTML/ })[0])
+    await userEvent.click(screen.getByRole('checkbox', { name: '语音' }))
+
+    const transcriptOption = await screen.findByRole('checkbox', {
+      name: '语音转文字，显示在语音条下方'
+    })
+    expect(transcriptOption).toBeEnabled()
+    expect(transcriptOption).toBeChecked()
+
+    await userEvent.click(screen.getByRole('button', { name: '开始导出' }))
+    await waitFor(() => expect(onStartExport).toHaveBeenCalledOnce())
+    expect(onStartExport.mock.calls[0][0]).toMatchObject({
+      format: 'html',
+      includeVoiceTranscripts: true,
+      kinds: expect.arrayContaining(['voice'])
+    })
+  })
+})