diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test-translator.js | 151 |
1 files changed, 11 insertions, 140 deletions
diff --git a/test/test-translator.js b/test/test-translator.js index 027fc4f5..e5dae5d8 100644 --- a/test/test-translator.js +++ b/test/test-translator.js @@ -18,150 +18,21 @@ const fs = require('fs'); const path = require('path'); const assert = require('assert'); -const {createDictionaryArchive, testMain} = require('../dev/util'); -const {DatabaseVM} = require('../dev/database-vm'); +const {testMain} = require('../dev/util'); +const {TranslatorVM} = require('../dev/translator-vm'); -function createTestDictionaryArchive(dictionary, dictionaryName) { - const dictionaryDirectory = path.join(__dirname, 'data', 'dictionaries', dictionary); - return createDictionaryArchive(dictionaryDirectory, dictionaryName); -} - function clone(value) { return JSON.parse(JSON.stringify(value)); } -async function createVM() { - // Set up VM - const vm = new DatabaseVM(); - vm.execute([ - 'js/core.js', - 'js/data/anki-note-data.js', - 'js/data/database.js', - 'js/data/json-schema.js', - 'js/general/cache-map.js', - 'js/general/regex-util.js', - 'js/general/text-source-map.js', - 'js/language/deinflector.js', - 'js/language/dictionary-data-util.js', - 'js/language/dictionary-importer.js', - 'js/language/dictionary-database.js', - 'js/language/japanese-util.js', - 'js/language/translator.js', - 'js/media/media-util.js' - ]); - const [ - DictionaryImporter, - DictionaryDatabase, - JapaneseUtil, - Translator, - AnkiNoteData - ] = vm.get([ - 'DictionaryImporter', - 'DictionaryDatabase', - 'JapaneseUtil', - 'Translator', - 'AnkiNoteData' - ]); - - // Dictionary - const testDictionary = createTestDictionaryArchive('valid-dictionary2'); - const testDictionaryContent = await testDictionary.generateAsync({type: 'string'}); - - // Setup database - const dictionaryImporter = new DictionaryImporter(); - const dictionaryDatabase = new DictionaryDatabase(); - await dictionaryDatabase.prepare(); - - const {result, errors} = await dictionaryImporter.importDictionary( - dictionaryDatabase, - testDictionaryContent, - {prefixWildcardsSupported: true}, - () => {} - ); - - assert.deepStrictEqual(errors.length, 0); - - // Setup translator - const japaneseUtil = new JapaneseUtil(null); - const translator = new Translator({japaneseUtil, database: dictionaryDatabase}); - const deinflectionReasions = JSON.parse(fs.readFileSync(path.join(__dirname, '..', 'ext', 'data/deinflect.json'))); - translator.prepare(deinflectionReasions); - - // Note data creation - const createPublicAnkiNoteData = (marker, data) => new AnkiNoteData(japaneseUtil, marker, data).createPublic(); - - // Done - return {vm, translator, dictionary: result, createPublicAnkiNoteData}; -} - -function buildOptions(optionsPresets, optionsArray, dictionaryTitle) { - const options = {}; - if (!Array.isArray(optionsArray)) { optionsArray = [optionsArray]; } - for (const entry of optionsArray) { - switch (typeof entry) { - case 'string': - if (!Object.prototype.hasOwnProperty.call(optionsPresets, entry)) { - throw new Error('Invalid options preset'); - } - Object.assign(options, clone(optionsPresets[entry])); - break; - case 'object': - Object.assign(options, clone(entry)); - break; - default: - throw new Error('Invalid options type'); - } - } - - // Construct regex - if (Array.isArray(options.textReplacements)) { - options.textReplacements = options.textReplacements.map((value) => { - if (Array.isArray(value)) { - value = value.map(({pattern, flags, replacement}) => ({pattern: new RegExp(pattern, flags), replacement})); - } - return value; - }); - } - - // Update structure - const placeholder = '${title}'; - if (options.mainDictionary === placeholder) { - options.mainDictionary = dictionaryTitle; - } - let {enabledDictionaryMap} = options; - if (Array.isArray(enabledDictionaryMap)) { - for (const entry of enabledDictionaryMap) { - if (entry[0] === placeholder) { - entry[0] = dictionaryTitle; - } - } - enabledDictionaryMap = new Map(enabledDictionaryMap); - options.enabledDictionaryMap = enabledDictionaryMap; - } - - return options; -} - - async function main() { const write = (process.argv[2] === '--write'); - const {translator, dictionary: {title}, createPublicAnkiNoteData} = await createVM(); - const createTestAnkiNoteData = (definition, mode) => createPublicAnkiNoteData('{marker}', { - definition, - resultOutputMode: mode, - mode: 'mode', - glossaryLayoutMode: 'default', - compactTags: false, - context: { - url: 'url:', - sentence: {text: '', offset: 0}, - documentTitle: 'title' - }, - injectedMedia: null - }); + const translatorVM = new TranslatorVM(); + const dictionaryDirectory = path.join(__dirname, 'data', 'dictionaries', 'valid-dictionary2'); + await translatorVM.prepare(dictionaryDirectory, 'Test Dictionary 2'); const testInputsFilePath = path.join(__dirname, 'data', 'translator-test-inputs.json'); const {optionsPresets, tests} = JSON.parse(fs.readFileSync(testInputsFilePath, {encoding: 'utf8'})); @@ -182,9 +53,9 @@ async function main() { case 'findTerms': { const {name, mode, text} = test; - const options = buildOptions(optionsPresets, test.options, title); - const [definitions, length] = clone(await translator.findTerms(mode, text, options)); - const noteDataList = mode !== 'simple' ? clone(definitions.map((definition) => createTestAnkiNoteData(clone(definition), mode))) : null; + const options = translatorVM.buildOptions(optionsPresets, test.options); + const [definitions, length] = clone(await translatorVM.translator.findTerms(mode, text, options)); + const noteDataList = mode !== 'simple' ? clone(definitions.map((definition) => translatorVM.createTestAnkiNoteData(clone(definition), mode))) : null; actualResults1.push({name, length, definitions}); actualResults2.push({name, noteDataList}); if (!write) { @@ -197,9 +68,9 @@ async function main() { case 'findKanji': { const {name, text} = test; - const options = buildOptions(optionsPresets, test.options, title); - const definitions = clone(await translator.findKanji(text, options)); - const noteDataList = clone(definitions.map((definition) => createTestAnkiNoteData(clone(definition), null))); + const options = translatorVM.buildOptions(optionsPresets, test.options); + const definitions = clone(await translatorVM.translator.findKanji(text, options)); + const noteDataList = clone(definitions.map((definition) => translatorVM.createTestAnkiNoteData(clone(definition), null))); actualResults1.push({name, definitions}); actualResults2.push({name, noteDataList}); if (!write) { |