TestNG中DataProvider的用法二：简单的数据驱动

@DataProvider 标记的方法除了可以返回数组外，还可以返回一个Iterator，这样的好处是不用把所有的测试数据都加载到内存中，而是需要的时候就读一条。

下面的例子就使用了Iterator，然后在Iterator对象中保存了一个文件对象，在需要参数的时候就从文件中读一行数据出来。

这个例子主要由如下四个文件组成：

1. Person类 被测试类
2. TxtIterator类 实现Iterator接口，用于从文件中读取数据，返回给被测试类
3. testng.xml TestNG配置文件
4. data.txt 参数池文件

Person主类（被测试类）

package ngtest;

import java.io.File;

import java.io.IOException;

import java.util.Iterator;

import org.testng.annotations.DataProvider;

import org.testng.annotations.Parameters;

import org.testng.annotations.Test;

/**

 * @author Administrator

 *

 */

public class Person {

 String dataFile;

 @Parameters({ "file" })  //parameter可以添加在类的构造方法上

 public Person(String dataFile) {

  super();

  this.dataFile = dataFile;

 }

 @Test(dataProvider = "provider") //被测试方法

 public void getName(String name, String age) {

  System.out.println(name + "_" + age);

 }

 @DataProvider(name = "provider")

 public Iterator<String[]> provider() throws IOException {

  return new TxtIterator(new File(dataFile));

 }

}

TxtIterator类

package ngtest;

import java.io.BufferedReader;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileReader;

import java.io.IOException;

import java.util.Iterator;

public class TxtIterator implements Iterator<String[]> {

 File txtFile;           //数据文件

 BufferedReader bs;

 String currentLine;

 public TxtIterator(File txtFile) throws IOException {

  super();

  this.txtFile = txtFile;

  try {

   bs = new BufferedReader(new FileReader(txtFile));

  } catch (FileNotFoundException e) {

   System.out.println("文件找不到");

   e.printStackTrace();

  }

  currentLine = bs.readLine();

 }

 @Override

 public boolean hasNext() {

  if (currentLine != null) {

   return true;

  } else {

   return false;

  }

 }

 @Override

 public String[] next() {

  String returnLine = currentLine;

  try {

   currentLine = bs.readLine();

  } catch (IOException e) {

   e.printStackTrace();

  }

  return returnLine.split(",");

 }

}

TestNG配置文件testng.xml

<?xml version="1.0" encoding="UTF-8"?>

<!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd">

<suite name="Suite">

 <parameter name="file" value="data.txt"/><!--数据文件的路径-->

 <test name="Test1">

  <classes>

   <class name="ngtest.Person" />

  </classes>

 </test>

</suite> 

数据文件data.txt

Liming,12

HanMeimei,13

Lily,11

Lucy,12

执行后输出

[TestNG] Running:

  D:\workspace\tester\testng.xml

Liming_12

HanMeimei_13

Lily_11

Lucy_12

===============================================

Suite

Total tests run: 4, Failures: 0, Skips: 0

===============================================

简单说一下上面的运行过程：

1. 运行testng.xml，发现要运行的方法在Person类中，这就要构造一个Person类的对象，发现Person类的构造方法包含一个file参数，于是从testng.xml文件中把file参数的值取出来，传给了构造方法，创建了Person对象
2. 开始执行测试，发现该测试方法需要一个DataProvider，于是在本类中找到了provider()方法，执行该方法，构造出Iterator对象，传递给测试方法。Iterator对象使用了Person对象中的dataFile成员变量的值构造出一个BufferedReader对象，每当测试方法需要一条数据时就由BufferedReader读一条数据出来，再拆分成数组，返回给测试方法调用。这样就实现了数据驱动和参数懒加载。