Fix Forbidden annotation use: org.junit.Test

Signed-off-by: Sumit Bansal <sumitsb@amazon.com>
sumitasr · Oct 1, 2024 · 09492e1 · 09492e1
1 parent 4e732b1
commit 09492e1
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/server/src/test/java/org/opensearch/common/breaker/ResponseLimitSettingsTests.java b/server/src/test/java/org/opensearch/common/breaker/ResponseLimitSettingsTests.java
@@ -23,7 +23,6 @@
 import org.opensearch.core.index.Index;
 import org.opensearch.core.index.shard.ShardId;
 import org.opensearch.test.OpenSearchTestCase;
-import org.junit.Test;
 
 import java.util.HashMap;
 import java.util.Map;
@@ -58,16 +57,20 @@ public void testIsResponseLimitBreachedForNullRoutingTableExpectNotBreached() {
         assertFalse(breached);
     }
 
-    @Test(expected = IllegalArgumentException.class)
     public void testIsResponseLimitBreachedForNullLimitEntityWithRoutingTableExpectException() {
         final ClusterState clusterState = buildClusterState("test-index-1", "test-index-2", "test-index-3");
-        ResponseLimitSettings.isResponseLimitBreached(clusterState.getRoutingTable(), null, 4);
+        expectThrows(
+            IllegalArgumentException.class,
+            () -> ResponseLimitSettings.isResponseLimitBreached(clusterState.getRoutingTable(), null, 4)
+        );
     }
 
-    @Test(expected = IllegalArgumentException.class)
     public void testIsResponseLimitBreachedForNullLimitEntityWithMetadataExpectException() {
         final ClusterState clusterState = buildClusterState("test-index-1", "test-index-2", "test-index-3");
-        ResponseLimitSettings.isResponseLimitBreached(clusterState.getMetadata(), null, 4);
+        expectThrows(
+            IllegalArgumentException.class,
+            () -> ResponseLimitSettings.isResponseLimitBreached(clusterState.getMetadata(), null, 4)
+        );
     }
 
     public void testIsResponseLimitBreachedForCatIndicesWithNoSettingPassedExpectNotBreached() {