From b1e8bf08db18b4fc6a4eebc116fe438e262a88db Mon Sep 17 00:00:00 2001 From: Tim te Beek Date: Mon, 12 Aug 2024 09:59:42 +0200 Subject: [PATCH] Context free SimplifyAssertJAssertion (#570) * Context free SimplifyAssertJAssertion * Apply suggestions from code review Co-authored-by: Knut Wannheden --------- Co-authored-by: Knut Wannheden --- .../java/testing/assertj/SimplifyAssertJAssertion.java | 5 ++--- 1 file changed, 2 insertions(+), 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 index 391a9e3f4..56a787bad 100644 --- 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()); } } }