fix: improve test reliability and server initialization handling

Added a wait for HugeGraph server initialization in the install script and test cases to prevent race conditions. Enhanced teardown methods in JobApiTest and TaskApiTest to cancel running tasks before deletion.

 Updated TaskApiTest to clean up vertices before insertion, ensuring tests run in a clean state.

Author: imbajin

hugegraph-client/assembly/travis/install-hugegraph-from-source.sh

@@ -47,6 +47,10 @@ sed -i 's|#auth.admin_pa=.*|auth.admin_pa=pa|' conf/rest-server.properties
 echo -e "pa" | bin/init-store.sh || exit 1
 bin/start-hugegraph.sh || exit 1
 
+# Wait for server to initialize
+echo "Waiting 5 seconds for HugeGraph server to initialize..."
+sleep 5
+
 cd ../${HTTPS_SERVER_DIR}
 REST_SERVER_CONFIG="conf/rest-server.properties"
 GREMLIN_SERVER_CONFIG="conf/gremlin-server.yaml"

hugegraph-client/src/test/java/org/apache/hugegraph/api/BaseApiTest.java

@@ -128,7 +128,8 @@ protected static void clearData() {
         vertexLabelAPI.list().forEach(vertexLabel -> {
             vlTaskIds.add(vertexLabelAPI.delete(vertexLabel.name()));
         });
-        vlTaskIds.forEach(BaseApiTest::waitUntilTaskCompleted);
+        // Vertex label deletion may take longer, use extended timeout
+        vlTaskIds.forEach(taskId -> waitUntilTaskCompleted(taskId, 30));
 
         List<Long> pkTaskIds = new ArrayList<>();
         propertyKeyAPI.list().forEach(propertyKey -> {

hugegraph-client/src/test/java/org/apache/hugegraph/api/GraphsApiTest.java

@@ -46,6 +46,33 @@ public class GraphsApiTest extends BaseApiTest {
     private static final String GRAPH3 = "hugegraph3";
     private static final String CONFIG3_PATH = "src/test/resources/hugegraph-clone.properties";
 
+    private static final int MAX_WAIT_SECONDS = 30;
+    private static final int RETRY_INTERVAL_MS = 1000;
+
+    /**
+     * Wait for HugeGraph server to be ready before running tests
+     */
+    private static void waitForServerReady() {
+        System.out.println("Waiting for HugeGraph server to be ready...");
+        for (int i = 0; i < MAX_WAIT_SECONDS; i++) {
+            try {
+                // Try to get graph list, if successful, server is ready
+                graphsAPI.list();
+                System.out.println("Server is ready after " + i + " seconds");
+                return;
+            } catch (Exception e) {
+                System.out.println("Waiting for server... (" + (i + 1) + "/" + MAX_WAIT_SECONDS + ")");
+                try {
+                    Thread.sleep(RETRY_INTERVAL_MS);
+                } catch (InterruptedException ie) {
+                    Thread.currentThread().interrupt();
+                    throw new RuntimeException("Interrupted while waiting for server", ie);
+                }
+            }
+        }
+        throw new RuntimeException("Server not ready after " + MAX_WAIT_SECONDS + " seconds");
+    }
+
     protected static void initPropertyKey(HugeClient client) {
         SchemaManager schema = client.schema();
         schema.propertyKey("name").asText().ifNotExist().create();
@@ -119,6 +146,7 @@ public void teardown() {
 
     @Test
     public void testCreateAndDropGraph() {
+        waitForServerReady();
         int initialGraphNumber = graphsAPI.list().size();
 
         // Create new graph dynamically
@@ -190,6 +218,7 @@ public void testCreateAndDropGraph() {
 
     @Test
     public void testCloneAndDropGraph() {
+        waitForServerReady();
         int initialGraphNumber = graphsAPI.list().size();
 
         // Clone a new graph from exist a graph dynamically
@@ -262,6 +291,7 @@ public void testCloneAndDropGraph() {
 
     @Test
     public void testCloneAndDropGraphWithoutConfig() {
+        waitForServerReady();
         int initialGraphNumber = graphsAPI.list().size();
 
         // Clone a new graph from exist a graph dynamically

hugegraph-client/src/test/java/org/apache/hugegraph/api/JobApiTest.java

@@ -38,7 +38,19 @@ public static void prepareSchema() {
 
     @After
     public void teardown() throws Exception {
-        taskAPI.list(null, -1).forEach(task -> taskAPI.delete(task.id()));
+        taskAPI.list(null, -1).forEach(task -> {
+            // Cancel running/cancelling tasks before deletion
+            if (!task.completed()) {
+                try {
+                    taskAPI.cancel(task.id());
+                    // Wait for cancellation to complete
+                    Thread.sleep(1000);
+                } catch (Exception e) {
+                    // Ignore if already completed or cancellation fails
+                }
+            }
+            taskAPI.delete(task.id());
+        });
     }
 
     @Test

hugegraph-client/src/test/java/org/apache/hugegraph/api/TaskApiTest.java

@@ -46,7 +46,19 @@ public static void prepareSchema() {
 
     @After
     public void teardown() throws Exception {
-        taskAPI.list(null, -1).forEach(task -> taskAPI.delete(task.id()));
+        taskAPI.list(null, -1).forEach(task -> {
+            // Cancel running/cancelling tasks before deletion
+            if (!task.completed()) {
+                try {
+                    taskAPI.cancel(task.id());
+                    // Wait for cancellation to complete
+                    Thread.sleep(1000);
+                } catch (Exception e) {
+                    // Ignore if already completed or cancellation fails
+                }
+            }
+            taskAPI.delete(task.id());
+        });
     }
 
     @Test
@@ -208,12 +220,17 @@ public void testDelete() {
     public void testCancel() {
         schema().vertexLabel("man").useAutomaticId().ifNotExist().create();
 
-        String groovy = "for (int i = 0; i < 10; i++) {" +
-                        "g.addV('man').iterate();" +
-                        "}";
-        // Insert 10 records in sync mode
+        // Clean up any existing 'man' vertices from previous tests
+        String groovy = "g.V().hasLabel('man').drop()";
         GremlinRequest request = new GremlinRequest(groovy);
         gremlin().execute(request);
+
+        groovy = "for (int i = 0; i < 10; i++) {" +
+                 "g.addV('man').iterate();" +
+                 "}";
+        // Insert 10 records in sync mode
+        request = new GremlinRequest(groovy);
+        gremlin().execute(request);
         // Verify insertion takes effect
         groovy = "g.V()";
         request = new GremlinRequest(groovy);