34 lines
1012 B
TypeScript
34 lines
1012 B
TypeScript
import { describe, expect, it, vi } from 'vitest';
|
|
import { createSentenceTranslator } from './providerTranslation';
|
|
|
|
vi.mock('./geminiTranslation', () => ({
|
|
createGeminiSentenceTranslator: vi.fn(() => 'gemini-translator'),
|
|
}));
|
|
|
|
vi.mock('./doubaoTranslation', () => ({
|
|
createDoubaoSentenceTranslator: vi.fn(() => 'doubao-translator'),
|
|
}));
|
|
|
|
describe('createSentenceTranslator', () => {
|
|
it('returns a doubao translator when provider is doubao', () => {
|
|
const translator = createSentenceTranslator({
|
|
provider: 'doubao',
|
|
apiKey: 'ark-key',
|
|
model: 'doubao-seed-2-0-pro-260215',
|
|
baseUrl: 'https://ark.cn-beijing.volces.com/api/v3/responses',
|
|
});
|
|
|
|
expect(translator).toBe('doubao-translator');
|
|
});
|
|
|
|
it('returns a gemini translator when provider is gemini', () => {
|
|
const translator = createSentenceTranslator({
|
|
provider: 'gemini',
|
|
apiKey: 'gemini-key',
|
|
model: 'gemini-2.5-flash',
|
|
});
|
|
|
|
expect(translator).toBe('gemini-translator');
|
|
});
|
|
});
|