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

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

测试驱动技术（TDD）系列之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 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次，如下图所示：