mirror of
https://wget.la/https://github.com/Wxw-Gu/WechatExplorer
synced 2026-08-18 03:57:02 +08:00
feat: 收敛 AI Search 检索边界与问答体验
This commit is contained in:
电摇小子
@@ -0,0 +1,315 @@
|
||||
import { render, screen, waitFor } from '@testing-library/react'
|
||||
import userEvent from '@testing-library/user-event'
|
||||
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
||||
import { AISearchWorkspace } from '../../src/renderer/src/components/search/AISearchWorkspace'
|
||||
import { SEARCH_CACHE_KEY, buildSearchCacheKey } from '../../src/renderer/src/components/search/searchUtils'
|
||||
|
||||
const api = {
|
||||
getSettings: vi.fn(),
|
||||
getAppLogPath: vi.fn(),
|
||||
getKnowledgeStatus: vi.fn(),
|
||||
onKnowledgeStatus: vi.fn(),
|
||||
onAiSearchProgress: vi.fn(),
|
||||
getAiSearchProviderStatus: vi.fn(),
|
||||
authorizeAiSearchExternalProvider: vi.fn(),
|
||||
runAiSearch: vi.fn()
|
||||
}
|
||||
|
||||
describe('AISearchWorkspace cache privacy boundary', () => {
|
||||
beforeEach(() => {
|
||||
localStorage.clear()
|
||||
sessionStorage.clear()
|
||||
vi.clearAllMocks()
|
||||
Object.defineProperty(window, 'api', { configurable: true, value: api })
|
||||
api.getSettings.mockResolvedValue({ settings: { debugEnabled: false } })
|
||||
api.getAppLogPath.mockResolvedValue('')
|
||||
api.getKnowledgeStatus.mockResolvedValue({ state: 'ready', processedMessages: 1, totalMessages: 1 })
|
||||
api.onKnowledgeStatus.mockReturnValue(() => undefined)
|
||||
api.onAiSearchProgress.mockReturnValue(() => undefined)
|
||||
api.getAiSearchProviderStatus.mockResolvedValue({
|
||||
configured: true,
|
||||
requiresConsent: true,
|
||||
providerId: 'remote-provider',
|
||||
recipient: 'https://remote.example.test/v1'
|
||||
})
|
||||
})
|
||||
|
||||
it('uses a local cache hit without opening a remote Provider consent dialog or making an AI request', async () => {
|
||||
const query = '最近聊过健身吗？'
|
||||
localStorage.setItem(
|
||||
SEARCH_CACHE_KEY,
|
||||
JSON.stringify([
|
||||
{
|
||||
version: 3,
|
||||
key: buildSearchCacheKey('global', '', '30d', query),
|
||||
createdAt: Date.now(),
|
||||
answer: '缓存结果',
|
||||
evidence: [],
|
||||
senderNames: {},
|
||||
messageCount: 1
|
||||
}
|
||||
])
|
||||
)
|
||||
const onNotice = vi.fn()
|
||||
const confirm = vi.spyOn(window, 'confirm').mockReturnValue(true)
|
||||
render(
|
||||
<AISearchWorkspace
|
||||
contacts={[]}
|
||||
selectedContact={null}
|
||||
dbReady
|
||||
aiModelConfig={{
|
||||
configured: true,
|
||||
providerName: 'Remote Provider',
|
||||
model: 'model',
|
||||
modelName: 'Model',
|
||||
status: 'connected'
|
||||
}}
|
||||
onSelectContact={vi.fn()}
|
||||
onOpenEvidence={vi.fn()}
|
||||
onOpenAISettings={vi.fn()}
|
||||
onNotice={onNotice}
|
||||
/>
|
||||
)
|
||||
|
||||
await userEvent.type(screen.getByRole('textbox'), query)
|
||||
await userEvent.click(screen.getByRole('button', { name: /开始分析/ }))
|
||||
|
||||
await waitFor(() => expect(onNotice).toHaveBeenCalledWith(expect.stringContaining('检索缓存')))
|
||||
expect(confirm).not.toHaveBeenCalled()
|
||||
expect(api.getAiSearchProviderStatus).not.toHaveBeenCalled()
|
||||
expect(api.authorizeAiSearchExternalProvider).not.toHaveBeenCalled()
|
||||
expect(api.runAiSearch).not.toHaveBeenCalled()
|
||||
})
|
||||
|
||||
it('does not reuse a result cached before the current identity-resolution contract', async () => {
|
||||
const query = '技术交流群最近聊了啥'
|
||||
localStorage.setItem(
|
||||
'wxe_ai_search_cache_v10',
|
||||
JSON.stringify([
|
||||
{
|
||||
version: 2,
|
||||
key: buildSearchCacheKey('global', '', '7d', query),
|
||||
createdAt: Date.now(),
|
||||
answer: '旧的全局关键词答案',
|
||||
evidence: [],
|
||||
senderNames: {},
|
||||
messageCount: 2
|
||||
}
|
||||
])
|
||||
)
|
||||
render(
|
||||
<AISearchWorkspace
|
||||
contacts={[]}
|
||||
selectedContact={null}
|
||||
dbReady
|
||||
aiModelConfig={{
|
||||
configured: true,
|
||||
providerName: 'Remote Provider',
|
||||
model: 'model',
|
||||
modelName: 'Model',
|
||||
status: 'connected'
|
||||
}}
|
||||
onSelectContact={vi.fn()}
|
||||
onOpenEvidence={vi.fn()}
|
||||
onOpenAISettings={vi.fn()}
|
||||
onNotice={vi.fn()}
|
||||
/>
|
||||
)
|
||||
|
||||
await userEvent.type(screen.getByRole('textbox'), query)
|
||||
await userEvent.click(screen.getByRole('button', { name: /开始分析/ }))
|
||||
|
||||
await screen.findByRole('dialog', { name: '确认发送本次搜索资料' })
|
||||
expect(screen.queryByText('旧的全局关键词答案')).not.toBeInTheDocument()
|
||||
expect(api.runAiSearch).not.toHaveBeenCalled()
|
||||
})
|
||||
|
||||
it('keeps a cached result visible when refresh is cancelled and restores it after the workspace remounts', async () => {
|
||||
const query = '最近聊过健身吗？'
|
||||
localStorage.setItem(
|
||||
SEARCH_CACHE_KEY,
|
||||
JSON.stringify([
|
||||
{
|
||||
version: 3,
|
||||
key: buildSearchCacheKey('global', '', '30d', query),
|
||||
createdAt: Date.now(),
|
||||
answer: '可恢复的缓存结果',
|
||||
evidence: [],
|
||||
senderNames: {},
|
||||
messageCount: 1
|
||||
}
|
||||
])
|
||||
)
|
||||
const props = {
|
||||
contacts: [],
|
||||
selectedContact: null,
|
||||
dbReady: true,
|
||||
aiModelConfig: {
|
||||
configured: true,
|
||||
providerName: 'Remote Provider',
|
||||
model: 'model',
|
||||
modelName: 'Model',
|
||||
status: 'connected' as const
|
||||
},
|
||||
onSelectContact: vi.fn(),
|
||||
onOpenEvidence: vi.fn(),
|
||||
onOpenAISettings: vi.fn(),
|
||||
onNotice: vi.fn()
|
||||
}
|
||||
const first = render(<AISearchWorkspace {...props} />)
|
||||
await userEvent.type(screen.getByRole('textbox'), query)
|
||||
await userEvent.click(screen.getByRole('button', { name: /开始分析/ }))
|
||||
await screen.findByText('可恢复的缓存结果')
|
||||
|
||||
await userEvent.click(screen.getByRole('button', { name: '刷新数据' }))
|
||||
await screen.findByRole('dialog', { name: '确认发送本次搜索资料' })
|
||||
await userEvent.click(screen.getByRole('button', { name: '取消' }))
|
||||
expect(screen.getByText('可恢复的缓存结果')).toBeInTheDocument()
|
||||
expect(api.runAiSearch).not.toHaveBeenCalled()
|
||||
|
||||
first.unmount()
|
||||
render(<AISearchWorkspace {...props} />)
|
||||
expect(await screen.findByText('可恢复的缓存结果')).toBeInTheDocument()
|
||||
})
|
||||
|
||||
it('cancels before starting a remote AI Search and never opens a native confirmation window', async () => {
|
||||
const onNotice = vi.fn()
|
||||
const confirm = vi.spyOn(window, 'confirm').mockReturnValue(true)
|
||||
render(
|
||||
<AISearchWorkspace
|
||||
contacts={[]}
|
||||
selectedContact={null}
|
||||
dbReady
|
||||
aiModelConfig={{
|
||||
configured: true,
|
||||
providerName: 'Remote Provider',
|
||||
model: 'model',
|
||||
modelName: 'Model',
|
||||
status: 'connected'
|
||||
}}
|
||||
onSelectContact={vi.fn()}
|
||||
onOpenEvidence={vi.fn()}
|
||||
onOpenAISettings={vi.fn()}
|
||||
onNotice={onNotice}
|
||||
/>
|
||||
)
|
||||
|
||||
await userEvent.type(screen.getByRole('textbox'), '最近聊过健身吗？')
|
||||
await userEvent.click(screen.getByRole('button', { name: /开始分析/ }))
|
||||
await screen.findByRole('dialog', { name: '确认发送本次搜索资料' })
|
||||
await userEvent.click(screen.getByRole('button', { name: '取消' }))
|
||||
|
||||
await waitFor(() => expect(onNotice).toHaveBeenCalledWith(expect.stringContaining('已取消本次 AI Search')))
|
||||
expect(confirm).not.toHaveBeenCalled()
|
||||
expect(api.authorizeAiSearchExternalProvider).not.toHaveBeenCalled()
|
||||
expect(api.runAiSearch).not.toHaveBeenCalled()
|
||||
})
|
||||
|
||||
it('scrolls to and flashes the matching Evidence card when an inline citation is clicked', async () => {
|
||||
const scrollIntoView = vi.fn()
|
||||
Object.defineProperty(HTMLElement.prototype, 'scrollIntoView', {
|
||||
configurable: true,
|
||||
value: scrollIntoView
|
||||
})
|
||||
api.getAiSearchProviderStatus.mockResolvedValue({ configured: true, requiresConsent: false })
|
||||
api.runAiSearch.mockResolvedValue({
|
||||
requestId: 'evidence-navigation',
|
||||
status: 'completed',
|
||||
answer: '请查看这条证据 [E7]。',
|
||||
plan: { intent: 'global_topic_search' },
|
||||
knowledge: { indexedMessageCount: 8, indexedChunkCount: 1, totalMessages: 8 },
|
||||
candidateEvidenceCount: 8,
|
||||
contextEvidenceCount: 8,
|
||||
evidence: Array.from({ length: 8 }, (_, index) => ({
|
||||
id: `E${index + 1}`,
|
||||
conversationId: 'fixture-contact',
|
||||
conversationName: '测试会话',
|
||||
conversationType: 'user',
|
||||
messageId: `message-${index + 1}`,
|
||||
sender: `发送者 ${index + 1}`,
|
||||
senderId: `sender-${index + 1}`,
|
||||
timestamp: 1_785_900_000_000 + index,
|
||||
text: `证据 ${index + 1}`
|
||||
})),
|
||||
aggregation: { messageCount: 8, peopleCount: 1, conversationCount: 1, people: [], conversations: [] },
|
||||
agent: { mode: 'agent', toolCalls: 1, trace: [] },
|
||||
timings: {},
|
||||
elapsedMs: 1
|
||||
} as never)
|
||||
render(
|
||||
<AISearchWorkspace
|
||||
contacts={[]}
|
||||
selectedContact={null}
|
||||
dbReady
|
||||
aiModelConfig={{
|
||||
configured: true,
|
||||
providerName: 'Local Provider',
|
||||
model: 'model',
|
||||
modelName: 'Model',
|
||||
status: 'connected'
|
||||
}}
|
||||
onSelectContact={vi.fn()}
|
||||
onOpenEvidence={vi.fn()}
|
||||
onOpenAISettings={vi.fn()}
|
||||
onNotice={vi.fn()}
|
||||
/>
|
||||
)
|
||||
|
||||
await userEvent.type(screen.getByRole('textbox'), '最近聊过健身吗？')
|
||||
await userEvent.click(screen.getByRole('button', { name: /开始分析/ }))
|
||||
await userEvent.click(await screen.findByRole('button', { name: '[E7]' }))
|
||||
|
||||
const card = screen.getByText('E7 · 发送者 7').closest('article')
|
||||
await waitFor(() => expect(card).toHaveClass('focus-flash'))
|
||||
expect(scrollIntoView).toHaveBeenCalledWith({ behavior: 'smooth', block: 'nearest' })
|
||||
})
|
||||
|
||||
it('keeps the submitted result title stable while drafting a new question and clears it from 新问题', async () => {
|
||||
api.getAiSearchProviderStatus.mockResolvedValue({ configured: true, requiresConsent: false })
|
||||
api.runAiSearch.mockResolvedValue({
|
||||
requestId: 'new-question',
|
||||
status: 'completed',
|
||||
answer: 'first answer',
|
||||
plan: { intent: 'global_topic_search' },
|
||||
knowledge: { indexedMessageCount: 1, indexedChunkCount: 1, totalMessages: 1 },
|
||||
candidateEvidenceCount: 1,
|
||||
contextEvidenceCount: 1,
|
||||
evidence: [],
|
||||
aggregation: { messageCount: 1, peopleCount: 1, conversationCount: 1, people: [], conversations: [] },
|
||||
agent: { mode: 'agent', toolCalls: 1, trace: [] },
|
||||
timings: {},
|
||||
elapsedMs: 1
|
||||
} as never)
|
||||
render(
|
||||
<AISearchWorkspace
|
||||
contacts={[]}
|
||||
selectedContact={null}
|
||||
dbReady
|
||||
aiModelConfig={{
|
||||
configured: true,
|
||||
providerName: 'Local Provider',
|
||||
model: 'model',
|
||||
modelName: 'Model',
|
||||
status: 'connected'
|
||||
}}
|
||||
onSelectContact={vi.fn()}
|
||||
onOpenEvidence={vi.fn()}
|
||||
onOpenAISettings={vi.fn()}
|
||||
onNotice={vi.fn()}
|
||||
/>
|
||||
)
|
||||
|
||||
const input = screen.getByRole('textbox')
|
||||
await userEvent.type(input, 'first question')
|
||||
await userEvent.click(screen.getByRole('button', { name: /开始分析/ }))
|
||||
await screen.findByRole('heading', { name: 'first question' })
|
||||
|
||||
await userEvent.clear(input)
|
||||
await userEvent.type(input, 'second question')
|
||||
expect(screen.getByRole('heading', { name: 'first question' })).toBeInTheDocument()
|
||||
|
||||
await userEvent.click(screen.getByRole('button', { name: '新问题' }))
|
||||
expect(screen.queryByRole('heading', { name: 'first question' })).not.toBeInTheDocument()
|
||||
expect(input).toHaveValue('')
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user