Prevent class cast exception with assertThat argument

Fixes #400
openrewrite · Sep 13, 2023 · 24742fb · 24742fb
1 parent 2fdde47
commit 24742fb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/org/openrewrite/java/testing/assertj/AdoptAssertJDurationAssertions.java b/src/main/java/org/openrewrite/java/testing/assertj/AdoptAssertJDurationAssertions.java
@@ -129,10 +129,11 @@ private J.MethodInvocation simplifyMultipleAssertions(J.MethodInvocation m, Exec
                 return m;
             }
 
-            J.MethodInvocation assertThatArg = (J.MethodInvocation) ((J.MethodInvocation) select).getArguments().get(0);
-            if (!(assertThatArg instanceof J.MethodInvocation)) {
+            Expression assertThatArgumentExpr = ((J.MethodInvocation) select).getArguments().get(0);
+            if (!(assertThatArgumentExpr instanceof J.MethodInvocation)) {
                 return m;
             }
+            J.MethodInvocation assertThatArg = (J.MethodInvocation) assertThatArgumentExpr;
 
             if (isZero(isEqualToArg) && checkIfRelatedToDuration(assertThatArg)) {
                 String formatted_template = formatTemplate("assertThat(#{any()}).%s();", m.getSimpleName(), asDescription);