From 5c75320d4dfabd3a330a51b7235267e8ab0bc651 Mon Sep 17 00:00:00 2001 From: Joanne Wang Date: Fri, 15 Mar 2024 15:18:25 -0700 Subject: [PATCH] fix integ tests for findings api Signed-off-by: Joanne Wang --- .../opensearch/securityanalytics/findings/FindingIT.java | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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 {