一、前言
- 最近测试一个开源项目，发现生成的
全局id有重复，也没有单元测试，就准备贡献个PR - 想到多线程并发测试，根据经验，第一想法是用 Testng，后面看了下 Junit5 也有实验性支持了，就对比下(以
maven为例) - spock 不太适合这种场景
二、Testng
1. 安装
- 选择 使用数 比较多、也比较新 的版本，7.7.1。<testng.version>7.7.1</testng.version>
- 多模块项目，可以在 根pom.xml里面添加依赖，避免每个模块都增加配置哦
<dependencies>
<!-- test -->
<dependency>
<groupId>org.junit.jupiter</groupId>
<artifactId>junit-jupiter</artifactId>
<scope>test</scope>
</dependency>
</dependencies>
2. 使用
- @Test 就可以指定 执行次数(invocationCount)、并发(threadPoolSize)，很方便
- 同一个测试对象，ids属性 不用加 static
public class UniqueIdGeneratorTest {
private Set<Long> ids = new ConcurrentHashSet<>(128);
@org.testng.annotations.Test(invocationCount = 128, threadPoolSize = 3)
public void testGenerateId() {
final Long id = UniqueIdGenerator.generateId();
assertTrue(ids.add(id), "id exists," + id);
}
}
3. 效果
testng.png
三、Junit5
1. 安装
- 选择 使用数 比较多、也比较新 的版本，5.8.2。<junit-jupiter.version>5.8.2</junit-jupiter.version>
- 最好通过 dependencyManagement 来统一版本，尤其是 多模块项目
- 建议放到 spring-boot-dependencies 前面，优先级更高
<dependencyManagement>
<dependencies>
<dependency>
<groupId>org.junit</groupId>
<artifactId>junit-bom</artifactId>
<version>${junit-jupiter.version}</version>
<type>pom</type>
<scope>import</scope>
</dependency>
</dependencies> </dependencyManagement>
多模块项目，可以在 根pom.xml里面添加依赖，避免每个模块都增加配置哦
<dependencies>
<!-- test -->
<dependency>
<groupId>org.testng</groupId>
<artifactId>testng</artifactId>
<version>${testng.version}</version>
<scope>test</scope>
</dependency>
</dependencies>
2 配置项
- 先设置 junit.jupiter.execution.parallel.enabled=true
- 全局并发测试：junit.jupiter.execution.parallel.mode.default=concurrent
- 局部并发测试：@RepeatedTest 和 @Execution(CONCURRENT)
- 按需配置并行策略 strategy，建议 fixed，线程数 可控
-
fixed 配置：如 fixed.parallelism=3，fixed.max-pool-size=3(默认 256 + parallelism)
junit5-fixed.png
-
dynamic策略(根据 factor、CPU核数 调整parallelism)，parallelism = max(1, factor * CPU核数)，后面和 fixed 逻辑一样
junit5-dynamic.png
3 配置方式
- System properties 配置方式，更适合多模块项目(根pom.xml配置，子模块就不用配置了)
<plugin>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-surefire-plugin</artifactId>
<version>2.18.1</version>
<configuration>
<argLine>-Djunit.jupiter.execution.parallel.enabled=true -Djunit.jupiter.execution.parallel.config.strategy=fixed
-Djunit.jupiter.execution.parallel.config.fixed.parallelism=3 -Djunit.jupiter.execution.parallel.config.fixed.max-pool-size=3
</argLine>
</configuration>
</plugin>
-
配置文件
test/resources 目录下，增加 junit-platform.properties 文件，内容如下：
#是否允许并行执行true/false
junit.jupiter.execution.parallel.enabled=true
#是否支持方法级别多线程same_thread/concurrent
junit.jupiter.execution.parallel.mode.default=concurrent
#是否支持类级别多线程same_thread/concurrent
junit.jupiter.execution.parallel.mode.classes.default=concurrent
# the maximum pool size can be configured using a ParallelExecutionConfigurationStrategy
junit.jupiter.execution.parallel.config.strategy=fixed
junit.jupiter.execution.parallel.config.fixed.parallelism=3
junit.jupiter.execution.parallel.config.fixed.max-pool-size=3
4. 使用
- 需改为 @RepeatedTest，和 普通测试不一致，并发线程数 是全局的
- 测试对象 不同，ids属性 需加 static
class UniqueIdGeneratorTest2 {
private static Set<Long> ids = new ConcurrentHashSet<>(128);
@org.junit.jupiter.api.RepeatedTest(128)
@org.junit.jupiter.api.parallel.Execution(ExecutionMode.CONCURRENT)
void generateId() {
final Long id = UniqueIdGenerator.generateId();
assertTrue(ids.add(id), "id exists," + id);
}
}
5. 效果
junit5.png
四、总结
- 多线程并发测试 场景，Testng 可能比 Junit5 更合适
- 使用不多也可以试试 hutool 的 ThreadUtil#concurrencyTest(int threadSize, Runnable runnable)
