From 9ed37037b76141813030dd6c06a7c065e366facc Mon Sep 17 00:00:00 2001
From: Sasha Sheikin <myminitrue@gmail.com>
Date: Tue, 4 Jun 2024 21:04:46 +0200
Subject: [PATCH] Adapt copied JUnit classes to TestNG

---
 .../TestNgAssertEqualsToAssertThat.java       | 18 +++++++--------
 .../testng/TestNgAssertFalseToAssertThat.java | 14 ++++++------
 .../TestNgAssertNotEqualsToAssertThat.java    | 18 +++++++--------
 .../TestNgAssertNotNullToAssertThat.java      | 14 ++++++------
 .../testng/TestNgAssertNullToAssertThat.java  | 14 ++++++------
 .../testng/TestNgAssertTrueToAssertThat.java  | 14 ++++++------
 .../resources/META-INF/rewrite/assertj.yml    | 22 +++++++++++++++++++
 7 files changed, 68 insertions(+), 46 deletions(-)

diff --git a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertEqualsToAssertThat.java b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertEqualsToAssertThat.java
index c596cca59..ffc36ec5f 100644
--- a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertEqualsToAssertThat.java
+++ b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertEqualsToAssertThat.java
@@ -36,17 +36,17 @@ public class TestNgAssertEqualsToAssertThat
 
     @Override
     public String getDisplayName() {
-        return "JUnit `assertEquals` to AssertJ";
+        return "TestNG `assertEquals` to AssertJ";
     }
 
     @Override
     public String getDescription() {
-        return "Convert JUnit-style `assertEquals()` to AssertJ's `assertThat().isEqualTo()`.";
+        return "Convert TestNG-style `assertEquals()` to AssertJ's `assertThat().isEqualTo()`.";
     }
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return Preconditions.check(new UsesType<>("org.junit.jupiter.api.Assertions", false), new AssertEqualsToAssertThatVisitor());
+        return Preconditions.check(new UsesType<>("org.testng.Assert", false), new AssertEqualsToAssertThatVisitor());
     }
 
     public static class AssertEqualsToAssertThatVisitor extends JavaIsoVisitor<ExecutionContext> {
@@ -60,23 +60,23 @@ public static class AssertEqualsToAssertThatVisitor extends JavaIsoVisitor<Execu
             return assertionsParser;
         }
 
-        private static final MethodMatcher JUNIT_ASSERT_EQUALS = new MethodMatcher("org.junit.jupiter.api.Assertions" + " assertEquals(..)");
+        private static final MethodMatcher TESTNG_ASSERT_EQUALS = new MethodMatcher("org.testng.Assert" + " assertEquals(..)");
 
         @Override
         public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, ExecutionContext ctx) {
-            if (!JUNIT_ASSERT_EQUALS.matches(method)) {
+            if (!TESTNG_ASSERT_EQUALS.matches(method)) {
                 return method;
             }
 
             List<Expression> args = method.getArguments();
-            Expression expected = args.get(0);
-            Expression actual = args.get(1);
+            Expression expected = args.get(1);
+            Expression actual = args.get(0);
 
             //always add the import (even if not referenced)
             maybeAddImport("org.assertj.core.api.Assertions", "assertThat", false);
 
-            // Remove import for "org.junit.jupiter.api.Assertions" if no longer used.
-            maybeRemoveImport("org.junit.jupiter.api.Assertions");
+            // Remove import for "org.testng.Assert" if no longer used.
+            maybeRemoveImport("org.testng.Assert");
 
             if (args.size() == 2) {
                 return JavaTemplate.builder("assertThat(#{any()}).isEqualTo(#{any()});")
diff --git a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertFalseToAssertThat.java b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertFalseToAssertThat.java
index 4fa3709b8..c5710403c 100644
--- a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertFalseToAssertThat.java
+++ b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertFalseToAssertThat.java
@@ -35,17 +35,17 @@ public class TestNgAssertFalseToAssertThat
 
     @Override
     public String getDisplayName() {
-        return "JUnit `assertFalse` to AssertJ";
+        return "TestNG `assertFalse` to AssertJ";
     }
 
     @Override
     public String getDescription() {
-        return "Convert JUnit-style `assertFalse()` to AssertJ's `assertThat().isFalse()`.";
+        return "Convert TestNG-style `assertFalse()` to AssertJ's `assertThat().isFalse()`.";
     }
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return Preconditions.check(new UsesType<>("org.junit.jupiter.api.Assertions", false), new AssertFalseToAssertThatVisitor());
+        return Preconditions.check(new UsesType<>("org.testng.Assert", false), new AssertFalseToAssertThatVisitor());
     }
 
     public static class AssertFalseToAssertThatVisitor extends JavaIsoVisitor<ExecutionContext> {
@@ -59,11 +59,11 @@ public static class AssertFalseToAssertThatVisitor extends JavaIsoVisitor<Execut
             return assertionsParser;
         }
 
-        private static final MethodMatcher JUNIT_ASSERT_FALSE = new MethodMatcher("org.junit.jupiter.api.Assertions" + " assertFalse(boolean, ..)");
+        private static final MethodMatcher TESTNG_ASSERT_FALSE = new MethodMatcher("org.testng.Assert" + " assertFalse(boolean, ..)");
 
         @Override
         public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, ExecutionContext ctx) {
-            if (!JUNIT_ASSERT_FALSE.matches(method)) {
+            if (!TESTNG_ASSERT_FALSE.matches(method)) {
                 return method;
             }
 
@@ -101,8 +101,8 @@ public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, Execu
             //Make sure there is a static import for "org.assertj.core.api.Assertions.assertThat" (even if not referenced)
             maybeAddImport("org.assertj.core.api.Assertions", "assertThat", false);
 
-            // Remove import for "org.junit.jupiter.api.Assertions" if no longer used.
-            maybeRemoveImport("org.junit.jupiter.api.Assertions");
+            // Remove import for "org.testng.Assert" if no longer used.
+            maybeRemoveImport("org.testng.Assert");
 
             return method;
         }
diff --git a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNotEqualsToAssertThat.java b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNotEqualsToAssertThat.java
index 5a0537246..2d1d1d567 100644
--- a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNotEqualsToAssertThat.java
+++ b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNotEqualsToAssertThat.java
@@ -36,17 +36,17 @@ public class TestNgAssertNotEqualsToAssertThat
 
     @Override
     public String getDisplayName() {
-        return "JUnit `assertNotEquals` to AssertJ";
+        return "TestNG `assertNotEquals` to AssertJ";
     }
 
     @Override
     public String getDescription() {
-        return "Convert JUnit-style `assertNotEquals()` to AssertJ's `assertThat().isNotEqualTo()`.";
+        return "Convert TestNG-style `assertNotEquals()` to AssertJ's `assertThat().isNotEqualTo()`.";
     }
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return Preconditions.check(new UsesType<>("org.junit.jupiter.api.Assertions", false), new AssertNotEqualsToAssertThatVisitor());
+        return Preconditions.check(new UsesType<>("org.testng.Assert", false), new AssertNotEqualsToAssertThatVisitor());
     }
 
     public static class AssertNotEqualsToAssertThatVisitor extends JavaIsoVisitor<ExecutionContext> {
@@ -60,18 +60,18 @@ public static class AssertNotEqualsToAssertThatVisitor extends JavaIsoVisitor<Ex
             return assertionsParser;
         }
 
-        private static final MethodMatcher JUNIT_ASSERT_EQUALS = new MethodMatcher("org.junit.jupiter.api.Assertions" + " assertNotEquals(..)");
+        private static final MethodMatcher TESTNG_ASSERT_EQUALS = new MethodMatcher("org.testng.Assert" + " assertNotEquals(..)");
 
         @Override
         public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, ExecutionContext ctx) {
-            if (!JUNIT_ASSERT_EQUALS.matches(method)) {
+            if (!TESTNG_ASSERT_EQUALS.matches(method)) {
                 return method;
             }
 
             List<Expression> args = method.getArguments();
 
-            Expression expected = args.get(0);
-            Expression actual = args.get(1);
+            Expression expected = args.get(1);
+            Expression actual = args.get(0);
 
             if (args.size() == 2) {
                 method = JavaTemplate.builder("assertThat(#{any()}).isNotEqualTo(#{any()});")
@@ -142,8 +142,8 @@ public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, Execu
             //Make sure there is a static import for "org.assertj.core.api.Assertions.assertThat" (even if not referenced)
             maybeAddImport("org.assertj.core.api.Assertions", "assertThat", false);
 
-            // Remove import for "org.junit.jupiter.api.Assertions" if no longer used.
-            maybeRemoveImport("org.junit.jupiter.api.Assertions");
+            // Remove import for "org.testng.Assert" if no longer used.
+            maybeRemoveImport("org.testng.Assert");
 
             return method;
         }
diff --git a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNotNullToAssertThat.java b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNotNullToAssertThat.java
index 1e57affff..63016ee29 100644
--- a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNotNullToAssertThat.java
+++ b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNotNullToAssertThat.java
@@ -35,17 +35,17 @@ public class TestNgAssertNotNullToAssertThat
 
     @Override
     public String getDisplayName() {
-        return "JUnit `assertNotNull` to AssertJ";
+        return "TestNG `assertNotNull` to AssertJ";
     }
 
     @Override
     public String getDescription() {
-        return "Convert JUnit-style `assertNotNull()` to AssertJ's `assertThat().isNotNull()`.";
+        return "Convert TestNG-style `assertNotNull()` to AssertJ's `assertThat().isNotNull()`.";
     }
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return Preconditions.check(new UsesType<>("org.junit.jupiter.api.Assertions", false), new AssertNotNullToAssertThatVisitor());
+        return Preconditions.check(new UsesType<>("org.testng.Assert", false), new AssertNotNullToAssertThatVisitor());
     }
 
     public static class AssertNotNullToAssertThatVisitor extends JavaIsoVisitor<ExecutionContext> {
@@ -59,11 +59,11 @@ public static class AssertNotNullToAssertThatVisitor extends JavaIsoVisitor<Exec
             return assertionsParser;
         }
 
-        private static final MethodMatcher JUNIT_ASSERT_NOT_NULL_MATCHER = new MethodMatcher("org.junit.jupiter.api.Assertions" + " assertNotNull(..)");
+        private static final MethodMatcher TESTNG_ASSERT_NOT_NULL_MATCHER = new MethodMatcher("org.testng.Assert" + " assertNotNull(..)");
 
         @Override
         public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, ExecutionContext ctx) {
-            if (!JUNIT_ASSERT_NOT_NULL_MATCHER.matches(method)) {
+            if (!TESTNG_ASSERT_NOT_NULL_MATCHER.matches(method)) {
                 return method;
             }
 
@@ -103,8 +103,8 @@ public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, Execu
             //Make sure there is a static import for "org.assertj.core.api.Assertions.assertThat" (even if not referenced)
             maybeAddImport("org.assertj.core.api.Assertions", "assertThat", false);
 
-            //And if there are no longer references to the JUnit assertions class, we can remove the import.
-            maybeRemoveImport("org.junit.jupiter.api.Assertions");
+            //And if there are no longer references to the TestNG assertions class, we can remove the import.
+            maybeRemoveImport("org.testng.Assert");
 
             return method;
         }
diff --git a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNullToAssertThat.java b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNullToAssertThat.java
index 8b6264a8b..8773006d7 100644
--- a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNullToAssertThat.java
+++ b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertNullToAssertThat.java
@@ -35,17 +35,17 @@ public class TestNgAssertNullToAssertThat
 
     @Override
     public String getDisplayName() {
-        return "JUnit `assertNull` to AssertJ";
+        return "TestNG `assertNull` to AssertJ";
     }
 
     @Override
     public String getDescription() {
-        return "Convert JUnit-style `assertNull()` to AssertJ's `assertThat().isNull()`.";
+        return "Convert TestNG-style `assertNull()` to AssertJ's `assertThat().isNull()`.";
     }
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return Preconditions.check(new UsesType<>("org.junit.jupiter.api.Assertions", false), new AssertNullToAssertThatVisitor());
+        return Preconditions.check(new UsesType<>("org.testng.Assert", false), new AssertNullToAssertThatVisitor());
     }
 
     public static class AssertNullToAssertThatVisitor extends JavaIsoVisitor<ExecutionContext> {
@@ -59,11 +59,11 @@ public static class AssertNullToAssertThatVisitor extends JavaIsoVisitor<Executi
             return assertionsParser;
         }
 
-        private static final MethodMatcher JUNIT_ASSERT_NULL_MATCHER = new MethodMatcher("org.junit.jupiter.api.Assertions" + " assertNull(..)");
+        private static final MethodMatcher TESTNG_ASSERT_NULL_MATCHER = new MethodMatcher("org.testng.Assert" + " assertNull(..)");
 
         @Override
         public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, ExecutionContext ctx) {
-            if (!JUNIT_ASSERT_NULL_MATCHER.matches(method)) {
+            if (!TESTNG_ASSERT_NULL_MATCHER.matches(method)) {
                 return method;
             }
 
@@ -102,8 +102,8 @@ public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, Execu
             // Make sure there is a static import for "org.assertj.core.api.Assertions.assertThat" (even if not referenced)
             maybeAddImport("org.assertj.core.api.Assertions", "assertThat", false);
 
-            // Remove import for "org.junit.jupiter.api.Assertions" if no longer used.
-            maybeRemoveImport("org.junit.jupiter.api.Assertions");
+            // Remove import for "org.testng.Assert" if no longer used.
+            maybeRemoveImport("org.testng.Assert");
 
             return method;
         }
diff --git a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertTrueToAssertThat.java b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertTrueToAssertThat.java
index 58b8544be..464ddf620 100644
--- a/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertTrueToAssertThat.java
+++ b/src/main/java/org/openrewrite/java/testing/assertj/testng/TestNgAssertTrueToAssertThat.java
@@ -34,17 +34,17 @@ public class TestNgAssertTrueToAssertThat
         extends Recipe {
     @Override
     public String getDisplayName() {
-        return "JUnit `assertTrue` to AssertJ";
+        return "TestNG `assertTrue` to AssertJ";
     }
 
     @Override
     public String getDescription() {
-        return "Convert JUnit-style `assertTrue()` to AssertJ's `assertThat().isTrue()`.";
+        return "Convert TestNG-style `assertTrue()` to AssertJ's `assertThat().isTrue()`.";
     }
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return Preconditions.check(new UsesType<>("org.junit.jupiter.api.Assertions", false), new AssertTrueToAssertThatVisitor());
+        return Preconditions.check(new UsesType<>("org.testng.Assert", false), new AssertTrueToAssertThatVisitor());
     }
 
     public static class AssertTrueToAssertThatVisitor extends JavaIsoVisitor<ExecutionContext> {
@@ -58,11 +58,11 @@ public static class AssertTrueToAssertThatVisitor extends JavaIsoVisitor<Executi
             return assertionsParser;
         }
 
-        private static final MethodMatcher JUNIT_ASSERT_TRUE = new MethodMatcher("org.junit.jupiter.api.Assertions" + " assertTrue(boolean, ..)");
+        private static final MethodMatcher TESTNG_ASSERT_TRUE = new MethodMatcher("org.testng.Assert" + " assertTrue(boolean, ..)");
 
         @Override
         public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, ExecutionContext ctx) {
-            if (!JUNIT_ASSERT_TRUE.matches(method)) {
+            if (!TESTNG_ASSERT_TRUE.matches(method)) {
                 return method;
             }
 
@@ -101,8 +101,8 @@ public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, Execu
             //Make sure there is a static import for "org.assertj.core.api.Assertions.assertThat" (even if not referenced)
             maybeAddImport("org.assertj.core.api.Assertions", "assertThat", false);
 
-            // Remove import for "org.junit.jupiter.api.Assertions" if no longer used.
-            maybeRemoveImport("org.junit.jupiter.api.Assertions");
+            // Remove import for "org.testng.Assert" if no longer used.
+            maybeRemoveImport("org.testng.Assert");
 
             return method;
         }
diff --git a/src/main/resources/META-INF/rewrite/assertj.yml b/src/main/resources/META-INF/rewrite/assertj.yml
index 18cac7ae6..6afe88c99 100644
--- a/src/main/resources/META-INF/rewrite/assertj.yml
+++ b/src/main/resources/META-INF/rewrite/assertj.yml
@@ -364,3 +364,25 @@ recipeList:
       version: 3.x
       onlyIfUsing: org.assertj.core.api.Assertions
       acceptTransitive: true
+
+---
+type: specs.openrewrite.org/v1beta/recipe
+name: org.openrewrite.java.testing.assertj.TestNgToAssertj
+displayName: Migrate TestNG asserts to AssertJ
+description: AssertJ provides a rich set of assertions, truly helpful error messages, improves test code readability. Converts assertions from `org.testng.Assert` to `org.assertj.core.api.Assertions`.
+tags:
+  - testing
+  - assertj
+recipeList:
+  - org.openrewrite.java.testing.assertj.testng.TestNgAssertEqualsToAssertThat
+  - org.openrewrite.java.testing.assertj.testng.TestNgAssertFalseToAssertThat
+  - org.openrewrite.java.testing.assertj.testng.TestNgAssertNotEqualsToAssertThat
+  - org.openrewrite.java.testing.assertj.testng.TestNgAssertNotNullToAssertThat
+  - org.openrewrite.java.testing.assertj.testng.TestNgAssertNullToAssertThat
+  - org.openrewrite.java.testing.assertj.testng.TestNgAssertTrueToAssertThat
+  - org.openrewrite.java.dependencies.AddDependency:
+      groupId: org.assertj
+      artifactId: assertj-core
+      version: 3.x
+      onlyIfUsing: org.assertj.core.api.Assertions
+      acceptTransitive: true