diff --git a/core/src/test/java/kafka/test/ClusterTestExtensionsTest.java b/core/src/test/java/kafka/test/ClusterTestExtensionsTest.java index c698632094e81..f9eec463cf9d7 100644 --- a/core/src/test/java/kafka/test/ClusterTestExtensionsTest.java +++ b/core/src/test/java/kafka/test/ClusterTestExtensionsTest.java @@ -37,6 +37,11 @@ import org.junit.jupiter.api.Assertions; import org.junit.jupiter.api.extension.ExtendWith; +import kafka.utils.TestUtils; +import org.junit.jupiter.api.extension.AfterAllCallback; +import org.junit.jupiter.api.extension.BeforeAllCallback; +import org.junit.jupiter.api.extension.ExtensionContext; + import java.util.Arrays; import java.util.Collections; import java.util.HashMap; @@ -56,7 +61,7 @@ @ClusterConfigProperty(id = 0, key = "queued.max.requests", value = "100"), }) // Set defaults for a few params in @ClusterTest(s) @ExtendWith(ClusterTestExtensions.class) -public class ClusterTestExtensionsTest { +public class ClusterTestExtensionsTest implements BeforeAllCallback, AfterAllCallback { private final ClusterInstance clusterInstance; @@ -64,6 +69,16 @@ public class ClusterTestExtensionsTest { this.clusterInstance = clusterInstance; } + @Override + public void afterAll(ExtensionContext extent) throws Exception { + TestUtils.verifyNoUnexpectedThreads("@AfterAllCallback"); + } + + @Override + public void beforeAll(ExtensionContext context) throws Exception { + TestUtils.verifyNoUnexpectedThreads("@BeforeAllCallback"); + } + // Static methods can generate cluster configurations static List generate1() { Map serverProperties = new HashMap<>();