Fix UT

Signed-off-by: b4sjoo <[email protected]>
opensearch-project · Sep 18, 2024 · c4a41cf · c4a41cf
1 parent 6edda9b
commit c4a41cf
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_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;
 import static org.opensearch.ml.settings.MLCommonsSettings.ML_COMMONS_REMOTE_INFERENCE_ENABLED;
@@ -46,7 +47,8 @@ public void setUp() {
                             ML_COMMONS_MULTI_TENANCY_ENABLED,
                             ML_COMMONS_REMOTE_INFERENCE_ENABLED,
                             ML_COMMONS_AGENT_FRAMEWORK_ENABLED,
-                            ML_COMMONS_LOCAL_MODEL_ENABLED
+                            ML_COMMONS_LOCAL_MODEL_ENABLED,
+                            ML_COMMONS_CONTROLLER_ENABLED
                         )
                 )
             );