WechatExplorer/tests/component/support/ai-search-fixtures.ts

import type { Contact } from '../../../src/shared/types'
import { buildSearchCacheKey } from '../../../src/renderer/src/components/search/searchUtils'

export const aiSearchContact: Contact = {
  md5: 'fixture-contact',
  m_nsUsrName: 'wxid_fixture',
  m_nsNickName: '测试会话',
  wechatNickname: 'Fixture User',
  remark: '测试联系人',
  type: 'user'
}

export const aiSearchGroup: Contact = {
  md5: 'fixture-group',
  m_nsUsrName: 'fixture-group@chatroom',
  m_nsNickName: '测试群聊',
  type: 'group'
}

export const makePipelineEvidence = (index: number, conversation = aiSearchContact) => ({
  id: `E${index}`,
  conversationId: conversation.md5,
  conversationName: conversation.m_nsNickName,
  conversationType: conversation.type,
  messageId: `message-${index}`,
  sender: `发送者 ${index}`,
  senderId: `sender-${index}`,
  timestamp: 1_700_000_000_000 + index * 1_000,
  text: `证据 ${index}`
})

export const makeSearchResult = ({
  requestId = 'request-1',
  status = 'completed',
  answer = '测试搜索答案',
  evidence = [],
  evidenceCollection = evidence,
  agentTrace = [],
  error,
  errorStage
}: {
  requestId?: string
  status?: 'completed' | 'no_evidence' | 'retrieval_incomplete' | 'ai_failed' | 'failed' | 'cancelled'
  answer?: string
  evidence?: ReturnType<typeof makePipelineEvidence>[]
  evidenceCollection?: ReturnType<typeof makePipelineEvidence>[]
  agentTrace?: Record<string, unknown>[]
  error?: string
  errorStage?: string
} = {}) =>
  ({
    requestId,
    status,
    answer,
    plan: {
      intent: 'global_topic_search',
      keywords: ['测试'],
      variants: ['测试'],
      source: 'local',
      scopeLabel: '所有聊天记录',
      rangeLabel: '近 30 天',
      timeRange: {
        startTime: 1_699_000_000,
        label: '近 30 天',
        reason: '测试',
        source: 'ui'
      },
      contactNames: []
    },
    knowledge: {
      source: 'knowledge',
      state: 'ready',
      indexedMessageCount: 20,
      indexedChunkCount: 4,
      totalMessages: 20,
      voiceCoverage: undefined
    },
    candidateEvidenceCount: evidenceCollection.length,
    retrieval: {
      intent: 'global_topic_search',
      timeRange: {
        startTime: 1_699_000_000,
        label: '近 30 天',
        reason: '测试',
        source: 'ui'
      },
      retrievalMode: 'global_fts',
      candidateCount: evidenceCollection.length,
      uniqueCandidateCount: evidenceCollection.length,
      sourceCoverage: 'complete',
      isComplete: true,
      fallbackUsed: false,
      suspicious: false
    },
    evidence,
    evidenceCollection,
    contextEvidenceCount: evidence.length,
    aggregation: {
      messageCount: evidenceCollection.length,
      peopleCount: evidenceCollection.length ? 1 : 0,
      conversationCount: evidenceCollection.length ? 1 : 0,
      people: [],
      conversations: []
    },
    agent: {
      mode: 'agent',
      toolCalls: agentTrace.length,
      trace: agentTrace
    },
    citationValidation: { status: 'valid', invalidCitationIds: [] },
    timings: {},
    elapsedMs: 12,
    error,
    errorStage
  }) as never

export const makeCacheRecord = ({
  query,
  scope = 'global',
  contactMd5 = '',
  range = '30d',
  answer = '缓存答案',
  evidence = []
}: {
  query: string
  scope?: 'global' | 'groups' | 'contacts' | 'conversation'
  contactMd5?: string
  range?: 'today' | '7d' | '30d' | 'all'
  answer?: string
  evidence?: ReturnType<typeof makePipelineEvidence>[]
}) => ({
  version: 3 as const,
  key: buildSearchCacheKey(scope, contactMd5, range, query),
  createdAt: 1_700_000_000_000,
  answer,
  evidence: evidence.map((item) => ({
    evidenceId: item.id,
    contact: scope === 'conversation' ? aiSearchContact : aiSearchContact,
    message: {
      id: item.messageId,
      from: item.senderId,
      type: '检索消息',
      datetime: new Date(item.timestamp).toLocaleString('zh-CN', { hour12: false }),
      content: item.text,
      isSender: false,
      name: item.sender,
      senderId: item.senderId,
      createTime: Math.floor(item.timestamp / 1000)
    }
  })),
  senderNames: Object.fromEntries(evidence.map((item) => [item.senderId, item.sender])),
  messageCount: evidence.length
})