feat: 完成 UI 基础层与核心页面迁移

2026-08-21 21:47:00 +08:00 · 2026-08-19 17:43:07 +08:00
parent a72ae49e2c
commit e9cc996ebb
62 changed files with 4279 additions and 5178 deletions
@@ -0,0 +1,102 @@
+import { describe, expect, it, vi } from 'vitest'
+import { ensureAiSearchDataConsent } from '../../src/renderer/src/components/search/services/aiSearchProviderConsent'
+
+const makeApi = (): {
+  getAiSearchProviderStatus: ReturnType<typeof vi.fn>
+  authorizeAiSearchExternalProvider: ReturnType<typeof vi.fn>
+} => ({
+  getAiSearchProviderStatus: vi.fn(),
+  authorizeAiSearchExternalProvider: vi.fn()
+})
+
+describe('ensureAiSearchDataConsent', () => {
+  it('bypasses consent when the configured provider does not require it', async () => {
+    const api = makeApi()
+    api.getAiSearchProviderStatus.mockResolvedValue({ configured: true, requiresConsent: false })
+    const requestConsent = vi.fn()
+
+    await expect(
+      ensureAiSearchDataConsent({
+        requestId: 'req-1',
+        api,
+        requestExternalProviderConsent: requestConsent
+      })
+    ).resolves.toBe(true)
+    expect(requestConsent).not.toHaveBeenCalled()
+    expect(api.authorizeAiSearchExternalProvider).not.toHaveBeenCalled()
+  })
+
+  it('preserves provider fields through consent and authorization', async () => {
+    const api = makeApi()
+    api.getAiSearchProviderStatus.mockResolvedValue({
+      configured: true,
+      requiresConsent: true,
+      providerId: 'provider-1',
+      providerName: 'Remote Provider',
+      recipient: 'remote@example.test'
+    })
+    api.authorizeAiSearchExternalProvider.mockResolvedValue({ success: true })
+    const requestConsent = vi.fn().mockResolvedValue(true)
+
+    await expect(
+      ensureAiSearchDataConsent({
+        requestId: 'req-1',
+        api,
+        requestExternalProviderConsent: requestConsent
+      })
+    ).resolves.toBe(true)
+    expect(requestConsent).toHaveBeenCalledWith('Remote Provider', 'remote@example.test')
+    expect(api.authorizeAiSearchExternalProvider).toHaveBeenCalledWith({
+      requestId: 'req-1',
+      providerId: 'provider-1',
+      recipient: 'remote@example.test'
+    })
+  })
+
+  it('does not authorize when the user rejects consent', async () => {
+    const api = makeApi()
+    api.getAiSearchProviderStatus.mockResolvedValue({
+      configured: true,
+      requiresConsent: true,
+      providerId: 'provider-1',
+      recipient: 'remote@example.test'
+    })
+    const requestConsent = vi.fn().mockResolvedValue(false)
+
+    await expect(
+      ensureAiSearchDataConsent({
+        requestId: 'req-1',
+        api,
+        requestExternalProviderConsent: requestConsent
+      })
+    ).resolves.toBe(false)
+    expect(api.authorizeAiSearchExternalProvider).not.toHaveBeenCalled()
+  })
+
+  it('preserves incomplete-status and authorization errors', async () => {
+    const api = makeApi()
+    api.getAiSearchProviderStatus.mockResolvedValue({ configured: true, requiresConsent: true })
+    await expect(
+      ensureAiSearchDataConsent({
+        requestId: 'req-1',
+        api,
+        requestExternalProviderConsent: vi.fn()
+      })
+    ).rejects.toThrow('当前 AI 服务信息不完整')
+
+    api.getAiSearchProviderStatus.mockResolvedValue({
+      configured: true,
+      requiresConsent: true,
+      providerId: 'provider-1',
+      recipient: 'remote@example.test'
+    })
+    api.authorizeAiSearchExternalProvider.mockResolvedValue({ success: false, error: '授权失败' })
+    await expect(
+      ensureAiSearchDataConsent({
+        requestId: 'req-1',
+        api,
+        requestExternalProviderConsent: vi.fn().mockResolvedValue(true)
+      })
+    ).rejects.toThrow('授权失败')
+  })
+})