chore(test): add test timeout (#2410)

Signed-off-by: Robin Han <[email protected]>

Author: superhx

automq-shell/src/test/java/com/automq/shell/util/UtilsTest.java

@@ -16,9 +16,11 @@
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 
 import io.netty.buffer.ByteBuf;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class UtilsTest {
 

core/src/test/java/kafka/autobalancer/ControllerActionExecutorServiceTest.java

@@ -31,12 +31,14 @@
 
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 import org.mockito.ArgumentCaptor;
 import org.mockito.Mockito;
 
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class ControllerActionExecutorServiceTest {
 

core/src/test/java/kafka/autobalancer/goals/GoalTestBase.java

@@ -26,6 +26,7 @@
 import org.apache.kafka.common.TopicPartition;
 
 import org.junit.jupiter.api.Tag;
+import org.junit.jupiter.api.Timeout;
 
 import java.util.Collection;
 import java.util.HashMap;
@@ -36,6 +37,7 @@
 import static kafka.autobalancer.common.types.Resource.NW_IN;
 import static kafka.autobalancer.common.types.Resource.NW_OUT;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class GoalTestBase {
     private final Map<String, Goal> goalMap = new HashMap<>();

core/src/test/java/kafka/autobalancer/goals/ResourceUsageDistributionGoalTest.java

@@ -32,6 +32,7 @@
 import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.ValueSource;
 
@@ -45,6 +46,7 @@
 import static kafka.autobalancer.common.types.Resource.NW_IN;
 import static kafka.autobalancer.common.types.Resource.NW_OUT;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class ResourceUsageDistributionGoalTest extends GoalTestBase {
 

core/src/test/java/kafka/autobalancer/metricsreporter/AutoBalancerMetricsReporterTest.java

@@ -25,11 +25,13 @@
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 import org.mockito.Mockito;
 
 import java.util.HashMap;
 import java.util.Map;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class AutoBalancerMetricsReporterTest {
 

core/src/test/java/kafka/autobalancer/metricsreporter/metric/MetricSerdeTest.java

@@ -23,13 +23,15 @@
 
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 
 import java.util.Map;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class MetricSerdeTest {
     private static final long TIME = 123L;

core/src/test/java/kafka/autobalancer/metricsreporter/metric/MetricsUtilsTest.java

@@ -26,10 +26,12 @@
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 
 import java.util.Collections;
 import java.util.Map;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class MetricsUtilsTest {
 

core/src/test/java/kafka/autobalancer/model/ClusterModelTest.java

@@ -37,12 +37,14 @@
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 
 import java.util.Collection;
 import java.util.Collections;
 import java.util.Map;
 import java.util.Set;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class ClusterModelTest {
 

core/src/test/java/kafka/log/streamaspect/ElasticLogSegmentManagerTest.java

@@ -13,6 +13,7 @@
 
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 
 import java.util.HashSet;
 import java.util.Set;
@@ -31,6 +32,7 @@
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class ElasticLogSegmentManagerTest {
     @Test

core/src/test/java/kafka/log/streamaspect/ElasticTimeIndexTest.java

@@ -20,13 +20,15 @@
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 
 import java.io.IOException;
 import java.util.List;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
 // TODO: replace S3Unit to AutoMQ
+@Timeout(60)
 @Tag("S3Unit")
 public class ElasticTimeIndexTest {
     int maxEntries = 30;

core/src/test/java/kafka/log/streamaspect/ElasticTransactionIndexTest.java

@@ -20,6 +20,7 @@
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 
 import java.io.File;
 import java.io.IOException;
@@ -28,6 +29,7 @@
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class ElasticTransactionIndexTest {
     @Test

core/src/test/java/kafka/log/streamaspect/PartitionStatusTrackerTest.java

@@ -16,9 +16,11 @@
 
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class PartitionStatusTrackerTest {
 

core/src/test/java/kafka/log/streamaspect/cache/FileCacheTest.java

@@ -20,6 +20,7 @@
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 
 import java.io.IOException;
 import java.util.Arrays;
@@ -32,6 +33,7 @@
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
+@Timeout(60)
 @Tag("S3Unit")
 public class FileCacheTest {
 

core/src/test/scala/kafka/log/streamaspect/ElasticLogCleanerTest.scala

@@ -14,6 +14,7 @@ import java.io.File
 import java.util.Properties
 import scala.jdk.CollectionConverters.CollectionHasAsScala
 
+@Timeout(60)
 @Tag("S3Unit")
 class ElasticLogCleanerTest extends LogCleanerTest {
   var client: Client = _

core/src/test/scala/kafka/log/streamaspect/ElasticLogTest.scala

@@ -24,7 +24,7 @@ import org.apache.kafka.coordinator.transaction.TransactionLogConfigs
 import org.apache.kafka.server.util.{MockTime, Scheduler}
 import org.apache.kafka.storage.internals.log._
 import org.junit.jupiter.api.Assertions._
-import org.junit.jupiter.api.{AfterEach, BeforeEach, Tag, Test}
+import org.junit.jupiter.api.{AfterEach, BeforeEach, Tag, Test, Timeout}
 
 import java.io.{File, IOException}
 import java.nio.charset.StandardCharsets
@@ -33,6 +33,7 @@ import java.util.regex.Pattern
 import scala.jdk.CollectionConverters.IterableHasAsScala
 
 // TODO: extends the LocalLogTest
+@Timeout(60)
 @Tag("S3Unit")
 class ElasticLogTest {
     val kafkaConfig: KafkaConfig = KafkaConfig.fromProps(TestUtils.createBrokerConfig(0, "127.0.0.1:1", port = -1))

core/src/test/scala/kafka/log/streamaspect/ElasticProducerStateManagerTest.scala

@@ -20,11 +20,10 @@ import org.apache.kafka.common.utils.{MockTime, Utils}
 import org.apache.kafka.coordinator.transaction.TransactionLogConfigs
 import org.apache.kafka.storage.internals.log._
 import org.junit.jupiter.api.Assertions._
-import org.junit.jupiter.api.{AfterEach, BeforeEach, Tag, Test}
+import org.junit.jupiter.api.{AfterEach, BeforeEach, Tag, Test, Timeout}
 import org.junit.jupiter.params.ParameterizedTest
 import org.junit.jupiter.params.provider.ValueSource
 import org.mockito.Mockito.{mock, when}
-
 import kafka.log.streamaspect.ElasticProducerStateManager.AWAIT_SEQ_ZERO_TIMEOUT
 
 import java.io.File
@@ -36,6 +35,7 @@ import java.util.{Collections, Optional, OptionalLong}
 import scala.compat.java8.OptionConverters.RichOptionalGeneric
 import scala.jdk.CollectionConverters._
 
+@Timeout(60)
 @Tag("S3Unit")
 class ElasticProducerStateManagerTest {
 

core/src/test/scala/kafka/log/streamaspect/ElasticUnifiedLogTest.scala

@@ -25,14 +25,15 @@ import org.apache.kafka.server.util.Scheduler
 import org.apache.kafka.storage.internals.log._
 import org.apache.kafka.storage.internals.utils.Throttler
 import org.junit.jupiter.api.Assertions._
-import org.junit.jupiter.api.{BeforeEach, Tag, Test}
+import org.junit.jupiter.api.{BeforeEach, Tag, Test, Timeout}
 import org.junit.jupiter.params.ParameterizedTest
 import org.junit.jupiter.params.provider.ValueSource
 
 import java.io.File
 import java.util.Optional
 import scala.jdk.CollectionConverters.IterableHasAsScala
 
+@Timeout(60)
 @Tag("S3Unit")
 class ElasticUnifiedLogTest extends UnifiedLogTest {
     var client: Client = _

core/src/test/scala/unit/kafka/server/BrokerQuotaManagerTest.java

@@ -29,6 +29,7 @@
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Timeout;
 
 import java.util.Properties;
 import java.util.concurrent.TimeUnit;
@@ -41,6 +42,7 @@
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
+@Timeout(60)
 @Tag("s3Unit")
 public class BrokerQuotaManagerTest {
     private final Time time = Time.SYSTEM;

core/src/test/scala/unit/kafka/server/ControllerConfigurationValidatorTableTest.scala

@@ -18,10 +18,11 @@ import org.apache.kafka.common.config.TopicConfig.TABLE_TOPIC_SCHEMA_TYPE_CONFIG
 import org.apache.kafka.common.errors.InvalidConfigurationException
 import org.apache.kafka.server.record.TableTopicSchemaType
 import org.junit.jupiter.api.Assertions.{assertEquals, assertThrows}
-import org.junit.jupiter.api.{Tag, Test}
+import org.junit.jupiter.api.{Tag, Test, Timeout}
 
 import java.util
 
+@Timeout(60)
 @Tag("S3Unit")
 class ControllerConfigurationValidatorTableTest {
     val config = new KafkaConfig(TestUtils.createDummyBrokerConfig())

core/src/test/scala/unit/kafka/server/streamaspect/ElasticControllerApisTest.scala

@@ -9,10 +9,11 @@ import org.apache.kafka.raft.QuorumConfig
 import org.apache.kafka.server.authorizer.Authorizer
 import org.apache.kafka.server.common.{FinalizedFeatures, MetadataVersion}
 import org.apache.kafka.server.config.KRaftConfigs
-import org.junit.jupiter.api.Tag
+import org.junit.jupiter.api.{Tag, Timeout}
 
 import java.util.Properties
 
+@Timeout(60)
 @Tag("S3Unit")
 class ElasticControllerApisTest extends ControllerApisTest {
 

core/src/test/scala/unit/kafka/server/streamaspect/ElasticKafkaApisTest.scala

@@ -11,13 +11,14 @@ import org.apache.kafka.raft.QuorumConfig
 import org.apache.kafka.server.authorizer.Authorizer
 import org.apache.kafka.server.common.{FinalizedFeatures, MetadataVersion}
 import org.apache.kafka.server.config.KRaftConfigs
-import org.junit.jupiter.api.Tag
+import org.junit.jupiter.api.{Tag, Timeout}
 import org.mockito.Mockito.mock
 
 import java.util.Collections
 import scala.collection.Map
 import scala.jdk.CollectionConverters.SetHasAsScala
 
+@Timeout(60)
 @Tag("S3Unit")
 class ElasticKafkaApisTest extends KafkaApisTest {
   override protected val replicaManager: ElasticReplicaManager = mock(classOf[ElasticReplicaManager])