使用TestNG 和 CSV文件进行数据驱动

package testNGPractice;

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 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;

import cn.gloryroad.Log;

import scr.comm.OpenBrowserInfo;

public class TestDataDriverByCSVFile {

	public WebDriver driver;

	private String url = "http://www.sogou.com/";

	// 使用注解 DataProvider ,,将数据结合命名为“TestData”

	@DataProvider(name = "testData")

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

		// 调用类中的静态方法 getTestData，获取CSV文件的测试数据；

		return getTestData("D:\\dataTest.csv");

	}

	@Test(dataProvider = "testData")

	public void TestSearch(String serachWord1, String searchWord2, String searchResult) {

		Log.startTextCase("TestSearch" + serachWord1 + "	" + searchWord2 + " " + searchResult);

		// 打开搜狗首页；

		driver.get(url);

		// 使用CSV文件中每个数据行的前俩个词汇作为搜索词！

		// 在搜索词中间增加一个空格；

		driver.findElement(By.id("query")).sendKeys(serachWord1 + "" + searchWord2);

		// 单击搜索按钮；

		driver.findElement(By.id("stb")).click();

		// 使用显示等待方式 ，确定页面已经加载完成，页面底部的关键字“搜索帮助已经显示在页面上”

		(new WebDriverWait(driver, 10)).until(new ExpectedCondition<Boolean>() {

			@Override

			public Boolean apply(WebDriver d) {

				return d.findElement(By.id("s_footer")).getText().contains("搜索帮助");

			}

		});

		Assert.assertTrue(driver.getPageSource().contains(searchResult));

		Log.endTestCase("TestSearch");

	}

	@BeforeMethod

	public void beforeMethod() {

		OpenBrowserInfo.Log4jInitialize();

		OpenBrowserInfo.SystemSetProperty();

		driver = new FirefoxDriver();

	}

	@AfterMethod

	public void afterMethod() {

		driver.quit();

	}

	// 读取CSV文件的静态方法，使用CSV文件的绝对文件路径作为函数参数；

	public static Object[][] getTestData(String fileName) throws IOException {

		List<Object[]> records = new ArrayList<Object[]>();

		String record;

		// 设定UTF-8字符编码集，使用带缓冲区的字符输入流BufferedReader 读取文件内容；

		BufferedReader file = new BufferedReader(new InputStreamReader(new FileInputStream(fileName), "UTF-8"));

		// 忽略读取CSV文件的标题行（第一行！）；

		file.readLine();

		/*

		 * 遍历读取CSV文件中除第一行外其他所有内容，并存在在名为records的ArrayList中

		 * 每一个records中存储的对象为一个String数组；

		 */

		while ((record = file.readLine()) != null) {

			String fields[] = record.split(",");

			records.add(fields);

		}

		// 关闭文件对象；

		file.close();

		// 第一函数返回值，即Object[][]

		// 将存储测试数据的list 转换为一个Objectde 的二维数组；

		Object[][] results = new Object[records.size()][];

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

			results[i] =  records.get(i);

		}

		return results;

	}

}