TestNG数据驱动

testng的功能很强大，利用@DataProvider可以做数据驱动，数据源文件可以是EXCEL，XML，YAML，甚至可以是TXT文本。

@DataProvider注解简介:

@DataProvider标记专门为测试方法提供参数的方法。这类方法必须返回Object[ ][ ]类型的二维数组或者Iterator<Object>[]，每一行的Object[]，都是测试方法的一个测试数据集，测试方法会为每个测试数据集执行一次。如果没有指定参数的名称，则默认为方法的名称，方法的名称没有限制。

 

@DataProvider的小例子:

import java.lang.reflect.Method;

 

import org.testng.annotations.DataProvider;

import org.testng.annotations.Test;

 

public class test {

    @DataProvider(name = "user")

    public Object[][] createUser(Method m) {

        System.out.println(m.getName());

        return new Object[][] { { "root", "root" }, { "test", "root" }, { "test", "test" } };

    }

 

    @Test(dataProvider = "user")

    public void verifyUser(String username, String password) {

        System.out.println("Verify User : " + username + ":" + password);

        assert username.equals(password);

    }

 

}

如上所示@DataProvider注解了createUser方法，返回的二位数组里有三行数据，每行两列。所以@Test(dataProvider = "user")注解的verifyUser方法有两个参数，用来接收每一行的两个数据，如果createUser返回的数据数组的列数和verifyUser的参数个数不同就会报错的。因为返回的有三行，所以verifyUser会被执行三次。结果如下：

PASSED: verifyUser("root", "root")

FAILED: verifyUser("test", "root")

PASSED: verifyUser("test", "test")

CSV文件数据读取和@DataProvider

我自己做了一个以csv为例的测试架子，部分代码可通用。

 

CSV文件读取类（可通用，目录自己可以修改，也可改变成读取EXCEL、TXT等文件）：

import java.io.BufferedReader;

import java.io.File;

import java.io.FileReader;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.TreeMap;

import java.util.regex.Matcher;

 

public class CSVData implements Iterator<Object[]> {

 

    private BufferedReader br        = null;

    //行数

    private int            rowNum    = 0;

    //获取次数

    private int            curRowNo  = 0;

    //列数

    private int            columnNum = 0;

    //key名

    private String[]       columnName;

    //csv中所有行数据

    private List<String>   csvList;

    //实际想要的行数据

    private List<String>   csvListNeed;

 

    /*

    * 在TestNG中由@DataProvider(dataProvider = "name")修饰的方法

    * 取csv时，调用此类构造方法（此方法会得到列名并将当前行移到下以后）执行后，转发哦

    * TestNG自己的方法中去，然后由它们调用此类实现的hasNext()、next()方法

    * 得到一行数据，然后返回给由@Test(dataProvider = "name")修饰的方法，如此

    * 反复到数据读完为止

    * 

    * 

    * @param filepath CSV文件名

    * @param casename 用例名

    */

    public CSVData(String fileName, String caseId) {

        try {

            File directory = new File(".");

            String ss = "resources.";

            File csv = new File(directory.getCanonicalFile() + "\\src\\test\\" + ss.replaceAll("\\.", Matcher.quoteReplacement("\\"))

                                + fileName + ".csv");

 

            br = new BufferedReader(new FileReader(csv));

            csvList = new ArrayList<String>();

            while (br.ready()) {

                csvList.add(br.readLine());

                this.rowNum++;

            }

            String stringValue[] = csvList.get(0).split(",");

            this.columnNum = stringValue.length;

 

            columnName = new String[stringValue.length];

 

            for (int i = 0; i < stringValue.length; i++) {

 

                columnName[i] = stringValue[i].toString();

 

            }

            this.curRowNo++;

            csvListNeed = new ArrayList<String>();

            for (int i = 1; i < rowNum; i++) {

                String values[] = csvList.get(i).split(",");

                if (caseId.equals(values[0])) {

                    csvListNeed.add(csvList.get(i));

                }

            }

            this.rowNum = 2;//就取一行

        } catch (Exception e) {

            e.printStackTrace();

        }

    }

 

    @Override

    public boolean hasNext() {

        if (this.rowNum == 0 || this.curRowNo >= this.rowNum) {

            try {

                br.close();

            } catch (Exception e) {

                e.printStackTrace();

            }

            return false;

        } else {

            return true;

        }

    }

 

    @Override

    public Object[] next() {

        /*

        * 将数据放入map 

        */

        Map<String, String> s = new TreeMap<String, String>();

        String csvCell[] = csvListNeed.get(0).split(",");

        for (int i = 0; i < this.columnNum; i++) {

            String temp = "";

            try {

                temp = csvCell[i].toString();

            } catch (ArrayIndexOutOfBoundsException ex) {

                temp = "";

            }

            s.put(this.columnName[i], temp);

        }

        Object r[] = new Object[1];

        r[0] = s;

        this.curRowNo++;

        return r;

    }

 

    @Override

    public void remove() {

        throw new UnsupportedOperationException("remove unsupported");

    }

 

}

 

这个类实现了Iterator<Object[]>迭代器，TestNG调用此类实现的hasNext()、next()方法得到一行数据，在next()方法中可以看到，我把数据是放在Map<String, String>中的，再把map放在Object[]里，所以测试方法的参数就必须是一个Map<String, String>。我这里改成了只读取一行，因为一个csv文件的一个caseId只应该有一行。

 

数据驱动类：

import java.lang.reflect.Method;

import java.util.Iterator;

 

import org.testng.annotations.DataProvider;

 

public class DataProviderTest {

 

    /**

     * @DataProvider的返回值类型只能是Object[][]与Iterator<Object>[]

     * 

     * @param method

     * @return

     */

    @DataProvider

    public Iterator<Object[]> dataSource(Method method) {

        return (Iterator<Object[]>) new CSVData(method.getDeclaringClass().getSimpleName(), method.getName());

    }

 

}

Method方法是通过反射获取的，总之哪个方法调用我Method就是那个方法。

method.getDeclaringClass().getSimpleName()可以获取方法所属的类的类名。

我这里规定了csv的文件名就是测试类的类名，用例名就是方法名。

return (Iterator<Object[]>) new CSVData(…)就是将CSV读取类读取的结果返回，返回的类型是Iterator<Object[]>的，符合@DataProvider的返回值类型要求。当@Test(dataProvider = "dataSource")注解的测试方法执行时就会调用Iterator的hasNext()判断是否有数据和next()获取数据。

 

测试类：

 

import java.util.Map;

 

import org.testng.annotations.Test;

 

public class DataTest extends DataProviderTest {

    @Test(dataProvider = "dataSource")

    public void id2(Map<String, String> data) {

        System.out.println(data);

    }

 

    @Test(dataProvider = "dataSource")

    public void id1(Map<String, String> data) {

        System.out.println(data);

    }

 

}

 

DataTest.csv文件如下：



 

输出结果如下：

PASSED: id1({caseId=id1, flag=Y, property=flowModel, type=com.mybank.bkloanapply.core.model.BaseModel, value=BaseModel.csv@1})

PASSED: id2({caseId=id2, flag=M, property=context, type=java.util.Map, value=a:Object.csv@1})

 

总结

通过以上例子可以看到，无论@DataProvider注解的方法返回的是Object[ ][ ]还是Iterator<Object>[]，最后测试方法获得的参数都是Object[ ]里放的东西，第一个例子里放了两列String，第二个例子里放了Map<String, String>，所以第一个测试类的测试方法的参数是两个String，第二个测试类的测试方法的参数是Map<String, String>，必须保持一致才行。

前几天和一个博士师兄聊天，师兄说：“你毕设做什么题目？”，我把毕设题目给师兄看了，师兄说：“数据驱动啊”。我突然有点懵，我做的不是深度学习吗？怎么变成了数据驱动。可能是因为我知道的太少了吧！

目前个人的理解是：深度学习是机器学习的子集，数据驱动是大数据？可能各有各的说法，但是都是属于人工智能的领域。

让我们通过一个小例子来简单理解一下什么是数据驱动

假设有A和B两家订票网站：

A公司将从网站收集到的数据进行分析，通过数据分析结果认为十一是出行高峰期，所以在十一前后提高价格，并手动在后头调整了十一前后的机票价格和酒店价格。（依靠人为来决定价格）

B公司则通过自动化手段实时收集数据，在线分析相关信息，总结规律，形成数据建模，当用户搜索时自动根据模型对价格进行调整。（可以说B公司进行了“数据采集——数据分析——得到模型”这样的一个过程）

这两个公司，哪家是数据驱动的公司呢？

显而易见，公司B是数据驱动型的公司，而公司A则是一家“以数据为中心来进行决策”的公司。

总的来说，数据驱动就是处理数据，跟数据相关吧！

以上仅为个人理解

 

 

数据驱动和模型驱动的区别
自己的理解，不对望纠正。
模型驱动
在以前的基于数学的时期时，因为不可能对大量的数据做处理，所以只能依靠在少量数据中例如数学中正态分布等等的假定来做假定的数学模型，进行手工计算，并推导出一些由这些模型所得结果的性质，模型驱动就是让数据去贴合某个模型，拿出一组数据来，对比更适合哪个模型。
数据驱动
数据驱动是目前比较火的研究方法，是指在得到一组数据后，如果想要得到某种效果和对数据做某些操作，让模型去贴合数据，从而改变该模型，以达到效果。
区别
模型驱动和数据驱动的区别在于对数据的操作，如果操作并不改变模型本身，就是模型驱动，如果操作改变了模型，让模型去贴合数据，就是数据驱动的研究方法。

数据驱动：
数据驱动从数据文件读取输入数据，通过变量的参数化将测试数据传入测试脚本，不同的数据文件对应不同的测试用例，数据和脚本分离。
数据来控制测试的业务流。 比如你测一个WEB程序，有很多页面，你可以通过一个数据来控制每次是再哪个页面下工作的（即通过数据来导航到相应的页面）。它是关键字驱动的低级版本，他控制的是函数级的，而关键字是控制动作级的。所以数据驱动应该是可以控制整个测试的


关键字驱动：
可以认为是对数字驱动的逻辑扩张。将测试逻辑按照关键字进行分解，关键字对应封装的逻辑业务。主要关键字包括三类：被操作对象（Item）、操作（operation）和值（value），利用面向对象的方式可以将其表现为Item.Operation（value）。