上一篇文章介绍了测试驱动的相关概念，并以junit4为例，带大家了解如何在测试框架中实现测试驱动。详情：
大家会发现Junit4在同一个测试类中实现多组数据的数据驱动，不是很方便，需要我们自己解决！我也说过在TestNG中这个问题很容易搞定！给自己挖了个坑！自己挖的坑总要填上，那么今天就给大家讲解一下数据驱动在TestNG中的应用。本篇文章只重点讲解TestNG参数化相关知识，关于TestNG的环境配置以及基础使用不在本文的讨论范围！TestNG中使用DataProvider实现参数化功能，在执行用例的时候dataProvider迭代中的每组数据都是作为一个用例执行。
一组参数化数据
定义参数化数据，代码如下：
@DataProvider
public Object[][] dp1() {
return new Object[][] {
new Object[] { 1, 1,0 },
new Object[] { 2, 1,1 },
new Object[] { 2, 1,2 },
};
}
用例中使用参数化数据（dp1定义的），代码如下：
@Test(dataProvider = "dp1")
public void f1(Integer n1, Integer n2, Integer n3) {
Integer result=n2+n3;
assertEquals(n1,result);
}
多组参数化数据
在一个测试类中，可以定义多组参数化数据（参数化数据个数不同，dp1三个，dp2二个），代码如下：
@DataProvider
public Object[][] dp2() {
return new Object[][] {
new Object[] { 1, 1},
new Object[] { 2, 1},
new Object[] { 2, 1},
};
}
用例中使用参数化数据（dp2定义的），代码如下：
@Test(dataProvider = "dp2")
public void f2(Integer n1, Integer n2) {
assertEquals(n1,n2);
}
测试方法通过DataProvider引用
我们可以自定义一些方法，对外部文件进行读取，然后把读取的数据作为参数在TestNG中引用，具体方法如下：
写一个读取各类文件的方法（txt、excel、db）具体代码就不给出了，但是一定注意方法的返回值必须是Object[][]（关于操控excel的知识会在后面的文章中介绍）
public Object[][] readfile(String p_file){
return new Object[][] {{ 1, 1},{ 1, 2}};
}
在@DataProvider修饰的方法中调用该方法（也可以直接实现）
@DataProvider(name = "dp_func")
public Object[][] testData(Method testMethod) {
return readfile("");
}
用例中使用参数化数据（dp_func），代码如下：
@Test(dataProvider = "dp_func")
public void f3(Integer n1, Integer n2) {
assertEquals(n1,n2);
}
完整代码
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
import staticorg.testng.Assert.assertEquals;
import java.lang.reflect.Method;
public class NewTest {
@DataProvider
public Object[][] dp1() {
return new Object[][] {
new Object[] { 1, 1,0 },
new Object[] { 2, 1,1 },
new Object[] { 2, 1,2 },
};
}
@DataProvider
public Object[][] dp2() {
return new Object[][] {
new Object[] { 1, 1},
new Object[] { 2, 1},
new Object[] { 2, 1},
};
}
@DataProvider(name = "dp_func")
public Object[][] testData(Method testMethod) {
return readfile("");
}
//读取各类文件，返回值是Object[][]
public Object[][] readfile(String p_file){
return new Object[][] {{ 1, 1},{ 1, 2}};
}
@Test(dataProvider = "dp1")
public void f1(Integer n1, Integer n2, Integer n3) {
Integer result=n2+n3;
assertEquals(n1,result);
}
@Test(dataProvider = "dp2")
public void f2(Integer n1, Integer n2) {
assertEquals(n1,n2);
}
@Test(dataProvider = "dp_func")
public void f3(Integer n1, Integer n2) {
assertEquals(n1,n2);
}
}
运行测试用例，可以看到用例成功地进行了参数化，f1执行了3次，f2执行了3次，f3执行了2次，如下图所示：
