87 lines
4.0 KiB
TypeScript
87 lines
4.0 KiB
TypeScript
import { describe, expect, it, vi } from 'vitest';
|
|
import { createLearningSpeechClient } from '@electron/services/learning-speech-client';
|
|
|
|
describe('Learning speech client', () => {
|
|
it('keeps authentication in Main and forwards bounded multipart audio', async () => {
|
|
const fetchImpl = vi.fn<typeof fetch>().mockResolvedValue(new Response(
|
|
JSON.stringify({ text: '什么是变量', model: 'stt' }),
|
|
{ status: 200, headers: { 'Content-Type': 'application/json' } },
|
|
));
|
|
const client = createLearningSpeechClient({
|
|
fetchImpl,
|
|
getAccessToken: vi.fn().mockResolvedValue('main-owned-token'),
|
|
apiBaseUrl: 'https://square.example',
|
|
});
|
|
|
|
await expect(client.transcribe({
|
|
audio: new Uint8Array([1, 2, 3]),
|
|
fileName: 'voice.webm',
|
|
mimeType: 'audio/webm',
|
|
language: 'zh-CN',
|
|
})).resolves.toEqual({ text: '什么是变量' });
|
|
|
|
const [url, init] = fetchImpl.mock.calls[0];
|
|
expect(url).toBe('https://square.example/api/speech/transcriptions');
|
|
expect(init?.headers).toEqual({ Authorization: 'Bearer main-owned-token' });
|
|
const form = init?.body as FormData;
|
|
expect(form.get('language')).toBe('zh-CN');
|
|
const file = form.get('audio') as File;
|
|
expect(file.name).toBe('voice.webm');
|
|
expect(new Uint8Array(await file.arrayBuffer())).toEqual(new Uint8Array([1, 2, 3]));
|
|
});
|
|
|
|
it('projects only transcript text and never forwards upstream error detail', async () => {
|
|
const fetchImpl = vi.fn<typeof fetch>()
|
|
.mockResolvedValueOnce(new Response(JSON.stringify({ text: ' 安全答案 ', model: 'secret-model', token: 'secret' }), { status: 200 }))
|
|
.mockResolvedValueOnce(new Response(JSON.stringify({ detail: 'provider key invalid at internal-host' }), { status: 502 }));
|
|
const client = createLearningSpeechClient({
|
|
fetchImpl,
|
|
getAccessToken: vi.fn().mockResolvedValue('token'),
|
|
apiBaseUrl: 'https://square.example',
|
|
});
|
|
|
|
await expect(client.transcribe({ audio: new Uint8Array([1]), fileName: 'a.webm', mimeType: 'audio/webm' }))
|
|
.resolves.toEqual({ text: '安全答案' });
|
|
await expect(client.transcribe({ audio: new Uint8Array([1]), fileName: 'a.webm', mimeType: 'audio/webm' }))
|
|
.rejects.toThrow('语音识别失败');
|
|
});
|
|
|
|
it('does not upload audio after its captured account changes during token lookup', async () => {
|
|
let resolveToken!: (token: string) => void;
|
|
const fetchImpl = vi.fn<typeof fetch>();
|
|
const client = createLearningSpeechClient({
|
|
fetchImpl,
|
|
getAccessToken: vi.fn(() => new Promise((resolve) => { resolveToken = resolve; })),
|
|
apiBaseUrl: 'https://square.example',
|
|
});
|
|
let current = true;
|
|
const result = client.transcribe(
|
|
{ audio: new Uint8Array([1]), fileName: 'a.webm', mimeType: 'audio/webm' },
|
|
() => { if (!current) throw new Error('account changed'); },
|
|
);
|
|
current = false;
|
|
resolveToken('new-account-token');
|
|
await expect(result).rejects.toThrow('account changed');
|
|
expect(fetchImpl).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('does not retry a 401 after its captured account changes during refresh', async () => {
|
|
let resolveRefresh!: (token: string) => void;
|
|
const getAccessToken = vi.fn()
|
|
.mockResolvedValueOnce('old-token')
|
|
.mockReturnValueOnce(new Promise((resolve) => { resolveRefresh = resolve; }));
|
|
const fetchImpl = vi.fn<typeof fetch>().mockResolvedValueOnce(new Response(null, { status: 401 }));
|
|
const client = createLearningSpeechClient({ fetchImpl, getAccessToken, apiBaseUrl: 'https://square.example' });
|
|
let current = true;
|
|
const result = client.transcribe(
|
|
{ audio: new Uint8Array([1]), fileName: 'a.webm', mimeType: 'audio/webm' },
|
|
() => { if (!current) throw new Error('account changed'); },
|
|
);
|
|
await vi.waitFor(() => expect(getAccessToken).toHaveBeenCalledTimes(2));
|
|
current = false;
|
|
resolveRefresh('new-account-token');
|
|
await expect(result).rejects.toThrow('account changed');
|
|
expect(fetchImpl).toHaveBeenCalledOnce();
|
|
});
|
|
});
|