modify test

Author: zcoo

fluss-server/src/test/java/org/apache/fluss/server/coordinator/CoordinatorServerTest.java

@@ -26,8 +26,10 @@
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 
+import java.time.Duration;
 import java.util.Optional;
 
+import static org.apache.fluss.testutils.common.CommonTestUtils.waitUntil;
 import static org.assertj.core.api.Assertions.assertThat;
 
 /** Test for {@link CoordinatorServer} . */
@@ -38,6 +40,7 @@ class CoordinatorServerTest extends ServerTestBase {
     @BeforeEach
     void beforeEach() throws Exception {
         coordinatorServer = startCoordinatorServer(createConfiguration());
+        waitUtilCoordinatorServerElected();
     }
 
     @AfterEach
@@ -71,4 +74,13 @@ protected void checkAfterStartServer() throws Exception {
                 optCoordinatorAddr.get().getEndpoints(),
                 coordinatorServer.getRpcServer().getBindEndpoints());
     }
+
+    public void waitUtilCoordinatorServerElected() {
+        waitUntil(
+                () -> {
+                    return zookeeperClient.getCoordinatorLeaderAddress().isPresent();
+                },
+                Duration.ofSeconds(10),
+                String.format("Fail to wait coordinator server reelected"));
+    }
 }