(paimon) branch master updated: [test] Fix unstable KvQueryTableTest.testServerRestartSamePorts test method. (#3735)

[lzljs3620320]

This is an automated email from the ASF dual-hosted git repository.

lzljs3620320 pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/paimon.git


The following commit(s) were added to refs/heads/master by this push:
     new 319ce7811 [test] Fix unstable 
KvQueryTableTest.testServerRestartSamePorts test method. (#3735)
319ce7811 is described below

commit 319ce7811bc4b7f88a09986e022475f16e70cdaf
Author: Kerwin <37063904+zhuangch...@users.noreply.github.com>
AuthorDate: Fri Jul 12 15:14:37 2024 +0800

    [test] Fix unstable KvQueryTableTest.testServerRestartSamePorts test 
method. (#3735)
---
 .../java/org/apache/paimon/service/KvQueryTableTest.java | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git 
a/paimon-service/paimon-service-runtime/src/test/java/org/apache/paimon/service/KvQueryTableTest.java
 
b/paimon-service/paimon-service-runtime/src/test/java/org/apache/paimon/service/KvQueryTableTest.java
index 72691a41d..d15247f9a 100644
--- 
a/paimon-service/paimon-service-runtime/src/test/java/org/apache/paimon/service/KvQueryTableTest.java
+++ 
b/paimon-service/paimon-service-runtime/src/test/java/org/apache/paimon/service/KvQueryTableTest.java
@@ -106,19 +106,27 @@ public class KvQueryTableTest extends 
PrimaryKeyTableTestBase {
     }
 
     @AfterEach
-    public void afterEach() {
+    public void afterEach() throws Exception {
         shutdownServers();
         if (client != null) {
-            client.shutdown();
+            client.shutdownFuture().get();
         }
     }
 
     private void shutdownServers() {
         if (server0 != null) {
-            server0.shutdown();
+            try {
+                server0.shutdownServer().get();
+            } catch (Exception e) {
+                throw new RuntimeException(e);
+            }
         }
         if (server1 != null) {
-            server1.shutdown();
+            try {
+                server1.shutdownServer().get();
+            } catch (Exception e) {
+                throw new RuntimeException(e);
+            }
         }
     }