Fix flaky testClusterStateBatchedUpdates test

Signed-off-by: Aman Khare <[email protected]>
amkhar · Oct 25, 2023 · afc5919 · afc5919
1 parent 3e64a7b
commit afc5919
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/server/src/test/java/org/opensearch/cluster/service/MasterServiceTests.java b/server/src/test/java/org/opensearch/cluster/service/MasterServiceTests.java
@@ -487,6 +487,9 @@ public void onFailure(String source, Exception e) {
                     }
                 });
                 assertBusy(mockAppender::assertAllExpectationsMatched);
+                // verify stats values after state is published
+                assertEquals(1, clusterManagerService.getClusterStateStats().getUpdateSuccess());
+                assertEquals(0, clusterManagerService.getClusterStateStats().getUpdateFailed());
             }
         }
     }
@@ -691,9 +694,6 @@ public void clusterStateProcessed(String source, ClusterState oldState, ClusterS
                     submittedTasksPerThread.get(entry.getKey()).get()
                 );
             }
-            // verify stats values after state is published
-            assertEquals(1, clusterManagerService.getClusterStateStats().getUpdateSuccess());
-            assertEquals(0, clusterManagerService.getClusterStateStats().getUpdateFailed());
         }
     }