前言

TestNG 是一款单元测试\集成框架，它提供了一系列的丰富注解来帮助我们很方便的编写和管理被测项目的测试类和测试方法。而且功能比 junit 更强大，支持依赖测试，忽略测试，异常测试，超时测试，分组测试等多种测试场景。广泛运用于接口自动化测试、UI 自动化测试。

在接口测试中，经常需要模拟用户行为，比如：高峰期访问接口，秒杀商品。这些都需对接口进行高并发模拟才能测试出效果。

TestNG 中的多线程

1、并发级别

在 TestNG 中，测试的级别分为 suite -> test -> class -> method，级别从大到小，一对多关系(如：一个 suite 下有多个 test)。

1.1、suite

一个 testng.xml 下只能有一个 suite，所以想要 suite 级别并发，一般不常用。只能通过官方给出的方案： 命令行的指定线程池的容量。

java org.testng.TestNG -suitethreadpoolsize 3 testng1.xml testng2.xml testng3.xml

1.2、test

在 testng.xml 中做如下设置：

效果：不同 test 标签下的用例方法在不同的线程执行，相同 test 标签下的用例方法在同一个线程中执行。为每个 test 开启 10 个线程。

1.3、class

在 testng.xml 中做如下设置：

效果：不同类下的用例方法在不同的线程执行，相同类下的用例方法在同一个线程中执行。 为每个类开启 10 个线程。

1.4、method

在 testng.xml 中做如下设置：

效果：每个用例方法在不同的线程执行。 为每个方法开启 10 个线程。

1.5、instances

在 testng.xml 中做如下设置：

效果：不相对象下的用例方法在不同线程执行，相同对象下的用力方法在相同线程中执行，一般比较少用。 为对象开启 10 个线程。

1.6、不同 DataProvider 的并发

DataProvider 经常被用来做数据驱动，但是设置了 method 级别并发时，一个 DataProvider 中的数据依然是顺序执行。

解决方式是：在@DataProvider 中添加 parallel 属性，并设置为 true。

@DataProvider(parallel = true)
public Object[] datas() {
List list = ExcelUtils.read(sheetIndex, 1);
return list.toArray();
}

默认情况下，DataProvider 并行执行的线程数量为 10，可在 suite 标签下指定 data-provider-thread-count 参数。

1.7、同一个方法并发

当我们需要对某个接口进行反复并发测试时，可以用一种更简单的方式解决。在@Test 注解中提供了 threadPoolSize 和 invocationCount 两个属性。

invocationCount：方法执行次数。

threadPoolSize：线程数量。

@Test(threadPoolSize = 30,invocationCount=10)
public void method2() {
System.out.println("------->" + Thread.currentThread().getName() );
}

准备了30个线程，需要执行10次方法，这种情况有多余空闲线程。

结果：

Java单台服务器并发量 java并发处理一个接口_Java单台服务器并发量

@Test(threadPoolSize = 30,invocationCount=100)
public void method2() {
System.out.println("------->" + Thread.currentThread().getName() );
}

准备了30个线程，需要执行100次方法，这种情况会存在线程竞争。

结果：

Java单台服务器并发量 java并发处理一个接口_Java单台服务器并发量_02

总结

TestNG 对多线程提供了多种支持，包括：suite、test、class、method、instances、DataProvider、@Test 注解，总有一种能够帮助我们在做自动化测试的同时兼顾并发的需求。