From 95a81251085a18300466256211e953167324f847 Mon Sep 17 00:00:00 2001 From: Riya Saxena Date: Wed, 11 Sep 2024 11:57:15 -0700 Subject: [PATCH] added secure correlation alerts tests Signed-off-by: Riya Saxena --- .../SecureCorrelationAlertsRestAPIT.java | 34 +++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 src/test/java/org/opensearch/securityanalytics/correlation/alerts/SecureCorrelationAlertsRestAPIT.java diff --git a/src/test/java/org/opensearch/securityanalytics/correlation/alerts/SecureCorrelationAlertsRestAPIT.java b/src/test/java/org/opensearch/securityanalytics/correlation/alerts/SecureCorrelationAlertsRestAPIT.java new file mode 100644 index 000000000..817662248 --- /dev/null +++ b/src/test/java/org/opensearch/securityanalytics/correlation/alerts/SecureCorrelationAlertsRestAPIT.java @@ -0,0 +1,34 @@ +package org.opensearch.securityanalytics.correlation.alerts; + +import org.apache.hc.core5.http.HttpHost; +import org.junit.After; +import org.junit.Before; +import org.opensearch.client.RestClient; +import org.opensearch.commons.rest.SecureRestClientBuilder; +import org.opensearch.securityanalytics.SecurityAnalyticsRestTestCase; +import java.io.IOException; + +public class SecureCorrelationAlertsRestAPIT { + static String SECURITY_ANALYTICS_FULL_ACCESS_ROLE = "security_analytics_full_access"; + static String SECURITY_ANALYTICS_READ_ACCESS_ROLE = "security_analytics_read_access"; + static String TEST_HR_BACKEND_ROLE = "HR"; + static String TEST_IT_BACKEND_ROLE = "IT"; + private final String user = "userAlert"; + private static final String[] EMPTY_ARRAY = new String[0]; + private RestClient userClient; + + @Before + public void create() throws IOException { + String[] backendRoles = { TEST_HR_BACKEND_ROLE }; + createUserWithData(user, user, SECURITY_ANALYTICS_FULL_ACCESS_ROLE, backendRoles ); + if (userClient == null) { + userClient = new SecureRestClientBuilder(getClusterHosts().toArray(new HttpHost[]{}), isHttps(), user, password).setSocketTimeout(60000).build(); + } + } + + @After + public void cleanup() throws IOException { + userClient.close(); + deleteUser(user); + } +}