a-ghorbani--pocketpal-ai
wehub-resource-sync
101 行
2.8 KiB
TypeScript
101 行
2.8 KiB
TypeScript
/**
|
|
* Jest mock for `@pocketpalai/react-native-speech`.
|
|
*
|
|
* The real package ships untranspiled source + TurboModule bindings, neither
|
|
* of which Jest can load. We expose the subset PocketPal uses, including
|
|
* `createSpeechStream` (v2.1.0+) which returns a handle with
|
|
* `append/finalize/cancel`. Each stream is a jest-tracked object so tests
|
|
* can assert on the calls forwarded into it.
|
|
*/
|
|
|
|
type FinishListener = (event: {utteranceId?: string}) => void;
|
|
const finishListeners = new Set<FinishListener>();
|
|
|
|
const onFinish = jest.fn((listener: FinishListener) => {
|
|
finishListeners.add(listener);
|
|
return {
|
|
remove: jest.fn(() => {
|
|
finishListeners.delete(listener);
|
|
}),
|
|
};
|
|
});
|
|
|
|
/**
|
|
* Test helper: emit an `onFinish` event to all registered listeners.
|
|
* Exported so tests can drive the streaming chain without a real native
|
|
* module.
|
|
*/
|
|
export const __emitFinish = (utteranceId?: string) => {
|
|
for (const l of Array.from(finishListeners)) {
|
|
l({utteranceId});
|
|
}
|
|
};
|
|
|
|
/** Test helper: number of currently-registered onFinish listeners. */
|
|
export const __finishListenerCount = () => finishListeners.size;
|
|
|
|
/** Test helper: clear listeners between tests. */
|
|
export const __resetFinishListeners = () => {
|
|
finishListeners.clear();
|
|
};
|
|
|
|
export interface MockSpeechStream {
|
|
voiceId?: string;
|
|
options?: Record<string, unknown>;
|
|
append: jest.Mock;
|
|
finalize: jest.Mock;
|
|
cancel: jest.Mock;
|
|
}
|
|
|
|
const createdStreams: MockSpeechStream[] = [];
|
|
|
|
const createSpeechStream = jest.fn(
|
|
(voiceId?: string, options?: Record<string, unknown>): MockSpeechStream => {
|
|
const stream: MockSpeechStream = {
|
|
voiceId,
|
|
options,
|
|
append: jest.fn(),
|
|
finalize: jest.fn().mockResolvedValue(undefined),
|
|
cancel: jest.fn().mockResolvedValue(undefined),
|
|
};
|
|
createdStreams.push(stream);
|
|
return stream;
|
|
},
|
|
);
|
|
|
|
/** Test helper: list of streams created since the last reset. */
|
|
export const __getCreatedStreams = (): MockSpeechStream[] => createdStreams;
|
|
|
|
/** Test helper: clear stream history between tests. */
|
|
export const __resetCreatedStreams = () => {
|
|
createdStreams.length = 0;
|
|
};
|
|
|
|
const Speech = {
|
|
initialize: jest.fn().mockResolvedValue(undefined),
|
|
speak: jest.fn().mockResolvedValue(undefined),
|
|
stop: jest.fn().mockResolvedValue(undefined),
|
|
release: jest.fn().mockResolvedValue(undefined),
|
|
getAvailableVoices: jest.fn().mockResolvedValue([
|
|
{
|
|
identifier: 'com.apple.voice.Sarah',
|
|
name: 'Sarah',
|
|
language: 'en-US',
|
|
quality: 0,
|
|
},
|
|
]),
|
|
createSpeechStream,
|
|
onFinish,
|
|
};
|
|
|
|
export default Speech;
|
|
|
|
/** Mirror of the fork's `TTSEngine` enum — minimum surface used by tests. */
|
|
export enum TTSEngine {
|
|
OS_NATIVE = 'os-native',
|
|
KOKORO = 'kokoro',
|
|
SUPERTONIC = 'supertonic',
|
|
POCKET = 'pocket',
|
|
KITTEN = 'kitten',
|
|
}
|