mirror of
https://wget.la/https://github.com/Wxw-Gu/WechatExplorer
synced 2026-10-04 03:01:42 +08:00
108 lines
3.9 KiB
TypeScript
108 lines
3.9 KiB
TypeScript
import { fireEvent, render, screen, waitFor } from '@testing-library/react'
|
||
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
||
import { LocalImageTextRecognition } from '../../src/renderer/src/features/settings/ai-model/LocalImageTextRecognition'
|
||
import type { SystemOcrCapability, SystemOcrResult } from '../../src/shared/system-ocr'
|
||
|
||
const availableCapability: SystemOcrCapability = {
|
||
available: true,
|
||
engine: 'windows-system-ocr',
|
||
platform: 'win32',
|
||
arch: 'x64',
|
||
runtimeVersion: '1.2.0',
|
||
language: 'zh-Hans-CN',
|
||
message: '本地图片文字识别可用（Windows 系统 OCR，zh-Hans-CN）。'
|
||
}
|
||
|
||
const unavailableCapability: SystemOcrCapability = {
|
||
available: false,
|
||
engine: 'windows-system-ocr',
|
||
platform: 'win32',
|
||
arch: 'x64',
|
||
runtimeVersion: '1.2.0',
|
||
language: null,
|
||
reason: 'LANGUAGE_UNAVAILABLE',
|
||
message: '当前 Windows 未安装可用的 OCR 语言支持。'
|
||
}
|
||
|
||
const successResult: SystemOcrResult = {
|
||
success: true,
|
||
text: 'TraceMemo 本地文字识别',
|
||
lines: [],
|
||
language: 'zh-Hans-CN',
|
||
engine: 'windows-system-ocr',
|
||
durationMs: 42
|
||
}
|
||
|
||
const emptyResult: SystemOcrResult = {
|
||
success: false,
|
||
text: '',
|
||
lines: [],
|
||
language: 'zh-Hans-CN',
|
||
engine: 'windows-system-ocr',
|
||
durationMs: 12,
|
||
errorCode: 'OCR_EMPTY_RESULT',
|
||
error: '没有在这张图片里识别到文字。'
|
||
}
|
||
|
||
const selectImage = (): void => {
|
||
const input = document.querySelector<HTMLInputElement>('input[type="file"]')
|
||
if (!input) throw new Error('file input missing')
|
||
const file = new File([new Uint8Array([0x89, 0x50, 0x4e, 0x47])], 'fixture.png', {
|
||
type: 'image/png'
|
||
})
|
||
fireEvent.change(input, { target: { files: [file] } })
|
||
}
|
||
|
||
describe('LocalImageTextRecognition', () => {
|
||
beforeEach(() => {
|
||
window.api = {
|
||
getSystemOcrCapability: vi.fn().mockResolvedValue(availableCapability),
|
||
recognizeLocalImageText: vi.fn().mockResolvedValue(successResult)
|
||
} as typeof window.api
|
||
})
|
||
|
||
it('shows local availability without requiring any AI provider', async () => {
|
||
render(<LocalImageTextRecognition />)
|
||
expect(await screen.findByText('本机可用')).toBeInTheDocument()
|
||
expect(screen.getByText(/组件 1\.2\.0/)).toBeInTheDocument()
|
||
expect(screen.getByRole('button', { name: '本地文字识别' })).toBeDisabled()
|
||
})
|
||
|
||
it('recognizes text locally and renders the extracted text', async () => {
|
||
render(<LocalImageTextRecognition />)
|
||
await screen.findByText('本机可用')
|
||
|
||
selectImage()
|
||
const button = await screen.findByRole('button', { name: '本地文字识别' })
|
||
await waitFor(() => expect(button).toBeEnabled())
|
||
fireEvent.click(button)
|
||
|
||
expect(await screen.findByText('TraceMemo 本地文字识别')).toBeInTheDocument()
|
||
expect(window.api.recognizeLocalImageText).toHaveBeenCalledWith({
|
||
imageDataUrl: expect.stringContaining('data:image/png;base64,')
|
||
})
|
||
})
|
||
|
||
it('surfaces an empty OCR result as a product message, not a native error', async () => {
|
||
vi.mocked(window.api.recognizeLocalImageText).mockResolvedValue(emptyResult)
|
||
render(<LocalImageTextRecognition />)
|
||
await screen.findByText('本机可用')
|
||
|
||
selectImage()
|
||
const button = await screen.findByRole('button', { name: '本地文字识别' })
|
||
await waitFor(() => expect(button).toBeEnabled())
|
||
fireEvent.click(button)
|
||
|
||
expect(await screen.findByText('没有在这张图片里识别到文字。')).toBeInTheDocument()
|
||
})
|
||
|
||
it('keeps the entry disabled and explains why when the language pack is missing', async () => {
|
||
vi.mocked(window.api.getSystemOcrCapability).mockResolvedValue(unavailableCapability)
|
||
render(<LocalImageTextRecognition />)
|
||
|
||
expect(await screen.findByText('本机不可用')).toBeInTheDocument()
|
||
expect(screen.getByText('当前 Windows 未安装可用的 OCR 语言支持。')).toBeInTheDocument()
|
||
expect(screen.getByRole('button', { name: '本地文字识别' })).toBeDisabled()
|
||
})
|
||
})
|