mirror of
https://gitlab.com/soapbox-pub/ditto.git
synced 2025-12-06 03:19:46 +00:00
Add mock LibreTranslate tests
This commit is contained in:
parent
1afb09e604
commit
91f9bd9442
2 changed files with 71 additions and 22 deletions
|
|
@ -1,21 +1,10 @@
|
||||||
import { DittoConf } from '@ditto/conf';
|
|
||||||
import { detectLanguage } from '@ditto/lang';
|
import { detectLanguage } from '@ditto/lang';
|
||||||
import { assertEquals } from '@std/assert';
|
import { assertEquals } from '@std/assert';
|
||||||
|
|
||||||
import { LibreTranslateTranslator } from './LibreTranslateTranslator.ts';
|
import { LibreTranslateTranslator } from './LibreTranslateTranslator.ts';
|
||||||
|
|
||||||
const {
|
Deno.test('LibreTranslate translation with source language omitted', async () => {
|
||||||
libretranslateBaseUrl: baseUrl,
|
const translator = mockLibreTranslate();
|
||||||
libretranslateApiKey: apiKey,
|
|
||||||
translationProvider,
|
|
||||||
} = new DittoConf(Deno.env);
|
|
||||||
|
|
||||||
const libretranslate = 'libretranslate';
|
|
||||||
|
|
||||||
Deno.test('LibreTranslate translation with source language omitted', {
|
|
||||||
ignore: !(translationProvider === libretranslate && apiKey),
|
|
||||||
}, async () => {
|
|
||||||
const translator = new LibreTranslateTranslator({ fetch: fetch, baseUrl, apiKey: apiKey! });
|
|
||||||
|
|
||||||
const data = await translator.translate(
|
const data = await translator.translate(
|
||||||
[
|
[
|
||||||
|
|
@ -33,10 +22,8 @@ Deno.test('LibreTranslate translation with source language omitted', {
|
||||||
assertEquals(detectLanguage(data.results[2], 0), 'ca');
|
assertEquals(detectLanguage(data.results[2], 0), 'ca');
|
||||||
});
|
});
|
||||||
|
|
||||||
Deno.test('LibreTranslate translation with source language set', {
|
Deno.test('LibreTranslate translation with source language set', async () => {
|
||||||
ignore: !(translationProvider === libretranslate && apiKey),
|
const translator = mockLibreTranslate();
|
||||||
}, async () => {
|
|
||||||
const translator = new LibreTranslateTranslator({ fetch: fetch, baseUrl, apiKey: apiKey! });
|
|
||||||
|
|
||||||
const data = await translator.translate(
|
const data = await translator.translate(
|
||||||
[
|
[
|
||||||
|
|
@ -53,3 +40,50 @@ Deno.test('LibreTranslate translation with source language set', {
|
||||||
assertEquals(detectLanguage(data.results[1], 0), 'ca');
|
assertEquals(detectLanguage(data.results[1], 0), 'ca');
|
||||||
assertEquals(detectLanguage(data.results[2], 0), 'ca');
|
assertEquals(detectLanguage(data.results[2], 0), 'ca');
|
||||||
});
|
});
|
||||||
|
|
||||||
|
function mockLibreTranslate(): LibreTranslateTranslator {
|
||||||
|
return new LibreTranslateTranslator({
|
||||||
|
apiKey: 'libretranslate',
|
||||||
|
fetch: async (input, init) => {
|
||||||
|
const req = new Request(input, init);
|
||||||
|
const body = await req.json();
|
||||||
|
|
||||||
|
switch (body.q) {
|
||||||
|
case 'Bom dia amigos':
|
||||||
|
return jsonResponse({
|
||||||
|
detectedLanguage: { language: 'pt' },
|
||||||
|
translatedText: 'Bon dia, amics.',
|
||||||
|
});
|
||||||
|
case 'Meu nome é Patrick, um nome belo ou feio? A questão é mais profunda do que parece.':
|
||||||
|
return jsonResponse({
|
||||||
|
detectedLanguage: { language: 'pt' },
|
||||||
|
translatedText: 'Em dic Patrick, un nom molt o lleig? La pregunta és més profunda del que sembla.',
|
||||||
|
});
|
||||||
|
case 'A respiração é mais importante do que comer e tomar agua.':
|
||||||
|
return jsonResponse({
|
||||||
|
detectedLanguage: { language: 'pt' },
|
||||||
|
translatedText: 'La respiració és més important que menjar i prendre aigua.',
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return new Response(JSON.stringify({ error: 'Not found' }), { status: 404 });
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
interface LibreTranslateResponse {
|
||||||
|
translatedText: string;
|
||||||
|
detectedLanguage?: {
|
||||||
|
language: string;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function jsonResponse(json: LibreTranslateResponse): Response {
|
||||||
|
const body = JSON.stringify(json);
|
||||||
|
|
||||||
|
return new Response(body, {
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json',
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -71,12 +71,20 @@ export class LibreTranslateTranslator implements DittoTranslator {
|
||||||
|
|
||||||
const response = await this.fetch(request);
|
const response = await this.fetch(request);
|
||||||
const json = await response.json();
|
const json = await response.json();
|
||||||
if (!response.ok) {
|
|
||||||
throw new Error(json['error']);
|
|
||||||
}
|
|
||||||
const data = LibreTranslateTranslator.schema().parse(json);
|
|
||||||
|
|
||||||
return data;
|
console.log(json);
|
||||||
|
|
||||||
|
if (!response.ok) {
|
||||||
|
const result = LibreTranslateTranslator.errorSchema().safeParse(json);
|
||||||
|
|
||||||
|
if (result.success) {
|
||||||
|
throw new Error(result.data.error);
|
||||||
|
} else {
|
||||||
|
throw new Error(`Unexpected LibreTranslate error: ${response.statusText} (${response.status})`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return LibreTranslateTranslator.schema().parse(json);
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Libretranslate response schema.
|
/** Libretranslate response schema.
|
||||||
|
|
@ -90,4 +98,11 @@ export class LibreTranslateTranslator implements DittoTranslator {
|
||||||
}).optional(),
|
}).optional(),
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/** Libretranslate error response schema. */
|
||||||
|
private static errorSchema() {
|
||||||
|
return z.object({
|
||||||
|
error: z.string(),
|
||||||
|
});
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue