Fix compileTestsJava

Signed-off-by: Darshit Chanpura <dchanp@amazon.com>

Author: DarshitChanpura

plugin/src/test/java/org/opensearch/ml/action/controller/CreateControllerTransportActionTests.java

@@ -149,6 +149,7 @@ public void setup() throws IOException {
                 actionFilters,
                 mlIndicesHandler,
                 client,
+                Settings.EMPTY,
                 clusterService,
                 modelAccessControlHelper,
                 mlModelCacheHelper,

plugin/src/test/java/org/opensearch/ml/action/controller/DeleteControllerTransportActionTests.java

@@ -147,6 +147,7 @@ public void setup() throws IOException {
                 transportService,
                 actionFilters,
                 client,
+                Settings.EMPTY,
                 xContentRegistry,
                 clusterService,
                 mlModelManager,

plugin/src/test/java/org/opensearch/ml/action/controller/GetControllerTransportActionTests.java

@@ -103,6 +103,7 @@ public void setup() throws IOException {
                 transportService,
                 actionFilters,
                 client,
+                settings,
                 xContentRegistry,
                 clusterService,
                 mlModelManager,

plugin/src/test/java/org/opensearch/ml/action/controller/UpdateControllerTransportActionTests.java

@@ -151,6 +151,7 @@ public void setup() throws IOException {
                 transportService,
                 actionFilters,
                 client,
+                Settings.EMPTY,
                 clusterService,
                 modelAccessControlHelper,
                 mlModelCacheHelper,

plugin/src/test/java/org/opensearch/ml/action/model_group/SearchModelGroupTransportActionTests.java

@@ -97,6 +97,7 @@ public void setup() {
             transportService,
             actionFilters,
             client,
+            Settings.EMPTY,
             sdkClient,
             clusterService,
             modelAccessControlHelper,

plugin/src/test/java/org/opensearch/ml/action/models/SearchModelTransportActionTests.java

@@ -114,7 +114,7 @@ public class SearchModelTransportActionTests extends OpenSearchTestCase {
     public void setup() {
         MockitoAnnotations.openMocks(this);
         sdkClient = SdkClientFactory.createSdkClient(client, NamedXContentRegistry.EMPTY, Collections.emptyMap());
-        mlSearchHandler = spy(new MLSearchHandler(client, namedXContentRegistry, modelAccessControlHelper, clusterService));
+        mlSearchHandler = spy(new MLSearchHandler(client, namedXContentRegistry, modelAccessControlHelper, clusterService, Settings.EMPTY));
         searchModelTransportAction = new SearchModelTransportAction(
             transportService,
             actionFilters,
@@ -184,7 +184,7 @@ public void test_DoExecute_addBackendRoles() throws IOException {
             listener.onResponse(searchResponse);
             return null;
         }).when(client).search(any(), any());
-        when(modelAccessControlHelper.createSearchSourceBuilder(any())).thenReturn(searchSourceBuilder);
+        when(modelAccessControlHelper.createSearchSourceBuilder(any(), Settings.EMPTY)).thenReturn(searchSourceBuilder);
         searchModelTransportAction.doExecute(null, mlSearchActionRequest, actionListener);
         verify(mlSearchHandler).search(sdkClient, mlSearchActionRequest, null, actionListener);
         verify(client, times(2)).search(any(), any());
@@ -196,7 +196,7 @@ public void test_DoExecute_addBackendRoles_without_groupIds() {
             listener.onResponse(searchResponse);
             return null;
         }).when(client).search(any(), isA(ActionListener.class));
-        when(modelAccessControlHelper.createSearchSourceBuilder(any())).thenReturn(searchSourceBuilder);
+        when(modelAccessControlHelper.createSearchSourceBuilder(any(), Settings.EMPTY)).thenReturn(searchSourceBuilder);
         searchModelTransportAction.doExecute(null, mlSearchActionRequest, actionListener);
         verify(mlSearchHandler).search(sdkClient, mlSearchActionRequest, null, actionListener);
         verify(client, times(2)).search(any(), any());
@@ -208,7 +208,7 @@ public void test_DoExecute_addBackendRoles_exception() {
             listener.onFailure(new RuntimeException("runtime exception"));
             return null;
         }).when(client).search(any(), isA(ActionListener.class));
-        when(modelAccessControlHelper.createSearchSourceBuilder(any())).thenReturn(searchSourceBuilder);
+        when(modelAccessControlHelper.createSearchSourceBuilder(any(), Settings.EMPTY)).thenReturn(searchSourceBuilder);
         searchModelTransportAction.doExecute(null, mlSearchActionRequest, actionListener);
         verify(mlSearchHandler).search(sdkClient, mlSearchActionRequest, null, actionListener);
         verify(client, times(1)).search(any(), any());
@@ -281,7 +281,7 @@ public void test_DoExecute_addBackendRoles_boolQuery() throws IOException {
             listener.onResponse(searchResponse);
             return null;
         }).when(client).search(any(), isA(ActionListener.class));
-        when(modelAccessControlHelper.createSearchSourceBuilder(any())).thenReturn(searchSourceBuilder);
+        when(modelAccessControlHelper.createSearchSourceBuilder(any(), Settings.EMPTY)).thenReturn(searchSourceBuilder);
         searchRequest.source().query(QueryBuilders.boolQuery().must(QueryBuilders.matchQuery("name", "model_IT")));
         searchModelTransportAction.doExecute(null, mlSearchActionRequest, actionListener);
         verify(mlSearchHandler).search(sdkClient, mlSearchActionRequest, null, actionListener);
@@ -295,7 +295,7 @@ public void test_DoExecute_addBackendRoles_termQuery() throws IOException {
             listener.onResponse(searchResponse);
             return null;
         }).when(client).search(any(), isA(ActionListener.class));
-        when(modelAccessControlHelper.createSearchSourceBuilder(any())).thenReturn(searchSourceBuilder);
+        when(modelAccessControlHelper.createSearchSourceBuilder(any(), Settings.EMPTY)).thenReturn(searchSourceBuilder);
         searchRequest.source().query(QueryBuilders.termQuery("name", "model_IT"));
         searchModelTransportAction.doExecute(null, mlSearchActionRequest, actionListener);
         verify(mlSearchHandler).search(sdkClient, mlSearchActionRequest, null, actionListener);
@@ -330,7 +330,7 @@ public void testDoExecute_MultiTenancyEnabled_TenantFilteringEnabled() throws In
             return null;
         }).when(client).search(any(), any());
 
-        when(modelAccessControlHelper.createSearchSourceBuilder(any())).thenReturn(searchSourceBuilder);
+        when(modelAccessControlHelper.createSearchSourceBuilder(any(), Settings.EMPTY)).thenReturn(searchSourceBuilder);
         searchRequest.source().query(QueryBuilders.termQuery("name", "model_IT"));
         mlSearchActionRequest = new MLSearchActionRequest(searchRequest, "123456");
 

plugin/src/test/java/org/opensearch/ml/action/tasks/CancelBatchJobTransportActionTests.java

@@ -140,6 +140,7 @@ public void setup() throws IOException {
                 transportService,
                 actionFilters,
                 client,
+                settings,
                 xContentRegistry,
                 clusterService,
                 scriptService,

plugin/src/test/java/org/opensearch/ml/action/upload_chunk/MLModelChunkUploaderTests.java

@@ -117,7 +117,7 @@ public void setup() throws IOException {
 
         threadContext.putTransient(ConfigConstants.OPENSEARCH_SECURITY_USER_INFO_THREAD_CONTEXT, "alex|IT,HR|engineering,operations");
 
-        mlModelChunkUploader = new MLModelChunkUploader(mlIndicesHandler, client, xContentRegistry, modelAccessControlHelper);
+        mlModelChunkUploader = new MLModelChunkUploader(mlIndicesHandler, client, settings, xContentRegistry, modelAccessControlHelper);
 
         MLModel mlModel = MLModel
             .builder()

plugin/src/test/java/org/opensearch/ml/action/upload_chunk/TransportRegisterModelMetaActionTests.java

@@ -86,6 +86,7 @@ public void setup() throws IOException {
             actionFilters,
             mlModelManager,
             client,
+            settings,
             modelAccessControlHelper,
             mlModelGroupManager
         );

plugin/src/test/java/org/opensearch/ml/helper/ModelAccessControlHelperTests.java

@@ -346,7 +346,16 @@ public void test_PrivateModelGroupWithDifferentOwner() throws IOException, Inter
         CountDownLatch latch = new CountDownLatch(1);
         LatchedActionListener<Boolean> latchedActionListener = new LatchedActionListener<>(actionListener, latch);
         modelAccessControlHelper
-            .validateModelGroupAccess(user, mlFeatureEnabledSetting, null, "testGroupID", client, sdkClient, latchedActionListener);
+            .validateModelGroupAccess(
+                user,
+                mlFeatureEnabledSetting,
+                null,
+                "testGroupID",
+                client,
+                sdkClient,
+                Settings.EMPTY,
+                latchedActionListener
+            );
         latch.await(500, TimeUnit.MILLISECONDS);
 
         ArgumentCaptor<Boolean> argumentCaptor = ArgumentCaptor.forClass(Boolean.class);
@@ -461,7 +470,7 @@ public void test_AddUserBackendRolesFilter() {
 
     public void test_CreateSearchSourceBuilder() {
         User user = User.parse("owner|IT,HR|myTenant");
-        assertNotNull(modelAccessControlHelper.createSearchSourceBuilder(user));
+        assertNotNull(modelAccessControlHelper.createSearchSourceBuilder(user, Settings.EMPTY));
     }
 
     private GetResponse modelGroupBuilder(List<String> backendRoles, String access, String owner) throws IOException {