Attempt to fix flaky test MessageDispatchThrottlingTest

apache · Feb 8, 2025 · e4f4689 · e4f4689
1 parent f9aab2e
commit e4f4689
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/broker/auth/MockedPulsarServiceBaseTest.java b/pulsar-broker/src/test/java/org/apache/pulsar/broker/auth/MockedPulsarServiceBaseTest.java
@@ -365,6 +365,9 @@ protected void afterPulsarStart(PulsarService pulsar) throws Exception {
     protected void restartBroker() throws Exception {
         stopBroker();
         startBroker();
+        if (pulsarClient == null) {
+            pulsarClient = newPulsarClient(lookupUrl.toString(), 0);
+        }
     }
 
     protected void stopBroker() throws Exception {

diff --git a/...ker/src/test/java/org/apache/pulsar/client/api/AbstractMessageDispatchThrottlingTest.java b/...ker/src/test/java/org/apache/pulsar/client/api/AbstractMessageDispatchThrottlingTest.java
@@ -40,7 +40,7 @@ public static <T> T[] merge(T[] first, T[] last) {
         return result;
     }
 
-    @BeforeClass
+    @BeforeClass(alwaysRun = true)
     @Override
     protected void setup() throws Exception {
         AsyncTokenBucket.switchToConsistentTokensView();