[FLINK-24584][qs] Improve assertion

gimral · Oct 19, 2021 · 4749a77 · 4749a77
1 parent f75cf38
commit 4749a77
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/...yable-state-runtime/src/test/java/org/apache/flink/queryablestate/network/ClientTest.java b/...yable-state-runtime/src/test/java/org/apache/flink/queryablestate/network/ClientTest.java
@@ -73,6 +73,7 @@
 import java.net.InetSocketAddress;
 import java.net.UnknownHostException;
 import java.nio.channels.ClosedChannelException;
+import java.time.Duration;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
@@ -266,15 +267,10 @@ public void testRequestUnavailableHost() throws Exception {
                     new KvStateInternalRequest(new KvStateID(), new byte[0]);
             CompletableFuture<KvStateResponse> future = client.sendRequest(serverAddress, request);
 
-            try {
-                future.get();
-                fail("Did not throw expected ConnectException");
-            } catch (ExecutionException e) {
-                if (!(e.getCause() instanceof ConnectException)) {
-                    fail("Did not throw expected ConnectException");
-                }
-                // else expected
-            }
+            assertThat(
+                    future,
+                    FlinkMatchers.futureWillCompleteExceptionally(
+                            ConnectException.class, Duration.ofHours(1)));
         } finally {
             if (client != null) {
                 try {