Speedup slow tests due to rate limit

slackapi · Sep 19, 2024 · 9cd4793 · 9cd4793
1 parent d5e0036
commit 9cd4793
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/slack-api-client/src/test/java/test_locally/api/methods/TeamTest.java b/slack-api-client/src/test/java/test_locally/api/methods/TeamTest.java
@@ -2,11 +2,15 @@
 
 import com.slack.api.Slack;
 import com.slack.api.SlackConfig;
+import com.slack.api.methods.MethodsConfig;
+import com.slack.api.methods.MethodsCustomRateLimitResolver;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 import util.MockSlackApiServer;
 
+import java.util.Optional;
+
 import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static util.MockSlackApi.ValidToken;
@@ -21,6 +25,19 @@ public class TeamTest {
     public void setup() throws Exception {
         server.start();
         config.setMethodsEndpointUrlPrefix(server.getMethodsEndpointPrefix());
+        MethodsConfig methodsConfig = new MethodsConfig();
+        methodsConfig.setCustomRateLimitResolver(new MethodsCustomRateLimitResolver() {
+            @Override
+            public Optional<Integer> getCustomAllowedRequestsPerMinute(String teamId, String methodName) {
+                return Optional.of(20);
+            }
+
+            @Override
+            public Optional<Integer> getCustomAllowedRequestsForChatPostMessagePerMinute(String teamId, String channel) {
+                return Optional.empty();
+            }
+        });
+        config.setMethodsConfig(methodsConfig);
     }
 
     @After

diff --git a/slack-api-client/src/test/java/test_locally/api/methods_admin_api/AdminApiAsyncTest.java b/slack-api-client/src/test/java/test_locally/api/methods_admin_api/AdminApiAsyncTest.java
@@ -3,6 +3,8 @@
 import com.slack.api.Slack;
 import com.slack.api.SlackConfig;
 import com.slack.api.methods.AsyncMethodsClient;
+import com.slack.api.methods.MethodsConfig;
+import com.slack.api.methods.MethodsCustomRateLimitResolver;
 import com.slack.api.model.admin.AppConfig;
 import org.junit.After;
 import org.junit.Before;
@@ -11,6 +13,7 @@
 
 import java.lang.reflect.Array;
 import java.util.Arrays;
+import java.util.Optional;
 
 import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -26,6 +29,19 @@ public class AdminApiAsyncTest {
     public void setup() throws Exception {
         server.start();
         config.setMethodsEndpointUrlPrefix(server.getMethodsEndpointPrefix());
+        MethodsConfig methodsConfig = new MethodsConfig();
+        methodsConfig.setCustomRateLimitResolver(new MethodsCustomRateLimitResolver() {
+            @Override
+            public Optional<Integer> getCustomAllowedRequestsPerMinute(String teamId, String methodName) {
+                return Optional.of(50);
+            }
+
+            @Override
+            public Optional<Integer> getCustomAllowedRequestsForChatPostMessagePerMinute(String teamId, String channel) {
+                return Optional.empty();
+            }
+        });
+        config.setMethodsConfig(methodsConfig);
     }
 
     @After