上一篇文章介绍了测试驱动的相关概念，并以junit4为例，带大家了解如何在测试框架中实现测试驱动。详情：

测试驱动技术系列之1：一文带你上手测试数据驱动

大家会发现Junit4在同一个测试类中实现多组数据的数据驱动，不是很方便，需要我们自己解决!我也说过在TestNG中这个问题很容易搞定!给自己挖了个坑!自己挖的坑总要填上，那么今天就给大家讲解一下数据驱动在TestNG中的应用。本篇文章只重点讲解TestNG参数化相关知识，关于TestNG的环境配置以及基础使用不在本文的讨论范围!TestNG中使用DataProvider实现参数化功能，在执行用例的时候dataProvider迭代中的每组数据都是作为一个用例执行。

一组参数化数据

定义参数化数据，代码如下：

1. @DataProvider 
2.       public Object[][] dp1() { 
3.         return new Object[][] { 
4.           new Object[] { 1, 1,0 }, 
5.           new Object[] { 2, 1,1 }, 
6.           new Object[] { 2, 1,2 }, 
7.         }; 
8.       } 

用例中使用参数化数据(dp1定义的)，代码如下：

1. @Test(dataProvider = "dp1") 
2.   public void f1(Integer n1, Integer n2, Integer n3) { 
3.      Integer result=n2+n3; 
4.      assertEquals(n1,result); 
5.   } 

多组参数化数据

在一个测试类中，可以定义多组参数化数据(参数化数据个数不同，dp1三个，dp2二个)，代码如下：

1. @DataProvider 
2.      public Object[][] dp2() { 
3.        return new Object[][] { 
4.          new Object[] { 1, 1}, 
5.          new Object[] { 2, 1}, 
6.          new Object[] { 2, 1}, 
7.        }; 
8.      } 

用例中使用参数化数据(dp2定义的)，代码如下：

1. @Test(dataProvider = "dp2") 
2.   public void f2(Integer n1, Integer n2) { 
3.        assertEquals(n1,n2); 
4.   } 

测试方法通过DataProvider引用

我们可以自定义一些方法，对外部文件进行读取，然后把读取的数据作为参数在TestNG中引用，具体方法如下：

写一个读取各类文件的方法(txt、excel、db)具体代码就不给出了，但是一定注意方法的返回值必须是Object[][](关于操控excel的知识会在后面的文章中介绍)

1. public Object[][] readfile(String p_file){ 
2.          return  new Object[][] {{ 1, 1},{ 1, 2}}; 
3.      } 

在@DataProvider修饰的方法中调用该方法(也可以直接实现)

1. @DataProvider(name = "dp_func") 
2.        public Object[][] testData(Method testMethod) { 
3.          return readfile(""); 
4.        } 

用例中使用参数化数据(dp_func)，代码如下：

1. @Test(dataProvider = "dp_func") 
2.   public void f3(Integer n1, Integer n2) { 
3.         assertEquals(n1,n2); 
4.   } 

完整代码

1. import org.testng.annotations.DataProvider; 
2. import org.testng.annotations.Test; 
3. import static org.testng.Assert.assertEquals; 
4. import java.lang.reflect.Method; 
5. public class NewTest { 
6.  
7.       @DataProvider 
8.       public Object[][] dp1() { 
9.         return new Object[][] { 
10.           new Object[] { 1, 1,0 }, 
11.           new Object[] { 2, 1,1 }, 
12.           new Object[] { 2, 1,2 }, 
13.         }; 
14.       } 
15.  
16.       @DataProvider 
17.       public Object[][] dp2() { 
18.  
19.         return new Object[][] { 
20.           new Object[] { 1, 1}, 
21.           new Object[] { 2, 1}, 
22.           new Object[] { 2, 1}, 
23.         }; 
24.       } 
25.  
26.       @DataProvider(name = "dp_func") 
27.         public Object[][] testData(Method testMethod) { 
28.           return readfile(""); 
29.         } 
30.  
31.      //读取各类文件，返回值是Object[][] 
32.  
33.       public Object[][] readfile(String p_file){ 
34.           return  new Object[][] {{ 1, 1},{ 1, 2}}; 
35.       } 
36.  
37.   @Test(dataProvider = "dp1") 
38.   public void f1(Integer n1, Integer n2, Integer n3) { 
39.       Integer result=n2+n3; 
40.       assertEquals(n1,result); 
41.   } 
42.  
43.   @Test(dataProvider = "dp2") 
44.   public void f2(Integer n1, Integer n2) { 
45.      assertEquals(n1,n2); 
46.   } 
47.  
48.   @Test(dataProvider = "dp_func") 
49.   public void f3(Integer n1, Integer n2) { 
50.       assertEquals(n1,n2); 
51.   } 
52. } 

运行测试用例，可以看到用例成功地进行了参数化，f1执行了3次，f2执行了3次，f3执行了2次，如下图所示：