Initial Tests

Author: Jammy-Louie

src/test/java/org/openrewrite/staticanalysis/NeedBracesTest.java

@@ -17,6 +17,7 @@
 
 import org.junit.jupiter.api.Test;
 import org.openrewrite.DocumentExample;
+import org.openrewrite.Issue;
 import org.openrewrite.Tree;
 import org.openrewrite.java.JavaParser;
 import org.openrewrite.java.style.Checkstyle;
@@ -328,4 +329,107 @@ static void method() {
           )
         );
     }
+
+    @Issue("https://github.com/openrewrite/rewrite-static-analysis/issues/315")
+    @Test
+    void commentsBeforeElseBlockWithNoBraces() {
+        rewriteRun(
+          //language=java
+          java(
+            """
+              class Test {
+                  static void method() {
+                      if (true) {
+                          return;
+                      }
+                      /*
+                       * else comment
+                       */
+                      else return;
+                  }
+              }
+              """,
+            """
+              class Test {
+                  static void method() {
+                      if (true) {
+                          return; // if comment
+                      } else {
+                      /*
+                       * else comment
+                       */
+                          return;
+                      }
+                  }
+              }
+              """
+          )
+        );
+    }
+
+    @Issue("https://github.com/openrewrite/rewrite-static-analysis/issues/315")
+    @Test
+    void commentsBeforeElseBlockWithBraces() {
+        rewriteRun(
+          //language=java
+          java(
+            """
+              class Test {
+                  static void method() {
+                      if (true)
+                          return;
+                      // if comment
+                      else{
+                          return;
+                      }
+                  }
+              }
+              """,
+            """
+              class Test {
+                  static void method() {
+                      if (true) {
+                          return;
+                      // if comment
+                      } else {
+                          return;
+                      }
+                  }
+              }
+              """
+          )
+        );
+    }
+
+    @Issue("https://github.com/openrewrite/rewrite-static-analysis/issues/315")
+    @Test
+    void trailingCommentsElseBlock() {
+        rewriteRun(
+          //language=java
+          java(
+            """
+              class Test {
+                  static void method() {
+                      if (true) return; // if comment
+                      else if (false) return; // else if comment
+                      else return; // else comment
+                  }
+              }
+              """,
+            """
+              class Test {
+                  static void method() {
+                      if (true) {
+                          return; // comment
+                      } else if (false) {
+                          return; // else if comment
+                      } else {
+                          return; // else comment
+                      }
+                  }
+              }
+              """
+          )
+        );
+    }
 }