transduck 0.1.4 → 0.2.1

package/tests/providers.test.ts

@@ -0,0 +1,289 @@
+import { describe, it, expect, vi } from 'vitest';
+import {
+  buildMessages,
+  buildPluralMessages,
+  buildSinglePrompt,
+  buildPluralSinglePrompt,
+  safeRender,
+} from '../src/providers/prompts.js';
+import { getProvider } from '../src/providers/index.js';
+import type { TransduckConfig } from '../src/config.js';
+import { loadConfig } from '../src/config.js';
+import { writeFileSync } from 'fs';
+import { join } from 'path';
+import { mkdtempSync } from 'fs';
+import { tmpdir } from 'os';
+
+function makeConfig(overrides: Partial<TransduckConfig> = {}): TransduckConfig {
+  return {
+    projectName: 'test',
+    projectContext: 'A test site',
+    sourceLang: 'EN',
+    targetLangs: ['DE'],
+    storagePath: '/tmp/test.duckdb',
+    provider: 'openai',
+    apiKeyEnv: 'OPENAI_API_KEY',
+    tokenEnv: 'CLAUDE_CODE_OAUTH_TOKEN',
+    backendModel: 'gpt-4.1-mini',
+    backendTimeout: 10,
+    backendMaxRetries: 2,
+    readOnly: false,
+    ...overrides,
+  };
+}
+
+// ---- Prompts ----
+
+describe('prompts', () => {
+  describe('buildMessages', () => {
+    it('builds system and user messages', () => {
+      const messages = buildMessages({
+        sourceText: 'Hello',
+        sourceLang: 'EN',
+        targetLang: 'DE',
+        projectContext: 'A travel site',
+        stringContext: 'greeting',
+      });
+      expect(messages).toHaveLength(2);
+      expect(messages[0].role).toBe('system');
+      expect(messages[0].content).toContain('EN');
+      expect(messages[0].content).toContain('DE');
+      expect(messages[1].role).toBe('user');
+      expect(messages[1].content).toContain('Hello');
+      expect(messages[1].content).toContain('greeting');
+    });
+
+    it('uses "none" for null string context', () => {
+      const messages = buildMessages({
+        sourceText: 'Hello',
+        sourceLang: 'EN',
+        targetLang: 'DE',
+        projectContext: 'test',
+        stringContext: null,
+      });
+      expect(messages[1].content).toContain('none');
+    });
+  });
+
+  describe('buildPluralMessages', () => {
+    it('builds plural messages with CLDR info', () => {
+      const messages = buildPluralMessages({
+        one: '{count} item',
+        other: '{count} items',
+        sourceLang: 'EN',
+        targetLang: 'DE',
+        projectContext: 'test',
+        stringContext: null,
+      });
+      expect(messages).toHaveLength(2);
+      expect(messages[0].content).toContain('CLDR');
+      expect(messages[1].content).toContain('{count} item');
+      expect(messages[1].content).toContain('{count} items');
+    });
+  });
+
+  describe('buildSinglePrompt', () => {
+    it('combines system and user into one string', () => {
+      const prompt = buildSinglePrompt({
+        sourceText: 'Hello',
+        sourceLang: 'EN',
+        targetLang: 'DE',
+        projectContext: 'A travel site',
+        stringContext: 'greeting',
+      });
+      expect(prompt).toContain('EN');
+      expect(prompt).toContain('DE');
+      expect(prompt).toContain('Hello');
+      expect(prompt).toContain('greeting');
+    });
+  });
+
+  describe('buildPluralSinglePrompt', () => {
+    it('combines plural system and user into one string', () => {
+      const prompt = buildPluralSinglePrompt({
+        one: '{count} item',
+        other: '{count} items',
+        sourceLang: 'EN',
+        targetLang: 'DE',
+        projectContext: 'test',
+        stringContext: null,
+      });
+      expect(prompt).toContain('CLDR');
+      expect(prompt).toContain('{count} item');
+    });
+  });
+
+  describe('safeRender', () => {
+    it('replaces known keys only', () => {
+      const result = safeRender('Hello {name}, you have {count} items', {
+        name: 'Tim',
+      });
+      expect(result).toBe('Hello Tim, you have {count} items');
+    });
+  });
+});
+
+// ---- Provider router ----
+
+describe('getProvider', () => {
+  it('returns openai provider for openai config', async () => {
+    const openaiProvider = await import('../src/providers/openai-provider.js');
+    const provider = await getProvider(makeConfig({ provider: 'openai' }));
+    expect(provider.translate).toBe(openaiProvider.translate);
+    expect(provider.translatePlural).toBe(openaiProvider.translatePlural);
+  });
+
+  it('returns claude_api provider for claude_api config', async () => {
+    const claudeApiProvider = await import('../src/providers/claude-api.js');
+    const provider = await getProvider(makeConfig({ provider: 'claude_api' }));
+    expect(provider.translate).toBe(claudeApiProvider.translate);
+  });
+
+  it('returns claude_code provider for claude_code config', async () => {
+    const claudeCodeProvider = await import('../src/providers/claude-code.js');
+    const provider = await getProvider(makeConfig({ provider: 'claude_code' }));
+    expect(provider.translate).toBe(claudeCodeProvider.translate);
+  });
+
+  it('throws for invalid provider', async () => {
+    await expect(
+      getProvider(makeConfig({ provider: 'invalid' }))
+    ).rejects.toThrow('Unknown provider');
+  });
+});
+
+// ---- OpenAI provider ----
+
+describe('openai-provider', () => {
+  it('translate calls OpenAI and returns result', async () => {
+    const { translate } = await import('../src/providers/openai-provider.js');
+    const mockCreate = vi.fn().mockResolvedValue({
+      choices: [{ message: { content: 'Hallo' } }],
+    });
+
+    const cfg = makeConfig();
+    process.env.OPENAI_API_KEY = 'test-key';
+
+    const result = await translate(
+      'Hello', 'EN', 'DE', 'test', null, cfg,
+      { chat: { completions: { create: mockCreate } } } as any,
+    );
+    expect(result).toBe('Hallo');
+    expect(mockCreate).toHaveBeenCalledOnce();
+  });
+
+  it('translatePlural calls OpenAI and returns parsed JSON', async () => {
+    const { translatePlural } = await import('../src/providers/openai-provider.js');
+    const mockCreate = vi.fn().mockResolvedValue({
+      choices: [{ message: { content: '{"one": "{count} Nachricht", "other": "{count} Nachrichten"}' } }],
+    });
+
+    const cfg = makeConfig();
+    process.env.OPENAI_API_KEY = 'test-key';
+
+    const result = await translatePlural(
+      '{count} message', '{count} messages',
+      'EN', 'DE', 'test', null, cfg,
+      { chat: { completions: { create: mockCreate } } } as any,
+    );
+    expect(result).toEqual({
+      one: '{count} Nachricht',
+      other: '{count} Nachrichten',
+    });
+  });
+});
+
+// ---- Config with provider fields ----
+
+describe('config provider fields', () => {
+  it('defaults to openai provider', () => {
+    const tmpDir = mkdtempSync(join(tmpdir(), 'transduck-cfg-'));
+    const configPath = join(tmpDir, 'transduck.yaml');
+    writeFileSync(configPath, `
+project:
+  name: test
+  context: "test"
+languages:
+  source: EN
+  targets: [DE]
+storage:
+  path: ./translations.duckdb
+backend:
+  api_key_env: OPENAI_API_KEY
+  model: gpt-4.1-mini
+  timeout_seconds: 10
+  max_retries: 2
+`);
+    const cfg = loadConfig(configPath);
+    expect(cfg.provider).toBe('openai');
+    expect(cfg.tokenEnv).toBe('CLAUDE_CODE_OAUTH_TOKEN');
+  });
+
+  it('loads claude_api provider', () => {
+    const tmpDir = mkdtempSync(join(tmpdir(), 'transduck-cfg-'));
+    const configPath = join(tmpDir, 'transduck.yaml');
+    writeFileSync(configPath, `
+project:
+  name: test
+  context: "test"
+languages:
+  source: EN
+  targets: [DE]
+storage:
+  path: ./translations.duckdb
+backend:
+  provider: claude_api
+  api_key_env: ANTHROPIC_API_KEY
+  model: claude-haiku-4-5-20251001
+  timeout_seconds: 15
+  max_retries: 3
+`);
+    const cfg = loadConfig(configPath);
+    expect(cfg.provider).toBe('claude_api');
+    expect(cfg.apiKeyEnv).toBe('ANTHROPIC_API_KEY');
+    expect(cfg.backendModel).toBe('claude-haiku-4-5-20251001');
+  });
+
+  it('claude_code forces readOnly', () => {
+    const tmpDir = mkdtempSync(join(tmpdir(), 'transduck-cfg-'));
+    const configPath = join(tmpDir, 'transduck.yaml');
+    writeFileSync(configPath, `
+project:
+  name: test
+  context: "test"
+languages:
+  source: EN
+  targets: [DE]
+storage:
+  path: ./translations.duckdb
+backend:
+  provider: claude_code
+  token_env: CLAUDE_CODE_OAUTH_TOKEN
+`);
+    const cfg = loadConfig(configPath);
+    expect(cfg.provider).toBe('claude_code');
+    expect(cfg.tokenEnv).toBe('CLAUDE_CODE_OAUTH_TOKEN');
+    expect(cfg.readOnly).toBe(true);
+  });
+
+  it('claude_code minimal config uses defaults', () => {
+    const tmpDir = mkdtempSync(join(tmpdir(), 'transduck-cfg-'));
+    const configPath = join(tmpDir, 'transduck.yaml');
+    writeFileSync(configPath, `
+project:
+  name: test
+  context: "test"
+languages:
+  source: EN
+  targets: [DE]
+storage:
+  path: ./translations.duckdb
+backend:
+  provider: claude_code
+`);
+    const cfg = loadConfig(configPath);
+    expect(cfg.provider).toBe('claude_code');
+    expect(cfg.readOnly).toBe(true);
+    expect(cfg.backendModel).toBe('gpt-4.1-mini');
+  });
+});