Added Changes after successful build

Signed-off-by: Ajay Kumar Movva <movvaam@amazon.com>
opensearch-project · Jan 31, 2024 · 74760ac · 74760ac
1 parent f3fbd35
commit 74760ac
Show file tree

Hide file tree

Showing 2 changed files with 246 additions and 246 deletions.
diff --git a/...ensearch/ratelimitting/admissioncontrol/controllers/CPUBasedAdmissionControllerTests.java b/...ensearch/ratelimitting/admissioncontrol/controllers/CPUBasedAdmissionControllerTests.java
@@ -6,104 +6,104 @@
  * compatible open source license.
  */
 
-// package org.opensearch.ratelimitting.admissioncontrol.controllers;
-//
-// import org.opensearch.cluster.service.ClusterService;
-// import org.opensearch.common.settings.ClusterSettings;
-// import org.opensearch.common.settings.Settings;
-// import org.opensearch.ratelimitting.admissioncontrol.enums.AdmissionControlMode;
-// import org.opensearch.ratelimitting.admissioncontrol.settings.CPUBasedAdmissionControllerSettings;
-// import org.opensearch.test.OpenSearchTestCase;
-// import org.opensearch.threadpool.TestThreadPool;
-// import org.opensearch.threadpool.ThreadPool;
-//
-// public class CPUBasedAdmissionControllerTests extends OpenSearchTestCase {
-// private ClusterService clusterService;
-// private ThreadPool threadPool;
-// CPUBasedAdmissionController admissionController = null;
-//
-// String action = "TEST_ACTION";
-//
-// @Override
-// public void setUp() throws Exception {
-// super.setUp();
-// threadPool = new TestThreadPool("admission_controller_settings_test");
-// clusterService = new ClusterService(
-// Settings.EMPTY,
-// new ClusterSettings(Settings.EMPTY, ClusterSettings.BUILT_IN_CLUSTER_SETTINGS),
-// threadPool
-// );
-// }
-//
-// @Override
-// public void tearDown() throws Exception {
-// super.tearDown();
-// threadPool.shutdownNow();
-// }
-//
-// public void testCheckDefaultParameters() {
-// admissionController = new CPUBasedAdmissionController(
-// CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER,
-// Settings.EMPTY,
-// clusterService.getClusterSettings()
-// );
-// assertEquals(admissionController.getName(), CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER);
-// assertEquals(admissionController.getRejectionCount(), 0);
-// assertEquals(admissionController.settings.getTransportLayerAdmissionControllerMode(), AdmissionControlMode.DISABLED);
-// assertFalse(
-// admissionController.isEnabledForTransportLayer(admissionController.settings.getTransportLayerAdmissionControllerMode())
-// );
-// }
-//
-// public void testCheckUpdateSettings() {
-// admissionController = new CPUBasedAdmissionController(
-// CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER,
-// Settings.EMPTY,
-// clusterService.getClusterSettings()
-// );
-// Settings settings = Settings.builder()
-// .put(
-// CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER_TRANSPORT_LAYER_MODE.getKey(),
-// AdmissionControlMode.ENFORCED.getMode()
-// )
-// .build();
-// clusterService.getClusterSettings().applySettings(settings);
-//
-// assertEquals(admissionController.getName(), CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER);
-// assertEquals(admissionController.getRejectionCount(), 0);
-// assertEquals(admissionController.settings.getTransportLayerAdmissionControllerMode(), AdmissionControlMode.ENFORCED);
-// assertTrue(admissionController.isEnabledForTransportLayer(admissionController.settings.getTransportLayerAdmissionControllerMode()));
-// }
-//
-// public void testApplyControllerWithDefaultSettings() {
-// admissionController = new CPUBasedAdmissionController(
-// CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER,
-// Settings.EMPTY,
-// clusterService.getClusterSettings()
-// );
-// assertEquals(admissionController.getRejectionCount(), 0);
-// assertEquals(admissionController.settings.getTransportLayerAdmissionControllerMode(), AdmissionControlMode.DISABLED);
-// action = "indices:data/write/bulk[s][p]";
-// admissionController.apply(action);
-// assertEquals(admissionController.getRejectionCount(), 0);
-// }
-//
-// public void testApplyControllerWhenSettingsEnabled() {
-// Settings settings = Settings.builder()
-// .put(
-// CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER_TRANSPORT_LAYER_MODE.getKey(),
-// AdmissionControlMode.ENFORCED.getMode()
-// )
-// .build();
-// admissionController = new CPUBasedAdmissionController(
-// CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER,
-// settings,
-// clusterService.getClusterSettings()
-// );
-// assertTrue(admissionController.isEnabledForTransportLayer(admissionController.settings.getTransportLayerAdmissionControllerMode()));
-// assertEquals(admissionController.getRejectionCount(), 0);
-// action = "indices:data/write/bulk[s][p]";
-// admissionController.apply(action);
-// assertEquals(admissionController.getRejectionCount(), 1);
-// }
-// }
+package org.opensearch.ratelimitting.admissioncontrol.controllers;
+
+import org.opensearch.cluster.service.ClusterService;
+import org.opensearch.common.settings.ClusterSettings;
+import org.opensearch.common.settings.Settings;
+import org.opensearch.ratelimitting.admissioncontrol.enums.AdmissionControlMode;
+import org.opensearch.ratelimitting.admissioncontrol.settings.CPUBasedAdmissionControllerSettings;
+import org.opensearch.test.OpenSearchTestCase;
+import org.opensearch.threadpool.TestThreadPool;
+import org.opensearch.threadpool.ThreadPool;
+
+public class CPUBasedAdmissionControllerTests extends OpenSearchTestCase {
+    private ClusterService clusterService;
+    private ThreadPool threadPool;
+    CPUBasedAdmissionController admissionController = null;
+
+    String action = "TEST_ACTION";
+
+    @Override
+    public void setUp() throws Exception {
+        super.setUp();
+        threadPool = new TestThreadPool("admission_controller_settings_test");
+        clusterService = new ClusterService(
+            Settings.EMPTY,
+            new ClusterSettings(Settings.EMPTY, ClusterSettings.BUILT_IN_CLUSTER_SETTINGS),
+            threadPool
+        );
+    }
+
+    @Override
+    public void tearDown() throws Exception {
+        super.tearDown();
+        threadPool.shutdownNow();
+    }
+
+    public void testCheckDefaultParameters() {
+        admissionController = new CPUBasedAdmissionController(
+            CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER,
+            Settings.EMPTY,
+            clusterService.getClusterSettings()
+        );
+        assertEquals(admissionController.getName(), CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER);
+        assertEquals(admissionController.getRejectionCount(), 0);
+        assertEquals(admissionController.settings.getTransportLayerAdmissionControllerMode(), AdmissionControlMode.DISABLED);
+        assertFalse(
+            admissionController.isEnabledForTransportLayer(admissionController.settings.getTransportLayerAdmissionControllerMode())
+        );
+    }
+
+    public void testCheckUpdateSettings() {
+        admissionController = new CPUBasedAdmissionController(
+            CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER,
+            Settings.EMPTY,
+            clusterService.getClusterSettings()
+        );
+        Settings settings = Settings.builder()
+            .put(
+                CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER_TRANSPORT_LAYER_MODE.getKey(),
+                AdmissionControlMode.ENFORCED.getMode()
+            )
+            .build();
+        clusterService.getClusterSettings().applySettings(settings);
+
+        assertEquals(admissionController.getName(), CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER);
+        assertEquals(admissionController.getRejectionCount(), 0);
+        assertEquals(admissionController.settings.getTransportLayerAdmissionControllerMode(), AdmissionControlMode.ENFORCED);
+        assertTrue(admissionController.isEnabledForTransportLayer(admissionController.settings.getTransportLayerAdmissionControllerMode()));
+    }
+
+    public void testApplyControllerWithDefaultSettings() {
+        admissionController = new CPUBasedAdmissionController(
+            CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER,
+            Settings.EMPTY,
+            clusterService.getClusterSettings()
+        );
+        assertEquals(admissionController.getRejectionCount(), 0);
+        assertEquals(admissionController.settings.getTransportLayerAdmissionControllerMode(), AdmissionControlMode.DISABLED);
+        action = "indices:data/write/bulk[s][p]";
+        admissionController.apply(action);
+        assertEquals(admissionController.getRejectionCount(), 0);
+    }
+
+    public void testApplyControllerWhenSettingsEnabled() {
+        Settings settings = Settings.builder()
+            .put(
+                CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER_TRANSPORT_LAYER_MODE.getKey(),
+                AdmissionControlMode.ENFORCED.getMode()
+            )
+            .build();
+        admissionController = new CPUBasedAdmissionController(
+            CPUBasedAdmissionControllerSettings.CPU_BASED_ADMISSION_CONTROLLER,
+            settings,
+            clusterService.getClusterSettings()
+        );
+        assertTrue(admissionController.isEnabledForTransportLayer(admissionController.settings.getTransportLayerAdmissionControllerMode()));
+        assertEquals(admissionController.getRejectionCount(), 0);
+        action = "indices:data/write/bulk[s][p]";
+        admissionController.apply(action);
+        assertEquals(admissionController.getRejectionCount(), 1);
+    }
+}