[hotfix] Replace AskTimeException usages in tests

Author: zentol

flink-runtime/src/test/java/org/apache/flink/runtime/resourcemanager/slotmanager/DeclarativeSlotManagerTest.java

@@ -56,7 +56,6 @@
 
 import org.apache.flink.shaded.guava18.com.google.common.collect.Iterators;
 
-import akka.pattern.AskTimeoutException;
 import org.junit.Test;
 
 import java.util.ArrayList;
@@ -602,7 +601,7 @@ public void testSlotAllocationTimeout() throws Exception {
         final BlockingQueue<Supplier<CompletableFuture<Acknowledge>>> responseQueue =
                 new ArrayBlockingQueue<>(2);
         responseQueue.add(
-                () -> FutureUtils.completedExceptionally(new AskTimeoutException("timeout")));
+                () -> FutureUtils.completedExceptionally(new TimeoutException("timeout")));
         responseQueue.add(
                 () -> {
                     secondSlotRequestFuture.complete(null);

flink-runtime/src/test/java/org/apache/flink/runtime/resourcemanager/slotmanager/DefaultSlotStatusSyncerTest.java

@@ -36,11 +36,11 @@
 import org.apache.flink.util.TestLogger;
 import org.apache.flink.util.concurrent.FutureUtils;
 
-import akka.pattern.AskTimeoutException;
 import org.junit.Test;
 
 import java.util.Arrays;
 import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.TimeoutException;
 
 import static org.hamcrest.Matchers.contains;
 import static org.hamcrest.Matchers.empty;
@@ -126,7 +126,7 @@ public void testAllocateSlotFailsWithException() {
                         .setRequestSlotFunction(
                                 ignored ->
                                         FutureUtils.completedExceptionally(
-                                                new AskTimeoutException("timeout")))
+                                                new TimeoutException("timeout")))
                         .createTestingTaskExecutorGateway();
         final TaskExecutorConnection taskExecutorConnection =
                 new TaskExecutorConnection(ResourceID.generate(), taskExecutorGateway);
@@ -151,7 +151,7 @@ public void testAllocateSlotFailsWithException() {
         try {
             allocatedFuture.get();
         } catch (Exception e) {
-            assertThat(e.getCause(), instanceOf(AskTimeoutException.class));
+            assertThat(e.getCause(), instanceOf(TimeoutException.class));
         }
         assertThat(resourceTracker.getAcquiredResources(jobId), is(empty()));
         assertThat(