diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApiTest.java b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApiTest.java index c08cc1cc1626a..30c6cc7d4d2c5 100644 --- a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApiTest.java +++ b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApiTest.java @@ -154,6 +154,8 @@ @Test(groups = "broker-admin") public class AdminApiTest extends MockedPulsarServiceBaseTest { + private static final String BROKER_SHUTDOWN_TIMEOUT_MS = "brokerShutdownTimeoutMs"; + private MockedPulsarService mockPulsarSetup; private PulsarService otherPulsar; @@ -216,6 +218,14 @@ private void setupConfigAndStart(java.util.function.Consumer assertNotEquals(pulsar.getConfiguration().getBrokerShutdownTimeoutMs(), overriddenTimeoutMs)); + } + private void setupClusters() throws PulsarAdminException { admin.clusters().createCluster("test", ClusterData.builder().serviceUrl(pulsar.getWebServiceAddress()).build()); TenantInfoImpl tenantInfo = new TenantInfoImpl(Set.of("role1", "role2"), Set.of("test")); @@ -754,6 +783,8 @@ public void testGetDynamicLocalConfiguration() throws Exception { admin.brokers().updateDynamicConfiguration(configName, Long.toString(shutdownTime)); // Now, znode is created: updateConfigurationAndRegisterListeners and check if configuration updated assertEquals(Long.parseLong(admin.brokers().getAllDynamicConfigurations().get(configName)), shutdownTime); + // wait until the broker has applied the value, so that the @AfterMethod always has to restore it + Awaitility.await().until(() -> pulsar.getConfiguration().getBrokerShutdownTimeoutMs() == shutdownTime); } @Test