欢迎光临
我们一直在努力
当前位置：爱分享 > 文章教程 > Java教程 > 正文

TestNG数据驱动简介

2022-02-08 分类：Java教程 阅读(81) 评论(0)
版权声明
1. 本站文章和资源均来自互联网收集和整理，本站不承担任何责任及版权问题。 2. 相关版权归作者及其公司所有，仅供学习研究用途，请勿用于商业目的。 3. 若侵犯您的版权，请发邮件至webmaster@ishare1.cn联系我们，我们确认后将立即删除。

TestNG数据驱动

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

TestNG数据驱动简介插图

@DataProvider注解简介:

@DataProvider标记专门为测试方法提供参数的方法。这类方法必须返回Object[ ][ ]类型的二维数组或者Iterator[]，每一行的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 {
    private BufferedReader br        = null;
    //行数
    private int            rowNum    = 0;
    //获取次数
    private int            curRowNo  = 0;
    //列数
    private int            columnNum = 0;
    //key名
    private String[]       columnName;
    //csv中所有行数据
    private List   csvList;
    //实际想要的行数据
    private List   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	est" + ss.replaceAll(".", Matcher.quoteReplacement(""))
                                + fileName + ".csv");
            br = new BufferedReader(new FileReader(csv));
            csvList = new ArrayList();
            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 ();
            for (int i = 1; i = this.rowNum) {
            try {
                br.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
            return false;
        } else {
            return true;
        }
    }
    @Override
    public Object[] next() {
        /*
        * 将数据放入map 
        */
        Map s = new TreeMap();
        String csvCell[] = csvListNeed.get(0).split(",");
        for (int i = 0; i 

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


数据驱动类：


import java.lang.reflect.Method;
import java.util.Iterator;
import org.testng.annotations.DataProvider;
public class DataProviderTest {
    /**
     * @DataProvider的返回值类型只能是Object[][]与Iterator[]
     * 
     * @param method
     * @return
     */
    @DataProvider
    public Iterator dataSource(Method method) {
        return (Iterator) new CSVData(method.getDeclaringClass().getSimpleName(), method.getName());
    }
}


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


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


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


return (Iterator) new CSVData(…)就是将CSV读取类读取的结果返回，返回的类型是Iterator的，符合@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 data) {
        System.out.println(data);
    }
    @Test(dataProvider = "dataSource")
    public void id1(Map data) {
        System.out.println(data);
    }
}


输出结果如下：


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[ ]里放的东西，第一个例子里放了两列String，第二个例子里放了Map，所以第一个测试类的测试方法的参数是两个String，第二个测试类的测试方法的参数是Map，必须保持一致才行。


                                                


分享本文到 

			
微海报分享


猜你喜欢：
    

  1. java jsp页面乱码解决方法 
    2. 

  2. java中如何使float类型数据保留两位小数 
    3. 

  3. java怎么从键盘获取数据 
    4. 

  4. java不能被继承的类是什么类 
    5. 

  5. java判断不是数字的方法介绍 
    6.
AD：【爱分享精选好货商城，大量淘宝天猫隐藏优惠券】优惠券折扣直播第一站！每天更新万款实惠好券，人工筛选验货，限时限量特卖，全场1折包邮！
赞(0)
未经允许不得转载：爱分享 » TestNG数据驱动简介
分享按钮

评论 抢沙发

取消

分享本文到

分类目录

标签云

Alita的读书笔记 riskyy user_76241248 一勺思想 专题 个人成长 中信出版社 乔安 会计 八字 公司/个股 其他 华为 商业模式 团队管理 国际关系 墨小隐 学习教育 学习资料 小飞 心理学 战略 战略管理 投资 教育 数字化转型 明冬亮-深夜读书思考者 机械工业出版社 沟通 科技 管理 组织发展 经管职场 职场 职场增值 职场精英进化论 营销 行研 读书笔记 财务 起点书院 起点学院 领导力 高考 麦肯锡

热门文章

爱分享，生活常用知识教程百科分享、学习、交流平台

爱分享精选好货商城