Fix tests to clean up after themselves.

apache · Oct 26, 2023 · 6d0a3f7 · 6d0a3f7
1 parent c42078f
commit 6d0a3f7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/helix-core/src/test/java/org/apache/helix/integration/rebalancer/TestInstanceOperation.java b/helix-core/src/test/java/org/apache/helix/integration/rebalancer/TestInstanceOperation.java
@@ -405,6 +405,8 @@ public void testMarkEvacuationAfterEMM() throws Exception {
       Assert.assertTrue(newPAssignedParticipants.containsAll(currentActiveInstances));
     }
     Assert.assertTrue(_admin.isReadyForPreparingJoiningCluster(CLUSTER_NAME, instanceToEvacuate));
+
+    _stateModelDelay = 3L;
   }
 
   @Test(dependsOnMethods = "testMarkEvacuationAfterEMM")
@@ -441,6 +443,9 @@ public void testEvacuationWithOfflineInstancesInCluster() throws Exception {
         return true;
       }, 30000);
     }
+
+    resetInstances();
+    dropTestDBs(ImmutableSet.of("TEST_DB3_DELAYED_CRUSHED", "TEST_DB4_DELAYED_WAGED"));
   }
 
   @Test(expectedExceptions = HelixException.class, dependsOnMethods = "testEvacuationWithOfflineInstancesInCluster")
@@ -450,7 +455,6 @@ public void testAddingNodeWithSwapOutInstanceOperation() throws Exception {
             System.currentTimeMillis()));
 
     resetInstances();
-    dropTestDBs(ImmutableSet.of("TEST_DB3_DELAYED_CRUSHED", "TEST_DB4_DELAYED_WAGED"));
 
     // Set instance's InstanceOperation to SWAP_OUT
     String instanceToSwapOutName = _participants.get(0).getInstanceName();