From 0bf9fd12984efe96a26f1a81df8d533d13ffddda Mon Sep 17 00:00:00 2001
From: Darshit Chanpura <dchanp@amazon.com>
Date: Wed, 18 Dec 2024 17:57:51 -0500
Subject: [PATCH] Completes the integrations tests

Signed-off-by: Darshit Chanpura <dchanp@amazon.com>
---
 .../DefaultResourceAccessControlPluginIT.java | 62 ++++++++++++++++++-
 1 file changed, 61 insertions(+), 1 deletion(-)

diff --git a/server/src/internalClusterTest/java/org/opensearch/accesscontrol/resources/fallback/DefaultResourceAccessControlPluginIT.java b/server/src/internalClusterTest/java/org/opensearch/accesscontrol/resources/fallback/DefaultResourceAccessControlPluginIT.java
index 0848b9fb48f38..63d23ba9adc27 100644
--- a/server/src/internalClusterTest/java/org/opensearch/accesscontrol/resources/fallback/DefaultResourceAccessControlPluginIT.java
+++ b/server/src/internalClusterTest/java/org/opensearch/accesscontrol/resources/fallback/DefaultResourceAccessControlPluginIT.java
@@ -8,6 +8,10 @@
 
 package org.opensearch.accesscontrol.resources.fallback;
 
+import org.opensearch.accesscontrol.resources.EntityType;
+import org.opensearch.accesscontrol.resources.ResourceSharing;
+import org.opensearch.accesscontrol.resources.ShareWith;
+import org.opensearch.accesscontrol.resources.SharedWithScope;
 import org.opensearch.client.Client;
 import org.opensearch.core.xcontent.XContentBuilder;
 import org.opensearch.plugins.Plugin;
@@ -19,6 +23,7 @@
 import java.io.IOException;
 import java.util.Collection;
 import java.util.List;
+import java.util.Map;
 import java.util.Set;
 
 import static org.opensearch.accesscontrol.resources.fallback.TestResourcePlugin.SAMPLE_TEST_INDEX;
@@ -27,6 +32,7 @@
 import static org.hamcrest.Matchers.hasProperty;
 import static org.hamcrest.Matchers.hasSize;
 import static org.hamcrest.Matchers.is;
+import static org.hamcrest.Matchers.nullValue;
 
 public class DefaultResourceAccessControlPluginIT extends OpenSearchIntegTestCase {
     @Override
@@ -57,7 +63,7 @@ public void testGetResources() throws IOException {
         }
     }
 
-    public void testSampleResourcePluginCallsDefaultRACPlugin() throws IOException {
+    public void testSampleResourcePluginListResources() throws IOException {
         createIndex(SAMPLE_TEST_INDEX);
         indexSampleDocuments();
 
@@ -75,6 +81,60 @@ public void testSampleResourcePluginCallsDefaultRACPlugin() throws IOException {
         MatcherAssert.assertThat(resources, hasItem(hasProperty("id", is("2"))));
     }
 
+    public void testSampleResourcePluginCallsHasPermission() {
+
+        ResourceAccessControlPlugin racPlugin = TestResourcePlugin.GuiceHolder.getResourceService().getResourceAccessControlPlugin();
+        MatcherAssert.assertThat(racPlugin.getClass(), is(DefaultResourceAccessControlPlugin.class));
+
+        boolean canAccess = racPlugin.hasPermission("1", SAMPLE_TEST_INDEX, "some_scope");
+
+        MatcherAssert.assertThat(canAccess, is(true));
+
+    }
+
+    public void testSampleResourcePluginCallsShareWith() {
+
+        ResourceAccessControlPlugin racPlugin = TestResourcePlugin.GuiceHolder.getResourceService().getResourceAccessControlPlugin();
+        MatcherAssert.assertThat(racPlugin.getClass(), is(DefaultResourceAccessControlPlugin.class));
+
+        SharedWithScope.SharedWithPerScope sharedWithPerScope = new SharedWithScope.SharedWithPerScope(Set.of(), Set.of(), Set.of());
+        SharedWithScope sharedWithScope = new SharedWithScope("some_scope", sharedWithPerScope);
+        ResourceSharing sharingInfo = racPlugin.shareWith("1", SAMPLE_TEST_INDEX, new ShareWith(Set.of(sharedWithScope)));
+
+        MatcherAssert.assertThat(sharingInfo, is(nullValue()));
+    }
+
+    public void testSampleResourcePluginCallsRevokeAccess() {
+
+        ResourceAccessControlPlugin racPlugin = TestResourcePlugin.GuiceHolder.getResourceService().getResourceAccessControlPlugin();
+        MatcherAssert.assertThat(racPlugin.getClass(), is(DefaultResourceAccessControlPlugin.class));
+
+        Map<EntityType, Set<String>> entityTypes = Map.of(EntityType.USERS, Set.of("some_user"));
+        ResourceSharing sharingInfo = racPlugin.revokeAccess("1", SAMPLE_TEST_INDEX, entityTypes, Set.of("some_scope"));
+
+        MatcherAssert.assertThat(sharingInfo, is(nullValue()));
+    }
+
+    public void testSampleResourcePluginCallsDeleteResourceSharingRecord() {
+        ResourceAccessControlPlugin racPlugin = TestResourcePlugin.GuiceHolder.getResourceService().getResourceAccessControlPlugin();
+        MatcherAssert.assertThat(racPlugin.getClass(), is(DefaultResourceAccessControlPlugin.class));
+
+        boolean recordDeleted = racPlugin.deleteResourceSharingRecord("1", SAMPLE_TEST_INDEX);
+
+        // no record to delete
+        MatcherAssert.assertThat(recordDeleted, is(false));
+    }
+
+    public void testSampleResourcePluginCallsDeleteAllResourceSharingRecordsForCurrentUser() {
+        ResourceAccessControlPlugin racPlugin = TestResourcePlugin.GuiceHolder.getResourceService().getResourceAccessControlPlugin();
+        MatcherAssert.assertThat(racPlugin.getClass(), is(DefaultResourceAccessControlPlugin.class));
+
+        boolean recordDeleted = racPlugin.deleteAllResourceSharingRecordsForCurrentUser();
+
+        // no records to delete
+        MatcherAssert.assertThat(recordDeleted, is(false));
+    }
+
     private void indexSampleDocuments() throws IOException {
         XContentBuilder doc1 = jsonBuilder().startObject().field("id", "1").field("name", "Test Document 1").endObject();