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

	}

}