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

数据驱动的意义

数据驱动，指在自动化测试中处理测试数据的方式。通常测试数据与功能函数分离，存储在功能函数的外部位置。在自动化测试运行时，数据驱动框架会读取数据源中的数据，把数据作为参数传递到功能函数中，并会根据数据的条数多次运行同一个功能函数。数据驱动的数据源可以是函数外的数据集合、CSV 文件、Excel 表格、TXT 文件，以及数据库等。如果没有数据驱动我们想对一个方法Add()，进行三组数据的测试{1,2 },{0,2},{0,3}，那么我们需要写三个测试方法；而通过数据驱动只需要写一个测试方法，然后通过循环读取测试数据的方式就能完成对三组数据的测试。数据驱动在测试框架中都是通过对测试数据的参数化来实现，本文会详细讲解测试驱动在junit4测试框架中的应用。

Junit4中数据驱动基础实现

代码如下：

import java.util.Arrays;

import java.util.Collection;

import org.junit.Assert;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.junit.runners.Parameterized;

import org.junit.runners.Parameterized.Parameters;

//@RunWith表示测试用例将会以参数化的形式执行

@RunWith(Parameterized.class)

public class ParaDemo {

private int input1;

private int input2;

private int expected;

@Parameters

//方法prepareData()中包括需要参数化的数据，示例中有三组数据，每组数据有三个值

public static Collection prepareData(){

Object [][]object= {{1,2,3},{0,2,2},{0,3,3}};

return Arrays.asList(object);

}

//构造方法中的参数与prepareData()中的数据是一一对应的，以第一组数据{1,2,3}为例:

1对应input1，2对应input2，3对应expected

public ParaDemo(int input1,int input2,int expected){

this.input1 =input1;

this.input2 =input2;

this.expected =expected;

}

@Test

//测试用例中把测试数据进行了参数化

public void testAdd(){

int result=this.input1+this.input2;

Assert.assertEquals(this.expected,result);

}

}

运行测试用例，可以看到同一个测试用例被prepareData()中赋值的三组测试数据执行，如下图所示：

测试数据与不同测试用例关联

假设想增加一个testSubstract()测试方法，然后使得testAdd()和testSubstract()分别用各自的参数化测试数据该如何实现呢？思路如下：

1.两个方法设计到不同的测试类中。（这里不做讨论）

2.增加标签对用例数据进行标识。使用枚举类实现enum Type {SUBSTRACT, ADD}；执行用例时，根据标签判定用例是否需要执行。通过org.jnit.Assume.assumeTrue来进行判定

具体代码实现如下:

import static org.junit.Assert.assertEquals;

import java.util.Arrays;

import java.util.Collection;

import org.junit.Assume;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.junit.runners.Parameterized;

import org.junit.runners.Parameterized.Parameters;

@RunWith(Parameterized.class)

public class ParaJunit4Demo {

enum Type {SUBSTRACT,ADD};

@Parameters

public static Collection data(){

return Arrays.asList(newObject[][] {

{Type.SUBSTRACT, 3, 2, 1},

{Type.SUBSTRACT, 4, 2, 2},

{Type.SUBSTRACT, 4, 2, 2},

{Type.ADD, 3, 2, 5},

{Type.ADD, 3, 2, 5},

});

}

private Type type;

private int a, b, expected;

public ParaJunit4Demo(Typetype,int a,int b,int expected){

this.type =type;

this.a=a;this.b=b;this.expected=expected;

}

@Test

public void testAdd(){

Assume.assumeTrue(type == Type.ADD);

assertEquals(this.expected,this.a+this.b);

}

@Test

public void testSubstract(){

Assume.assumeTrue(type == Type.SUBSTRACT);

assertEquals(this.expected,this.a-this.b);

}

}

上面的例子是testAdd()和testSubstract()拥有相同的测试参数个数(都是3个)，如果在增加一个测试用例testOthers()，而它只需要两个测试参数，又该如何实现呢？

尝试方法一：直接设置参数数据为{Type.OTHERS, 3,3}，代码如下：

public static Collection data(){

return Arrays.asList(newObject[][] {

{Type.SUBSTRACT, 3, 2, 1},

{Type.SUBSTRACT, 4, 2, 2},

{Type.SUBSTRACT, 4, 2, 2},

{Type.ADD, 3, 2, 5},

{Type.ADD, 3, 2, 5},

{Type.OTHERS, 3,3}

});

}

//用例设置如下：

@Test

public voidtestOthers(){

Assume.assumeTrue(type == Type.OTHERS);

assertEquals(this.a,this.b);

}

运行代码，提示参数个数错误

尝试方法二：尝试添加一个构造方法，代码如下：

public ParaJunit4Demo(Typetype,int a,int b){

this.type =type;

this.a=a;

this.b=b;

}

运行代码，提示只能有一个构造方法的错误

尝试方法三：添加一组数据，将最后一个数据设置为3，3，0

public static Collection<Object[]>data(){

return Arrays.asList(new Object[][] {

{Type.SUBSTRACT, 3, 2, 1},

{Type.SUBSTRACT, 4, 2, 2},

{Type.SUBSTRACT, 4, 2, 2},

{Type.ADD, 3, 2, 5},

{Type.ADD, 3, 2, 5},

{Type.OTHERS, 3,3,0}

});

}

运行测试，测试通过。

由此可见，想要实现拥有不同测试参数个数的测试用例的参数化，需要使参数个数统一，参数少的用例，需要添加参数进行占位（是否被使用无所谓），实例中，通过添加0来实现参数个数统一，{Type.OTHERS, 3,3,0}

总结

有人会说在junit5或者TestNG中都可以很方便地实现该功能（后面会接受TestNG参数化技术），有必要搞得这么麻烦吗？大家说得没错，在这里我只是给大家展示一个解决问题的思路而已，好多测试同学不知道通过代码怎样解决问题！其实最简单有效的就是完善目前架构中的功能，遇到问题不要置之不理，觉得架构不提供，我们就没有必要再继续深入了，我们应该好好想一想通过编码能否解决问题，多动手多思考，慢慢的编码能力就会提升了！