tests fix

Signed-off-by: Riya Saxena <[email protected]>
riysaxen-amzn · Dec 17, 2024 · 0fce044 · 0fce044
1 parent 779afc3
commit 0fce044
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alerting/src/test/kotlin/org/opensearch/alerting/DocumentMonitorRunnerIT.kt b/alerting/src/test/kotlin/org/opensearch/alerting/DocumentMonitorRunnerIT.kt
@@ -2791,7 +2791,7 @@ class DocumentMonitorRunnerIT : AlertingRestTestCase() {
         indexDoc(aliasName, "6", testDoc)
         indexDoc(aliasName, "7", testDoc)
         OpenSearchTestCase.waitUntil(
-            { searchFindings(monitor).size == 6 }, 2, TimeUnit.MINUTES
+            { searchFindings(monitor).size == 6  && searchAlertsWithFilter(monitor).size == 1 }, 2, TimeUnit.MINUTES
         )
 
         rolloverDatastream(aliasName)
@@ -2802,7 +2802,7 @@ class DocumentMonitorRunnerIT : AlertingRestTestCase() {
         indexDoc(aliasName, "16", testDoc)
         indexDoc(aliasName, "17", testDoc)
         OpenSearchTestCase.waitUntil(
-            { searchFindings(monitor).size == 6 }, 2, TimeUnit.MINUTES
+            { searchFindings(monitor).size == 6  && searchAlertsWithFilter(monitor).size == 1 }, 2, TimeUnit.MINUTES
         )
 
         deleteDataStream(aliasName)