[review] Simplifies assertion

apache · Dec 6, 2021 · 172e638 · 172e638
1 parent 8aa2e83
commit 172e638
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/.../src/test/java/org/apache/flink/runtime/jobmaster/DefaultJobMasterServiceProcessTest.java b/.../src/test/java/org/apache/flink/runtime/jobmaster/DefaultJobMasterServiceProcessTest.java
@@ -30,7 +30,6 @@
 import org.apache.flink.runtime.rest.handler.legacy.utils.ArchivedExecutionGraphBuilder;
 import org.apache.flink.runtime.scheduler.ExecutionGraphInfo;
 import org.apache.flink.runtime.scheduler.exceptionhistory.RootExceptionHistoryEntry;
-import org.apache.flink.util.ExceptionUtils;
 import org.apache.flink.util.SerializedThrowable;
 import org.apache.flink.util.TestLogger;
 
@@ -138,14 +137,8 @@ private static void assertInitializationException(
                 actualException.deserializeError(Thread.currentThread().getContextClassLoader());
 
         assertThat(deserializedException)
-                .isOfAnyClassIn(JobInitializationException.class)
-                .satisfies(
-                        e ->
-                                assertThat(
-                                                ExceptionUtils.findThrowable(
-                                                                e, t -> t.equals(expectedCause))
-                                                        .isPresent())
-                                        .isTrue());
+                .isInstanceOf(JobInitializationException.class)
+                .hasCause(expectedCause);
 
         assertThat(actualTimestamp)
                 .isGreaterThanOrEqualTo(expectedLowerTimestampThreshold)