Fix comments after default switch case expressions (#1420)

Fix comments after default switch case expressions

Author: crogoz

palantir-java-format/src/main/java/com/palantir/javaformat/java/JavaInputAstVisitor.java

@@ -1854,6 +1854,7 @@ public Void visitSwitch(SwitchTree node, Void unused) {
     }
 
     protected void visitSwitch(ExpressionTree expression, List<? extends CaseTree> cases) {
+        builder.open(ZERO);
         token("switch");
         builder.space();
         token("(");
@@ -1874,7 +1875,8 @@ protected void visitSwitch(ExpressionTree expression, List<? extends CaseTree> c
         builder.close();
         builder.forcedBreak();
         builder.blankLineWanted(BlankLineWanted.NO);
-        token("}", plusFour);
+        token("}", plusTwo);
+        builder.close();
     }
 
     @Override

palantir-java-format/src/test/resources/com/palantir/javaformat/java/testdata/ExpressionSwitch.input

@@ -58,4 +58,20 @@ class ExpressionSwitch {
           case 2 -> throw new IllegalArgumentException();
           default -> throw new IllegalStateException();};
   }
+
+
+    public void testWithComments(int y) {
+        int x;
+        x = switch (y) {
+    case 1 ->
+1;
+// after case 1
+    case 2 ->
+throw new IllegalArgumentException();
+// after case 2
+    default ->
+throw new IllegalStateException();
+// after default case
+};
+    }
 }

palantir-java-format/src/test/resources/com/palantir/javaformat/java/testdata/ExpressionSwitch.output

@@ -65,4 +65,16 @@ class ExpressionSwitch {
             default -> throw new IllegalStateException();
         };
     }
+
+    public void testWithComments(int y) {
+        int x;
+        x = switch (y) {
+            case 1 -> 1;
+            // after case 1
+            case 2 -> throw new IllegalArgumentException();
+            // after case 2
+            default -> throw new IllegalStateException();
+            // after default case
+        };
+    }
 }