HDDS-11927 Fix possible flaky

apache · Dec 14, 2024 · 00642e9 · 00642e9
1 parent 2c00c34
commit 00642e9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...t/java/org/apache/hadoop/hdds/scm/container/balancer/TestContainerBalancerStatusInfo.java b/...t/java/org/apache/hadoop/hdds/scm/container/balancer/TestContainerBalancerStatusInfo.java
@@ -94,9 +94,10 @@ void testGetCurrentStatisticsRequestInPeriodBetweenIterations() throws Exception
     config.setMaxSizeToMovePerIteration(50 * OzoneConsts.GB);
 
     ContainerBalancerTask task = mockedScm.startBalancerTaskAsync(config, false);
-    LambdaTestUtils.await(1000, 500,
+    LambdaTestUtils.await(5000, 10,
         () -> task.getCurrentIterationsStatistic().size() == 1 &&
-              task.getCurrentIterationsStatistic().get(0).getIterationResult().equals("ITERATION_COMPLETED"));
+              task.getCurrentIterationsStatistic().get(0) != null &&
+              "ITERATION_COMPLETED".equals(task.getCurrentIterationsStatistic().get(0).getIterationResult()));
     List<ContainerBalancerTaskIterationStatusInfo> iterationsStatic = task.getCurrentIterationsStatistic();
     assertEquals(1, iterationsStatic.size());