fixing unit test (#3052)

Signed-off-by: Dhrubo Saha <dhrubo@amazon.com>
opensearch-project · Oct 2, 2024 · 459f44d · 459f44d
1 parent 97cf307
commit 459f44d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/plugin/src/test/java/org/opensearch/ml/settings/MLFeatureEnabledSettingTests.java b/plugin/src/test/java/org/opensearch/ml/settings/MLFeatureEnabledSettingTests.java
@@ -10,6 +10,7 @@
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 import static org.opensearch.ml.settings.MLCommonsSettings.ML_COMMONS_AGENT_FRAMEWORK_ENABLED;
+import static org.opensearch.ml.settings.MLCommonsSettings.ML_COMMONS_CONNECTOR_PRIVATE_IP_ENABLED;
 import static org.opensearch.ml.settings.MLCommonsSettings.ML_COMMONS_CONTROLLER_ENABLED;
 import static org.opensearch.ml.settings.MLCommonsSettings.ML_COMMONS_LOCAL_MODEL_ENABLED;
 import static org.opensearch.ml.settings.MLCommonsSettings.ML_COMMONS_MULTI_TENANCY_ENABLED;
@@ -48,7 +49,8 @@ public void setUp() {
                             ML_COMMONS_REMOTE_INFERENCE_ENABLED,
                             ML_COMMONS_AGENT_FRAMEWORK_ENABLED,
                             ML_COMMONS_LOCAL_MODEL_ENABLED,
-                            ML_COMMONS_CONTROLLER_ENABLED
+                            ML_COMMONS_CONTROLLER_ENABLED,
+                            ML_COMMONS_CONNECTOR_PRIVATE_IP_ENABLED
                         )
                 )
             );