#3156 = add test and exclusion for main method

coiouhkc · Jun 17, 2023 · 2a27d55 · 2a27d55
1 parent 6d87f63
commit 2a27d55
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/rewrite-java/src/main/java/org/openrewrite/java/LombokUtilityClass.java b/rewrite-java/src/main/java/org/openrewrite/java/LombokUtilityClass.java
@@ -137,6 +137,10 @@ public J.MethodDeclaration visitMethodDeclaration(
             if (!method.hasModifier(J.Modifier.Type.Static)) {
                 shouldPerformChanges.set(false);
             }
+
+            if (method.getSimpleName().equalsIgnoreCase("main")) {
+                shouldPerformChanges.set(false);
+            }
             return super.visitMethodDeclaration(method, shouldPerformChanges);
         }
 

diff --git a/rewrite-java/src/test/java/org/openrewrite/java/LombokUtilityClassTest.java b/rewrite-java/src/test/java/org/openrewrite/java/LombokUtilityClassTest.java
@@ -14,7 +14,6 @@
  * - abstract classes
  * - instantiations of changed classes
  * - constructor
- * - public static void main
  */
 class LombokUtilityClassTest implements RewriteTest {
 
@@ -237,6 +236,21 @@ public int add(final int x, final int y) {
             );
         }
 
+        @Test
+        void givenMain() {
+            rewriteRun(
+                    recipeSpec -> recipeSpec.recipe(new LombokUtilityClass()),
+                    java(
+                            """
+                                    public class A {
+                                       public static void main(String[] args) {
+                                       }
+                                    }
+                                    """
+                    )
+            );
+        }
+
     }
 
     @Test