diff --git a/src/test/java/org/opensearch/securityanalytics/findings/FindingIT.java b/src/test/java/org/opensearch/securityanalytics/findings/FindingIT.java index 4615d7c09..ef6e9e5d7 100644 --- a/src/test/java/org/opensearch/securityanalytics/findings/FindingIT.java +++ b/src/test/java/org/opensearch/securityanalytics/findings/FindingIT.java @@ -369,7 +369,7 @@ public void testGetAllFindings_success() throws IOException { Map params = new HashMap<>(); Response getFindingsResponse = makeRequest(client(), "GET", SecurityAnalyticsPlugin.FINDINGS_BASE_URI + "/_search", params, null); Map getFindingsBody = entityAsMap(getFindingsResponse); - Assert.assertEquals(1, getFindingsBody.get("total_findings")); + Assert.assertEquals(2, getFindingsBody.get("total_findings")); } public void testGetFindings_byDetectionType_success() throws IOException { @@ -589,13 +589,13 @@ public void testGetFindings_bySeverity_success() throws IOException { params.put("severity", "high"); Response getFindingsResponse = makeRequest(client(), "GET", SecurityAnalyticsPlugin.FINDINGS_BASE_URI + "/_search", params, null); Map getFindingsBody = entityAsMap(getFindingsResponse); - Assert.assertEquals(1, getFindingsBody.get("total_findings")); + Assert.assertEquals(2, getFindingsBody.get("total_findings")); // Call GetFindings API for second detector by severity params.clear(); params.put("severity", "critical"); getFindingsResponse = makeRequest(client(), "GET", SecurityAnalyticsPlugin.FINDINGS_BASE_URI + "/_search", params, null); getFindingsBody = entityAsMap(getFindingsResponse); - Assert.assertEquals(1, getFindingsBody.get("total_findings")); + Assert.assertEquals(2, getFindingsBody.get("total_findings")); } public void testGetFindings_bySearchString_success() throws IOException { @@ -842,7 +842,7 @@ public void testGetFindings_byStartTimeAndEndTime_success() throws IOException { params.put("endTime", String.valueOf(endTime2.toEpochMilli())); getFindingsResponse = makeRequest(client(), "GET", SecurityAnalyticsPlugin.FINDINGS_BASE_URI + "/_search", params, null); getFindingsBody = entityAsMap(getFindingsResponse); - Assert.assertEquals(1, getFindingsBody.get("total_findings")); + Assert.assertEquals(2, getFindingsBody.get("total_findings")); } public void testGetFindings_rolloverByMaxAge_success() throws IOException, InterruptedException {