From 24742fb4458d71f11103116090a3337612ca8187 Mon Sep 17 00:00:00 2001
From: Tim te Beek <tim@moderne.io>
Date: Wed, 13 Sep 2023 21:30:19 +0200
Subject: [PATCH] Prevent class cast exception with assertThat argument

Fixes #400
---
 .../java/testing/assertj/AdoptAssertJDurationAssertions.java | 5 +++--
 1 file changed, 3 insertions(+), 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
index dd9a71640..ef26a84bd 100644
--- 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);