Context free SimplifyAssertJAssertion (#570)

* Context free SimplifyAssertJAssertion * Apply suggestions from code review Co-authored-by: Knut Wannheden <[email protected]> --------- Co-authored-by: Knut Wannheden <[email protected]>
openrewrite · Aug 12, 2024 · b1e8bf0 · b1e8bf0
1 parent 8a037e8
commit b1e8bf0
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/main/java/org/openrewrite/java/testing/assertj/SimplifyAssertJAssertion.java b/src/main/java/org/openrewrite/java/testing/assertj/SimplifyAssertJAssertion.java
@@ -95,11 +95,10 @@ public J.MethodInvocation visitMethodInvocation(J.MethodInvocation methodInvocat
             }
 
             // Assume zero argument replacement method
-            return JavaTemplate.builder(dedicatedAssertion + "()")
-                    .contextSensitive()
+            return JavaTemplate.builder("#{any()}." + dedicatedAssertion + "()")
                     .javaParser(JavaParser.fromJavaVersion().classpathFromResources(ctx, "assertj-core-3.24"))
                     .build()
-                    .apply(getCursor(), mi.getCoordinates().replaceMethod());
+                    .apply(getCursor(), mi.getCoordinates().replace(), mi.getSelect());
         }
     }
 }