fix flaky multi-node tests

Signed-off-by: Subhobrata Dey <sbcd90@gmail.com>
opensearch-project · Apr 18, 2024 · 68c40e1 · 68c40e1
1 parent d7562b1
commit 68c40e1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/...ing/src/main/kotlin/org/opensearch/alerting/transport/TransportGetWorkflowAlertsAction.kt b/...ing/src/main/kotlin/org/opensearch/alerting/transport/TransportGetWorkflowAlertsAction.kt
@@ -251,7 +251,7 @@ class TransportGetWorkflowAlertsAction @Inject constructor(
                 searchRequest.source().sort(sortBuilder).size(tableProp.size).from(tableProp.startIndex)
             }
             queryBuilder.must(QueryBuilders.termsQuery("_id", associatedAlertIds))
-            queryBuilder.must(QueryBuilders.termQuery(Alert.STATE_FIELD, Alert.State.AUDIT))
+            queryBuilder.must(QueryBuilders.termQuery(Alert.STATE_FIELD, Alert.State.AUDIT.name))
             searchRequest.source().query(queryBuilder)
             val response: SearchResponse = client.suspendUntil { search(searchRequest, it) }
             associatedAlerts.addAll(parseAlertsFromSearchResponse(response))

diff --git a/alerting/src/main/kotlin/org/opensearch/alerting/workflow/CompositeWorkflowRunner.kt b/alerting/src/main/kotlin/org/opensearch/alerting/workflow/CompositeWorkflowRunner.kt
@@ -402,9 +402,9 @@ object CompositeWorkflowRunner : WorkflowRunner() {
 
     fun getDelegateMonitorAlertState(
         workflow: Workflow,
-    ): Alert.State {
+    ): String {
         return if (workflow.triggers.isNotEmpty()) {
-            Alert.State.AUDIT
-        } else Alert.State.ACTIVE
+            Alert.State.AUDIT.name
+        } else Alert.State.ACTIVE.name
     }
 }