TestNG中DataProvider的用法一
为什么要使用DataProvider
- DataProvider可以实现类似Factory的数据驱动的测试,而且使用DataProvider的时候,被测试方法所在的类只被创建一次,只是方法被调用多次,所以比较省时省力
- DataProvider还可以实现懒加载,在需要测试大量数据的时候可以一条一条的加载数据,而不需要一次性把数据加载到内存中,避免了内存的浪费,和可能的内存不够用的危险
DataProvider的常规用法
- 定义一个函数,函数名随意取,使用@DataProvider注解,注解中用name给这个provider起名字,该函数返回一个二维数组
- 该函数返回的二维数组每一行就代表一次测试的参数,每一行的元素跟Test中需要的参数一一对应(通常情况下数组类型是Object,但是可以用其他自定义类型代替)
- 在需要使用provider的方法的Test注解上,用provider属性指向上面的DataProvider
例子
public class TestDataProvider {
@DataProvider(name="provider")
public Object[][] provider(){
Object [][] provider = new Object [5][2];
for (int i = 0; i < provider.length; i++) {
provider[i][0] = "name"+i;
provider[i][1] = i+10;
}
return provider;
}
@Test(dataProvider="provider")
public void getName(String name,int age){
System.out.println(name+"_"+age);
}
}
运行这个Test,输出
[TestNG] Running:
C:\Users\Administrator\AppData\Local\Temp\testng-eclipse-1631420346\testng-customsuite.xml
name0_10
name1_11
name2_12
name3_13
name4_14
PASSED: getName("name0", 10)
PASSED: getName("name1", 11)
PASSED: getName("name2", 12)
PASSED: getName("name3", 13)
PASSED: getName("name4", 14)
===============================================
Default test
Tests run: 5, Failures: 0, Skips: 0
===============================================
可以看到该测试方法运行了5次,每一次都把一个DataProvider提供的数组作为自己的参数输入然后执行了测试。
带Method参数的DataProvider
如果希望DataProvider对不同的Test方法使用不同的dataprovider,那么在DataProvider中可以使用Method参数
@Test(dataProvider = "provider")
public void getFirst(String name, int age) {
System.out.println("第一组"+name);
}
@Test(dataProvider = "provider")
public void getSecond(String name, int age) {
System.out.println("第二组 " + name);
}
@DataProvider(name = "provider")
public Object[][] provider(Method method) {
Object[][] objects;
if (method.getName().equals("getFirst")) { //如果调用该DataProvider的函数是getFirst,那么就返回这个数组
objects = new Object[][] { { "cq1", 20 }, { "cq2", 22 } };
} else if (method.getName().equals("getSecond")) {//如果调用该DataProvider的函数是getSecond,那么就返回这个数组
objects = new Object[][] { { "cq3", 20 }, { "cq4", 22 } };
} else { //如果调用该DataProvider的函数不是getFirst也不是getSecond,那么就返回这个数组
objects = new Object[][] { {"cq5",33}, {"cq6",34} };
}
return objects;
}
这里根据调用DataProvider的函数名,返回了不同的参数。
由于没有指定运行顺序,所以Test会按照字母顺序执行,先执行getFirst,然后getSecond。
执行getFirst时发现有两组参数,就先把第一组参数传进去执行一遍,再把第二组参数传进去执行一遍。
执行getSecond时也是同样的顺序,所以最后的执行顺序是
①getFirst输入第一组参数
②getFirst输入第二组参数
③getSecond输入第一组参数
④getSecond输入第二组参数
带ITestContext的DataProvider
在Dataprovider中可以带一个ITestContext 参数,从这个参数中可以获取该方法执行时的一些上下文信息,比如context.getIncludeGroups就可以得到这个测试执行的分组,这样在不同的分组中就可以可以向测试执行类传递不同的参数了。
DataProvider的其他位置
在@Test标签中使用dataProvider属性时还可以额外指定一个dataProviderClass属性,这个属性可以定义dataProvider函数的位置类,如果不指定这个属性就会在@Test所在的类中寻找dataProvider,如果指定了就会在指定的类中寻找dataProvider。指定类中的dataProvider必须是static方法。
Tips:dataProvider调用后的方法,在TestNG插件或报告中的方法名后面会显示参数值,这样当运行出现问题后就可以知道哪组参数导致程序出现了异常。