improve code

apache · Nov 17, 2021 · 5844955 · 5844955
1 parent df61a96
commit 5844955
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/computer-test/src/main/java/com/baidu/hugegraph/computer/k8s/MiniKubeTest.java b/computer-test/src/main/java/com/baidu/hugegraph/computer/k8s/MiniKubeTest.java
@@ -123,7 +123,7 @@ public void testJobSucceed() {
         Mockito.verify(jobObserver, Mockito.timeout(15000L).atLeast(1))
                .onJobStateChanged(Mockito.eq(jobState2));
 
-        future.getNow(null);
+        future.cancel(true);
     }
 
     @Test
@@ -154,7 +154,7 @@ public void testJobInternalSucceed() {
         Mockito.verify(jobObserver, Mockito.timeout(15000L).atLeast(1))
                .onJobStateChanged(Mockito.eq(jobState2));
 
-        future.getNow(null);
+        future.cancel(true);
     }
 
     @Test
@@ -192,7 +192,7 @@ public void testJobFailed() {
         String diagnostics = this.driver.diagnostics(jobId, params);
         Assert.assertContains("No such file or directory", diagnostics);
 
-        future.getNow(null);
+        future.cancel(true);
     }
 
     @Test
@@ -216,7 +216,7 @@ public void testUnSchedulable() {
         String diagnostics = this.driver.diagnostics(jobId, params);
         Assert.assertContains("Unschedulable", diagnostics);
 
-        future.getNow(null);
+        future.cancel(true);
     }
 
     @Test
@@ -251,7 +251,7 @@ public void testJobCancelled() {
         Mockito.verify(jobObserver, Mockito.timeout(15000L).atLeast(1))
                .onJobStateChanged(Mockito.eq(jobState2));
 
-        future.getNow(null);
+        future.cancel(true);
     }
 
     @Test
@@ -291,7 +291,7 @@ public void testTwiceCreate() {
 
         UnitTestBase.sleep(1000L);
 
-        future.getNow(null);
+        future.cancel(true);
     }
 
     @Test
@@ -314,7 +314,7 @@ public void testPullImageError() {
         String diagnostics = this.driver.diagnostics(jobId, params);
         Assert.assertContains("ImagePullBackOff", diagnostics);
 
-        future.getNow(null);
+        future.cancel(true);
     }
 
     @Test
@@ -348,6 +348,6 @@ public void testGetResourceListWithLabels() {
                                          new HashMap<String, String>());
         Assert.assertNotEquals(0, pods.size());
 
-        future.getNow(null);
+        future.cancel(true);
     }
 }