UNPKG

@translated/lara-mcp

Version:

Lara API official MCP server

67 lines (66 loc) 2.68 kB
import { describe, it, expect, beforeEach } from 'vitest'; import { translateHandler, translateSchema } from '../../mcp/tools/translate.js'; import { getMockTranslator, setupTranslatorMock } from '../utils/mocks.js'; // Setup mocks setupTranslatorMock(); describe('translateSchema', () => { it('should validate valid input', () => { const validInput = { text: [{ text: 'Hello world', translatable: true }], target: 'it-IT' }; expect(() => translateSchema.parse(validInput)).not.toThrow(); }); it('should reject invalid input', () => { const invalidInput = { text: 'not an array', target: 'it-IT' }; expect(() => translateSchema.parse(invalidInput)).toThrow(); }); it('should require target language', () => { const missingTarget = { text: [{ text: 'Hello world', translatable: true }] }; expect(() => translateSchema.parse(missingTarget)).toThrow(); }); }); describe('translateHandler', () => { let mockTranslator; beforeEach(() => { mockTranslator = getMockTranslator(); }); it('should call lara.translate with correct parameters', async () => { const mockTranslationResult = { translation: [{ text: 'Ciao mondo', translatable: true }] }; mockTranslator.translate.mockResolvedValue(mockTranslationResult); const args = { text: [{ text: 'Hello world', translatable: true }], target: 'it-IT' }; const result = await translateHandler(args, mockTranslator); expect(mockTranslator.translate).toHaveBeenCalledWith(args.text, null, args.target, { instructions: [], adaptTo: undefined }); expect(result).toEqual(mockTranslationResult.translation); }); it('should handle context and instructions', async () => { const mockTranslationResult = { translation: [{ text: 'Ciao mondo', translatable: true }] }; mockTranslator.translate.mockResolvedValue(mockTranslationResult); const args = { text: [{ text: 'Hello world', translatable: true }], target: 'it-IT', context: 'Casual conversation', instructions: ['Use informal tone'] }; await translateHandler(args, mockTranslator); expect(mockTranslator.translate).toHaveBeenCalledWith(args.text, null, args.target, { instructions: [ 'Use informal tone', 'Always consider the following contextual information: Casual conversation' ], adaptTo: undefined }); }); });