测试驱动技术（TDD）系列之2：详解TestNG参数化

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

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

大家会发现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 static org.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次，如下图所示：