基于testNg的测试框架设计（一）数据驱动

所谓的数据驱动是指测试的逻辑时一样的，发生改变的仅仅是数据，期望不同的测试数据输出不同或者相同的测试结果。

测试数据和测试脚本的分离，有利于分别维护测试数据和脚本，使得逻辑更加清晰，更容易交接和维护。

testNg本身提供了@DataProvider的注解去加载测试数据，还有通过testng.xml配置参数的方式进行。我在此基础上做改动，达到更号的效果。

实现目标：

1、测试脚本和测试数据分离，达到不脚本和数据分开维护。

2、测试脚本需要的测试数据，通过@DataProvider加载进来。

3、测试数据使用csv文件来维护，包括测试编号，测试描述，测试输入数据，测试期望数据

测试框架的大概结构如下：

从csv文件中读取数据并返回的代码实现过程：

/** *  * @author kaifeng.shi * @version $Id: DataProviderFromCsv.java, v 0.1 2014年8月24日 下午10:52:14  Exp $ * @return Object[][] 返回一个二维数组，第一维代表行号，第二维代表每行数据 */import java.io.FileReader;import java.io.IOException;import java.lang.reflect.Method;import java.util.ArrayList;import org.testng.ITestContext;import org.testng.annotations.DataProvider;import au.com.bytecode.opencsv.CSVReader;public class DataProviderFromCsv {// 定义csv文件的分割符public static char csvSeprator = ',';// 定义@DataProvider的namepublic static final String dataProviderName = "dataProviderName";@DataProvider(name = dataProviderName)public static Object[][] getCsvData(Method method, ITestContext context)throws IOException {// 获取当前文件编译后的绝对路径DataProviderFromCsv dataProviderFromCsv = new DataProviderFromCsv();String retPath = dataProviderFromCsv.getPath();// 获取调用方法的方法名String methodName = method.getName();// 获取调用方法的类名String className = null;String[] packageName = method.getDeclaringClass().getName().toString().split("\\.");className = packageName[packageName.length - 1];// 指定当前类当前方法对应的csv文件String csvPath = retPath + className + "." + methodName + ".csv";// 从CSV文件中读取数据CSVReader reader = new CSVReader(new FileReader(csvPath), csvSeprator);// 不读第一行,第一行统一为参数的字段名字reader.readNext();// csv中每行的数据都是一个string数组String[] csvRow = null;ArrayList<Object[]> csvList = new ArrayList<Object[]>();// 将读取的数据，按行存入到csvList中while ((csvRow = reader.readNext()) != null) {csvList.add(csvRow);}// 定义返回值Object[][] results = new Object[csvList.size()][];// 设置二维数组每行的值，每行是个object对象for (int i = 0; i < csvList.size(); i++) {results[i] = csvList.get(i);}return results;}public String getPath() {String absolutePath = null;try {absolutePath = this.getClass().getResource("").getPath();} catch (Exception e) {e.printStackTrace();}return absolutePath;}}

以上代码做了几件事情：

   1）获取测试数据（csv文件）的路径，约定csv文件的命名方式为 ----类名+方法名+csv后缀。@Dateprovider 的入参中有个Method method参数，可以获取调用方法的方法名。

  2）获取数据文件的路径和文件名称后，通过opencsv提供的方法，读取文件的内容。需要添加opencsv的jar文件。

        2.1）csv文件的第一行一般用于填写列名，不作为测试数据，所以不返回第一行的内容。

        2.2）csv文件每行是一个object，这个object包含每列的列值，所以每行时个string[]数组。一行数据是一个object对象，里面为每行的内容，而每行的内容都按列对应的值存入。

  3）@Dataprovider要求返回一个Object[][]的二维数组，第一维表示的是行号（第几行），第二维表示的每行里面的内容值。所以将csv中每行的内容放入第二维中返回。

通过上述代码就简单的在testng的@dataprovider基础上进行了优化，实现了数据驱动，测试脚本和数据分离的功能。

testng本身实现的相关资料可参考：

http://www.shenyanchao.cn/blog/2013/06/05/deep-in-testng/