From 79a1a82d9b6a644ea51418953e6383ff9254322b Mon Sep 17 00:00:00 2001
From: Petar Dzepina <petar.dzepina@gmail.com>
Date: Wed, 28 Jun 2023 21:15:44 +0200
Subject: [PATCH] added more asserts

Signed-off-by: Petar Dzepina <petar.dzepina@gmail.com>
---
 .../org/opensearch/alerting/MonitorRunnerServiceIT.kt  | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/alerting/src/test/kotlin/org/opensearch/alerting/MonitorRunnerServiceIT.kt b/alerting/src/test/kotlin/org/opensearch/alerting/MonitorRunnerServiceIT.kt
index 3155bae81..329cb6130 100644
--- a/alerting/src/test/kotlin/org/opensearch/alerting/MonitorRunnerServiceIT.kt
+++ b/alerting/src/test/kotlin/org/opensearch/alerting/MonitorRunnerServiceIT.kt
@@ -1196,7 +1196,7 @@ class MonitorRunnerServiceIT : AlertingRestTestCase() {
         indexDoc(
             index,
             "2",
-            """{"user_id": "1",
+            """{"user_id": "2",
             "ip_addr": "12345678",
             "user_agent": "chrome"
             }
@@ -1255,6 +1255,14 @@ class MonitorRunnerServiceIT : AlertingRestTestCase() {
         @Suppress("UNCHECKED_CAST")
         val buckets = searchResult.stringMap("aggregations")?.stringMap("hot")?.get("buckets") as List<Map<String, Any>>
         assertEquals("Incorrect search result", 2, buckets.size)
+        val distinctUserCountAgg1 = buckets.find {
+            it.get("key_as_string") == "12345678|chrome"
+        }!!.get("distinct_user_count") as Map<String, Integer>
+        assertEquals(2, distinctUserCountAgg1.get("value"))
+        val distinctUserCountAgg2 = buckets.find {
+            it.get("key_as_string") == "3443534|chrome"
+        }!!.get("distinct_user_count") as Map<String, Integer>
+        assertEquals(1, distinctUserCountAgg2.get("value"))
     }
 
     fun `test bucket-level monitor alert creation and completion`() {