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');
  });
});