Extend test coverage

javaparser · Sep 5, 2017 · 97f69aa · 97f69aa
1 parent 615f56a
commit 97f69aa
Showing 1 changed file with 36 additions and 3 deletions.
diff --git a/javaparser-testing/src/test/java/com/github/javaparser/printer/PrettyPrinterTest.java b/javaparser-testing/src/test/java/com/github/javaparser/printer/PrettyPrinterTest.java
@@ -98,7 +98,7 @@ public void printUseTestVisitor(){
     }
 
     @Test
-    public void prettyColumnAlignParameters() {
+    public void prettyColumnAlignParameters_enabled() {
         PrettyPrinterConfiguration config = new PrettyPrinterConfiguration();
         config.setIndent("\t");
         config.setColumnAlignParameters(true);
@@ -122,8 +122,24 @@ public void prettyColumnAlignParameters() {
     }
 
     @Test
-    public void prettyAlignMethodCallChains() {
+    public void prettyColumnAlignParameters_disabled() {
-        IntStream.range(0, 10).filter(x -> x % 2 == 0).map(x -> x * IntStream.of(1,3,5,1).sum()).forEach(System.out::println);
+        PrettyPrinterConfiguration config = new PrettyPrinterConfiguration();
+        final String EOL = config.getEndOfLineCharacter();
+
+        String code = "class Example { void foo(Object arg0,Object arg1){ myMethod(1, 2, 3, 5, Object.class); } }";
+        String expected = "class Example {" + EOL + 
+                "" + EOL + 
+                "    void foo(Object arg0, Object arg1) {" + EOL + 
+                "        myMethod(1, 2, 3, 5, Object.class);" + EOL + 
+                "    }" + EOL + 
+                "}" + EOL + 
+                "";
+
+        assertEquals(expected, new PrettyPrinter(config).print(JavaParser.parse(code)));
+    }
+
+    @Test
+    public void prettyAlignMethodCallChains_enabled() {
         PrettyPrinterConfiguration config = new PrettyPrinterConfiguration();
         config.setIndent("\t");
         config.setColumnAlignFirstMethodChain(true);
@@ -145,5 +161,22 @@ public void prettyAlignMethodCallChains() {
 
         assertEquals(expected, new PrettyPrinter(config).print(JavaParser.parse(code)));
     }
+
+    @Test
+    public void prettyAlignMethodCallChains_disabled() {
+        PrettyPrinterConfiguration config = new PrettyPrinterConfiguration();
+        final String EOL = config.getEndOfLineCharacter();
+
+        String code = "class Example { void foo() { IntStream.range(0, 10).filter(x -> x % 2 == 0).map(x -> x * IntStream.of(1,3,5,1).sum()).forEach(System.out::println); } }";
+        String expected = "class Example {" + EOL + 
+                "" + EOL + 
+                "    void foo() {" + EOL + 
+                "        IntStream.range(0, 10).filter(x -> x % 2 == 0).map(x -> x * IntStream.of(1, 3, 5, 1).sum()).forEach(System.out::println);" + EOL + 
+                "    }" + EOL + 
+                "}" + EOL + 
+                "";
+
+        assertEquals(expected, new PrettyPrinter(config).print(JavaParser.parse(code)));
+    }
 
 }