Add test

omegat-org · Jul 11, 2024 · c3a8525 · c3a8525
1 parent efdb31b
commit c3a8525
Show file tree

Hide file tree

Showing 2 changed files with 149 additions and 0 deletions.
diff --git a/test/data/mark/alternative.tmx b/test/data/mark/alternative.tmx
@@ -0,0 +1,27 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE tmx SYSTEM "tmx11.dtd">
+<tmx version="1.1">
+    <header srclang="en" />
+    <body>
+        <tu>
+            <tuv lang="en">
+                <seg>Edit</seg>
+            </tuv>
+            <tuv lang="fr">
+                <seg>default</seg>
+            </tuv>
+        </tu>
+        <tu>
+            <prop type="file">file1</prop>
+            <prop type="id">1_1</prop>
+            <prop type="prev">prev1</prop>
+            <prop type="next">next1</prop>
+            <tuv lang="en">
+                <seg>Edit</seg>
+            </tuv>
+            <tuv lang="fr">
+                <seg>alternative</seg>
+            </tuv>
+        </tu>
+    </body>
+</tmx>
diff --git a/test/src/org/omegat/gui/editor/mark/AltTranslationsMarkerTest.java b/test/src/org/omegat/gui/editor/mark/AltTranslationsMarkerTest.java
@@ -0,0 +1,122 @@
+/**************************************************************************
+ OmegaT - Computer Assisted Translation (CAT) tool
+          with fuzzy matching, translation memory, keyword search,
+          glossaries, and translation leveraging into updated projects.
+
+ Copyright (C) 2024 Lev Abashkin
+               2024 Hiroshi Miura
+               Home page: https://www.omegat.org/
+               Support center: https://omegat.org/support
+
+ This file is part of OmegaT.
+
+ OmegaT is free software: you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ OmegaT is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ **************************************************************************/
+package org.omegat.gui.editor.mark;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+
+import java.nio.file.Paths;
+import java.util.Collections;
+import java.util.List;
+
+import org.junit.Before;
+import org.junit.Test;
+
+import org.omegat.core.Core;
+import org.omegat.core.TestCoreInitializer;
+import org.omegat.core.data.EntryKey;
+import org.omegat.core.data.NotLoadedProject;
+import org.omegat.core.data.ProjectProperties;
+import org.omegat.core.data.ProjectTMX;
+import org.omegat.core.data.SourceTextEntry;
+import org.omegat.core.data.TMXEntry;
+import org.omegat.core.segmentation.SRX;
+import org.omegat.core.segmentation.Segmenter;
+import org.omegat.util.Language;
+
+public class AltTranslationsMarkerTest  extends MarkerTestBase {
+
+    private ProjectTMX projectTMX;
+
+    @Before
+    public void preUp() throws Exception {
+        TestCoreInitializer.initEditor(editor);
+        Core.setSegmenter(new Segmenter(SRX.getDefault()));
+        projectTMX = new ProjectTMX(new Language("en"), new Language("fr"), true,
+                Paths.get("test/data/mark/alternative.tmx").toFile(), null);
+        Core.setProject(new NotLoadedProject() {
+            @Override
+            public boolean isProjectLoaded() {
+                return true;
+            }
+
+            @Override
+            public ProjectProperties getProjectProperties() {
+                return new ProjectProperties() {
+                    @Override
+                    public Language getSourceLanguage() {
+                        return new Language("en");
+                    }
+
+                    @Override
+                    public Language getTargetLanguage() {
+                        return new Language("fr");
+                    }
+                };
+            }
+
+            @Override
+            public TMXEntry getTranslationInfo(SourceTextEntry ste) {
+                if (ste == null || projectTMX == null) {
+                    return EMPTY_TRANSLATION;
+                }
+                TMXEntry r = projectTMX.getMultipleTranslation(ste.getKey());
+                if (r == null) {
+                    r = projectTMX.getDefaultTranslation(ste.getSrcText());
+                }
+                if (r == null) {
+                    r = EMPTY_TRANSLATION;
+                }
+                return r;
+            }
+
+        });
+    }
+
+    @Test
+    public void testAltTranslationsMarker() throws Exception {
+        IMarker marker = new AltTranslationsMarker();
+        Core.getEditor().getSettings().setMarkAltTranslations(true);
+        String sourceText = "Edit";
+        String translationText = "default";
+        // default entry: file and
+        EntryKey key0 = new EntryKey("file0", sourceText, "1_0", "prev0", "next0", "path");
+        SourceTextEntry ste0 = new SourceTextEntry(key0, 1, new String[0], sourceText,
+                Collections.emptyList());
+        List<Mark> result = marker.getMarksForEntry(ste0, sourceText, translationText, true);
+        assertNull(result);
+        // alternative entry: file and 1_0
+        translationText = "alternative";
+        EntryKey key1 = new EntryKey("file1", sourceText, "1_1", "prev1", "next1", null);
+        SourceTextEntry ste1 = new SourceTextEntry(key1, 1, new String[0], sourceText,
+                Collections.emptyList());
+        result = marker.getMarksForEntry(ste1, sourceText, translationText, true);
+        assertNotNull(result);
+        assertEquals(1, result.size());
+        // TODO: further checks
+    }
+}