Enhance test case to check line number of the error

takahi-i · Apr 21, 2018 · da5ee0d · da5ee0d
1 parent 31d1e45
commit da5ee0d
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/redpen-core/src/test/java/cc/redpen/validator/document/UnexpandedAcronymValidatorTest.java b/redpen-core/src/test/java/cc/redpen/validator/document/UnexpandedAcronymValidatorTest.java
@@ -55,13 +55,15 @@ void testDocument() throws RedPenException {
                                        .build();
 
         RedPen redPen = new RedPen(config);
-        Map<Document, List<ValidationError>> errors = redPen.validate(documents);
-        assertEquals(1, errors.get(documents.get(0)).size());
+        List<ValidationError> errors = redPen.validate(documents).get(documents.get(0));
+        assertEquals(1, errors.size());
+        assertEquals(2, errors.get(0).getLineNumber());
     }
 
     @Test
     void testSimpleSentence() throws Exception {
-        List<Document> documents = new ArrayList<>();documents.add(
+        List<Document> documents = new ArrayList<>();
+        documents.add(
                 Document.builder()
                         .addSection(1)
                         .addParagraph()
@@ -70,8 +72,8 @@ void testSimpleSentence() throws Exception {
 
         Configuration config;
         config = Configuration.builder()
-                .addValidatorConfig(new ValidatorConfiguration("UnexpandedAcronym"))
-                .build();
+                         .addValidatorConfig(new ValidatorConfiguration("UnexpandedAcronym"))
+                         .build();
 
         RedPen redPen = new RedPen(config);
         Map<Document, List<ValidationError>> errors = redPen.validate(documents);