WechatExplorer/tests/component/personal-wechat-send-dialog.test.tsx

import { fireEvent, render, screen, waitFor } from '@testing-library/react'
import { beforeEach, describe, expect, it, vi } from 'vitest'
import { PersonalWechatSendDialog } from '../../src/renderer/src/components/chat/PersonalWechatSendDialog'

const getStatus = vi.fn()
const rebind = vi.fn()
const selectImage = vi.fn()
const selectVoice = vi.fn()
const sendMessage = vi.fn()
const getTextToSpeechSettings = vi.fn()
const listTextToSpeechVoices = vi.fn()
const synthesizeTextToSpeech = vi.fn()
const removeGeneratedTextToSpeechAudio = vi.fn()

const contact = {
  m_nsUsrName: 'fixture-room@chatroom',
  m_nsNickName: '技术交流群',
  md5: 'fixture-md5',
  type: 'group' as const
}

const readyStatus = {
  state: 'online' as const,
  platform: 'darwin',
  arch: 'arm64',
  sipDisabled: true,
  wechatRunning: true,
  wechatPid: 4668,
  boundWechatPid: 4668,
  oneBotPid: 5401,
  endpoint: '127.0.0.1:58080',
  endpointReady: true,
  wechatVersion: '4.1.11.53',
  runtimeReady: true,
  attachReady: true,
  baseAddress: '0x114ef8000',
  baseAddressReady: true,
  textHookInstalled: true,
  textHookReady: true,
  imageHookInstalled: true,
  imageHookReady: true,
  messageListenerReady: true,
  canSend: true,
  canSendText: true,
  canSendImage: true,
  canSendVoice: true,
  message: '个人微信已绑定'
}

describe('PersonalWechatSendDialog', () => {
  beforeEach(() => {
    getStatus.mockReset().mockResolvedValue(readyStatus)
    rebind.mockReset().mockResolvedValue(readyStatus)
    selectImage.mockReset().mockResolvedValue({
      canceled: false,
      path: '/Users/fixture/test.png',
      name: 'test.png'
    })
    selectVoice.mockReset().mockResolvedValue({
      canceled: false,
      path: '/Users/fixture/test.silk',
      name: 'test.silk'
    })
    sendMessage.mockReset().mockResolvedValue({ success: true, status: readyStatus })
    getTextToSpeechSettings.mockReset().mockResolvedValue({
      success: true,
      settings: {
        provider: 'fish-audio',
        hasApiKey: true,
        encryptionAvailable: true,
        selectedVoiceId: 'fish-warm-female',
        outputFormat: 'mp3',
        model: 's2.1-pro-free',
        phase: 'ready'
      },
      voices: []
    })
    listTextToSpeechVoices.mockReset().mockResolvedValue({
      success: true,
      items: [
        {
          id: 'fish-warm-female',
          name: '暖阳女声',
          description: '自然温和',
          tags: ['女声'],
          languages: ['中文'],
          source: 'fish-audio'
        }
      ],
      total: 1,
      pageNumber: 1,
      pageSize: 24,
      hasMore: false
    })
    synthesizeTextToSpeech.mockReset().mockResolvedValue({
      success: true,
      filePath: '/tmp/generated.mp3',
      audioDataUrl: 'data:audio/mpeg;base64,fixture'
    })
    removeGeneratedTextToSpeechAudio.mockReset().mockResolvedValue({ success: true })
    Object.defineProperty(window, 'api', {
      configurable: true,
      value: {
        getPersonalWechatSenderStatus: getStatus,
        rebindPersonalWechatSender: rebind,
        selectPersonalWechatImage: selectImage,
        selectPersonalWechatVoice: selectVoice,
        sendPersonalWechatMessage: sendMessage,
        getTextToSpeechSettings,
        listTextToSpeechVoices,
        synthesizeTextToSpeech,
        removeGeneratedTextToSpeechAudio
      }
    })
  })

  it('switches to voice mode and sends only the selected audio file', async () => {
    render(<PersonalWechatSendDialog contact={contact} isGroupChat onClose={vi.fn()} />)
    await screen.findByText('技术交流群')
    fireEvent.click(screen.getByRole('radio', { name: '语音' }))
    fireEvent.click(screen.getByRole('radio', { name: '选择本地文件' }))
    fireEvent.click(screen.getByRole('button', { name: '选择语音' }))
    expect(await screen.findByText('test.silk')).toBeInTheDocument()
    fireEvent.click(screen.getByRole('button', { name: '测试发送语音到群聊' }))
    await waitFor(() =>
      expect(sendMessage).toHaveBeenCalledWith({
        type: 'voice',
        to: 'fixture-room@chatroom',
        filePath: '/Users/fixture/test.silk',
        isGroup: true
      })
    )
  })

  it('shows the selected TTS voice and keeps generation separate from sending', async () => {
    const openSettings = vi.fn()
    render(
      <PersonalWechatSendDialog
        contact={contact}
        isGroupChat
        onClose={vi.fn()}
        onOpenTextToSpeechSettings={openSettings}
      />
    )
    await screen.findByText('技术交流群')
    fireEvent.click(screen.getByRole('radio', { name: '语音' }))
    expect(await screen.findByText('暖阳女声')).toBeInTheDocument()
    expect(screen.getByRole('textbox', { name: '要生成的文字' })).toHaveValue('1')
    expect(screen.getByRole('button', { name: '生成语音' })).toBeEnabled()
    fireEvent.click(screen.getByRole('button', { name: '前往文字转语音设置' }))
    expect(openSettings).toHaveBeenCalledTimes(1)
    expect(sendMessage).not.toHaveBeenCalled()
  })

  it('shows binding diagnostics and only offers image and voice modes', async () => {
    render(<PersonalWechatSendDialog contact={contact} isGroupChat onClose={vi.fn()} />)

    expect(await screen.findByText('PID 4668')).toBeInTheDocument()
    expect(screen.getByText('PID 5401 · 绑定 4668')).toBeInTheDocument()
    expect(screen.getByText('0x114ef8000')).toBeInTheDocument()
    expect(screen.getByText('已捕获，可发送')).toBeInTheDocument()
    expect(screen.getByRole('radio', { name: '图片' })).toHaveAttribute('aria-checked', 'true')
    expect(screen.getByRole('radio', { name: '语音' })).toBeVisible()
    expect(screen.queryByRole('radio', { name: '文字' })).not.toBeInTheDocument()
  })

  it('switches to image mode and sends only the selected image', async () => {
    render(<PersonalWechatSendDialog contact={contact} isGroupChat onClose={vi.fn()} />)
    await screen.findByText('技术交流群')
    fireEvent.click(screen.getByRole('radio', { name: '图片' }))
    expect(screen.queryByRole('textbox')).not.toBeInTheDocument()
    expect(
      screen.getByText(
        '如果想测试图片，请先在微信中给任意好友手动发送一张普通图片，再点击重新检测。',
        { exact: false }
      )
    ).toBeInTheDocument()
    fireEvent.click(screen.getByRole('button', { name: '选择图片' }))
    expect(await screen.findByText('test.png')).toBeInTheDocument()
    fireEvent.click(screen.getByRole('button', { name: '测试发送图片到群聊' }))

    await waitFor(() =>
      expect(sendMessage).toHaveBeenCalledWith({
        type: 'image',
        to: 'fixture-room@chatroom',
        filePath: '/Users/fixture/test.png',
        isGroup: true
      })
    )
  })

  it('offers a safe explicit rebind action', async () => {
    render(<PersonalWechatSendDialog contact={contact} isGroupChat onClose={vi.fn()} />)
    await screen.findByText('技术交流群')
    fireEvent.click(screen.getByRole('button', { name: '尝试重新绑定' }))
    await waitFor(() => expect(rebind).toHaveBeenCalledTimes(1))
  })

  it('blocks image sending until the media Hook and image directory are initialized', async () => {
    getStatus.mockResolvedValue({
      ...readyStatus,
      state: 'hook_not_ready',
      imageHookReady: false,
      canSendImage: false,
      message: '请先手动发送图片'
    })
    render(<PersonalWechatSendDialog contact={contact} isGroupChat onClose={vi.fn()} />)
    expect(await screen.findByText('已绑定，等待消息初始化')).toBeInTheDocument()
    expect(screen.getByRole('button', { name: '测试发送图片到群聊' })).toBeDisabled()
  })
})