数据驱动测试之——CSV+TestNG

对于利用Webdriver做自动化的童鞋，对于如何将元素或者输入数据如何和编码分离都应该不会太陌生，本着一边学习一边分享的心态，大概总结了一下关于利用CSV、XML以及Excel来存放数据，然后在结合TestNG来执行数据驱动测试。

下面是关于CSV+TestNG进行数据驱动的代码：

package com.util.datadriver;

import java.io.BufferedReader;

import java.io.FileInputStream;

import java.io.InputStreamReader;

import java.util.ArrayList;

import java.util.List;

import org.testng.annotations.DataProvider;

import org.testng.annotations.Test;

/**

 *

 * @author LTao 使用TestNG和CSV文件进行数据驱动

 */

public class DataProviderCSV {

    //TestNG定义一个DataProvider

    @DataProvider(name = "testData")

    public static Object[][] words() throws Exception {

        return getData("date\\testDate.csv");

    }

    //利用TestNG的特性，可以直接在方法中传入参数，这些参数从DataProvider中数据化传进来

    @Test(dataProvider = "testData")

    public void testDtaa(String input1, String input2, String result1) {

        System.out.println(input1);

        System.out.println(input2);

        System.out.println(result1);

    }

    public static Object[][] getData(String filePath) throws Exception {

        String line;

        //定义list来存放CSV中的数据，数组存放的是每行的数据

        List<String[]> list = new ArrayList<String[]>();

        //定义一个BufferedReader方便一行一行来读取csv中的数据

        BufferedReader br = new BufferedReader(new InputStreamReader(

                new FileInputStream(filePath)));

        //先读一行，第一行定义的是关键字对应的解释语言，不用存放到list中，

        br.readLine();

        while ((line = br.readLine()) != null) {

            String[] fileds = line.split(","); //获取每行的数据

            list.add(fileds);

        }

        br.close();

        //定义object[][]二维数组，用list.size()定义二维数组行的限度

        Object[][] result = new Object[list.size()][];

        for (int i = 0; i < list.size(); i++) {

            result[i] = list.get(i); //将CSV每行中的数据存放到二维数组中

        }

        return result;

    }

}

秒客网

数据驱动测试之——CSV+TestNG

相关文章