数据驱动测试二：使用TestNG和CSV文件进行数据驱动

使用@DataProvider注解定义当前方法中的返回对象CSV文件（存放测试数据）作为测试脚本的测试数据集进行数据驱动。

用法参考代码：

代码在搜索完成后使用显式等待方式，确认页面已经加载完成，页面底部的关键字"搜索帮助"已经显示在页面上

//从CSV文件中读取每行中前2个逗号分割的中文词作为搜索框中输入的搜索关键词
//断言搜索结果页面是否包含CSV文件中每行的最后一个词汇的关键字
import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.support.ui.ExpectedCondition;
import org.openqa.selenium.support.ui.WebDriverWait;
import org.testng.Assert;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

public class TestDataByCSVFile {
	private static WebDriver driver;
	@DataProvider(name="searchData")
	public static Object[][] data() throws IOException
	{
		return getSearchData("E:\\AutoData\\testData.csv");//获取CSV文件的测试数据
	}
  @Test(dataProvider="searchData")
  public void testSearch(String searchdata1,String searchdata2,String searchResult) {
	  //打开sogou首页
	  driver.get("http://www.sogou.com/");
	  //输入搜索条件
	  //从CSV文件中读取每行中前2个逗号分割的中文词作为搜索框中输入的搜索关键词,在两个搜索词中间增加一个空格
	  driver.findElement(By.id("query")).sendKeys(searchdata1+" "+searchdata2);
	  //单击搜索按钮
	  driver.findElement(By.id("stb")).click();
	  
	  //使用显式等待方式，确认页面已经加载完成，页面底部的关键字"搜索帮助"已经显示在页面上
	  (new WebDriverWait(driver,3)).until(new ExpectedCondition<Boolean>(){

		@Override
		public Boolean apply(WebDriver d) {
			return d.findElement(By.id("sogou_webhelp")).getText().contains("搜索帮助");
		}});

	  //断言搜索结果页面是否包含CSV文件中每行的最后一个词汇的关键字
	  Assert.assertTrue(driver.getPageSource().contains(searchResult));
  }
  @BeforeMethod
  public void beforeMethod() {
	  //若无法打开Firefox浏览器，可设定Firefox浏览器的安装路径
	  System.setProperty("webdriver.firefox.bin", "D:\\Program Files (x86)\\Mozilla Firefox\\firefox.exe");
	  //打开Firefox浏览器
	  driver=new FirefoxDriver();
	  //设定等待时间为5秒
	  driver.manage().timeouts().implicitlyWait(5, TimeUnit.SECONDS);
  }

  @AfterMethod
  public void afterMethod() {
	  //关闭打开的浏览器
	  driver.quit();
  }
  //读取CSV文件的静态方法，使用CSV文件的绝对文件路径作为函数参数
  public static Object[][] getSearchData(String FileNameroot) throws IOException{
	  List<Object[]> records=new ArrayList<Object[]>();
	  String record;
	  //设定UTF-8字符集，使用带缓冲区的字符输入流BufferedReader读取文件内容
	  BufferedReader file=new BufferedReader(new InputStreamReader(new FileInputStream(FileNameroot),"UTF-8"));
	  //忽略读取CSV文件的标题行（第一行）
	  file.readLine();
	  //遍历读取文件中除第一行外的其他所有内容并存储在名为records的ArrayList中，每一行records中存储的对象为一个String数组
	  while((record=file.readLine())!=null){
		  String fields[]=record.split(",");
		  records.add(fields);
	  }
	  //关闭文件对象
	  file.close();
	  //将存储测试数据的List转换为一个Object的二维数组
	  Object[][] results=new Object[records.size()][];
	  //设置二位数组每行的值，每行是一个Object对象
	  for(int i=0;i<records.size();i++){
		  results[i]=records.get(i);
	  }
	  return results;	  
  }
}

运行结果：

PASSED: testSearch("老九门", "演员", "赵丽颖")
PASSED: testSearch("X站警天启", "导演", "布莱恩·辛格")
PASSED: testSearch("诛仙青云志", "编剧", "张戬")

===============================================
    Default test
    Tests run: 3, Failures: 0, Skips: 0
===============================================

测试数据的CSV文件内容：

搜索关键词1,搜索关键词2,搜索结果
老九门,演员,赵丽颖
X站警天启,导演,布莱恩·辛格
诛仙青云志,编剧,张戬

注意：使用写字板程序编辑CSV文件内容，在保存文件时要将文件存储为UTF-8编码格式。