tagger make test cover overwrite mode

languagetool-org · Jan 3, 2015 · dbe7dc5 · dbe7dc5
1 parent 314c637
commit dbe7dc5
Showing 1 changed file with 26 additions and 7 deletions.
diff --git a/languagetool-core/src/test/java/org/languagetool/tagging/CombiningTaggerTest.java b/languagetool-core/src/test/java/org/languagetool/tagging/CombiningTaggerTest.java
@@ -13,27 +13,46 @@
 public class CombiningTaggerTest {
 
   @Test
-  public void testTag() throws Exception {
+  public void testTagNoOverwrite() throws Exception {
-    ManualTagger tagger1 = new ManualTagger(JLanguageTool.getDataBroker().getFromResourceDirAsStream("/xx/added1.txt"));
+    CombiningTagger tagger = getCombiningTagger(false);
-    ManualTagger tagger2 = new ManualTagger(JLanguageTool.getDataBroker().getFromResourceDirAsStream("/xx/added2.txt"));
-    CombiningTagger tagger = new CombiningTagger(tagger1, tagger2, false);
     assertThat(tagger.tag("nosuchword").size(), is(0));
     List<TaggedWord> result = tagger.tag("fullform");
     assertThat(result.size(), is(2));
+    String asString = getAsString(result);
+    assertTrue(asString.contains("baseform1/POSTAG1"));
+    assertTrue(asString.contains("baseform2/POSTAG2"));
+  }
+
+  @Test
+  public void testTagOverwrite() throws Exception {
+    CombiningTagger tagger = getCombiningTagger(true);
+    assertThat(tagger.tag("nosuchword").size(), is(0));
+    List<TaggedWord> result = tagger.tag("fullform");
+    assertThat(result.size(), is(1));
+    String asString = getAsString(result);
+    assertTrue(asString.contains("baseform2/POSTAG2"));
+  }
+
+  private CombiningTagger getCombiningTagger(boolean overwrite) throws IOException {
+    ManualTagger tagger1 = new ManualTagger(JLanguageTool.getDataBroker().getFromResourceDirAsStream("/xx/added1.txt"));
+    ManualTagger tagger2 = new ManualTagger(JLanguageTool.getDataBroker().getFromResourceDirAsStream("/xx/added2.txt"));
+    return new CombiningTagger(tagger1, tagger2, overwrite);
+  }
+
+  private String getAsString(List<TaggedWord> result) {
     StringBuilder sb = new StringBuilder();
     for (TaggedWord taggedWord : result) {
       sb.append(taggedWord.getLemma());
       sb.append("/");
       sb.append(taggedWord.getPosTag());
       sb.append("\n");
     }
-    assertTrue(sb.toString().contains("baseform1/POSTAG1"));
+    return sb.toString();
-    assertTrue(sb.toString().contains("baseform2/POSTAG2"));
   }
 
   @Test(expected = IOException.class)
   public void testInvalidFile() throws Exception {
     new ManualTagger(JLanguageTool.getDataBroker().getFromResourceDirAsStream("/xx/added-invalid.txt"));
- }
+  }
 
 }