轉載：https://blog.csdn.net/heart_1014/article/details/52013173

使用@DataProvider註解定義當前方法中的返回對象CSV文件（存放測試數據）作為測試腳本的測試數據集進行數據驅動。

用法參考代碼：

代碼在搜索完成後使用顯式等待方式，確認頁面已經加載完成，頁面底部的關鍵字"搜索幫助"已經顯示在頁面上

1. //從CSV文件中讀取每行中前2個逗號分割的中文詞作為搜索框中輸入的搜索關鍵詞
2. //斷言搜索結果頁面是否包含CSV文件中每行的最後一個詞匯的關鍵字
3. import java.io.BufferedReader;
4. import java.io.FileInputStream;
5. import java.io.IOException;
6. import java.io.InputStreamReader;
7. import java.util.ArrayList;
8. import java.util.List;
9. import java.util.concurrent.TimeUnit;
11. import org.openqa.selenium.By;
12. import org.openqa.selenium.WebDriver;
13. import org.openqa.selenium.firefox.FirefoxDriver;
14. import org.openqa.selenium.support.ui.ExpectedCondition;
15. import org.openqa.selenium.support.ui.WebDriverWait;
16. import org.testng.Assert;
17. import org.testng.annotations.AfterMethod;
18. import org.testng.annotations.BeforeMethod;
19. import org.testng.annotations.DataProvider;
20. import org.testng.annotations.Test;
22. public class TestDataByCSVFile {
23. private static WebDriver driver;
24. @DataProvider(name="searchData")
25. public static Object[][] data() throws IOException
26. {
27. return getSearchData("E:\\AutoData\\testData.csv");//獲取CSV文件的測試數據
28. }
29. @Test(dataProvider="searchData")
30. public void testSearch(String searchdata1,String searchdata2,String searchResult) {
31. //打開sogou首頁
32. driver.get("http://www.sogou.com/");
33. //輸入搜索條件
34. //從CSV文件中讀取每行中前2個逗號分割的中文詞作為搜索框中輸入的搜索關鍵詞,在兩個搜索詞中間增加一個空格
35. driver.findElement(By.id("query")).sendKeys(searchdata1+" "+searchdata2);
36. //單擊搜索按鈕
37. driver.findElement(By.id("stb")).click();
39. //使用顯式等待方式，確認頁面已經加載完成，頁面底部的關鍵字"搜索幫助"已經顯示在頁面上
40. (new WebDriverWait(driver,3)).until(new ExpectedCondition<Boolean>(){
42. @Override
43. public Boolean apply(WebDriver d) {
44. return d.findElement(By.id("sogou_webhelp")).getText().contains("搜索幫助");
45. }});
47. //斷言搜索結果頁面是否包含CSV文件中每行的最後一個詞匯的關鍵字
48. Assert.assertTrue(driver.getPageSource().contains(searchResult));
49. }
50. @BeforeMethod
51. public void beforeMethod() {
52. //若無法打開Firefox瀏覽器，可設定Firefox瀏覽器的安裝路徑
53. System.setProperty("webdriver.firefox.bin", "D:\\Program Files (x86)\\Mozilla Firefox\\firefox.exe");
54. //打開Firefox瀏覽器
55. driver=new FirefoxDriver();
56. //設定等待時間為5秒
57. driver.manage().timeouts().implicitlyWait(5, TimeUnit.SECONDS);
58. }
60. @AfterMethod
61. public void afterMethod() {
62. //關閉打開的瀏覽器
63. driver.quit();
64. }
65. //讀取CSV文件的靜態方法，使用CSV文件的絕對文件路徑作為函數參數
66. public static Object[][] getSearchData(String FileNameroot) throws IOException{
67. List<Object[]> records=new ArrayList<Object[]>();
68. String record;
69. //設定UTF-8字符集，使用帶緩沖區的字符輸入流BufferedReader讀取文件內容
70. BufferedReader file=new BufferedReader(new InputStreamReader(new FileInputStream(FileNameroot),"UTF-8"));
71. //忽略讀取CSV文件的標題行（第一行）
72. file.readLine();
73. //遍歷讀取文件中除第一行外的其他所有內容並存儲在名為records的ArrayList中，每一行records中存儲的對象為一個String數組
74. while((record=file.readLine())!=null){
75. String fields[]=record.split(",");
76. records.add(fields);
77. }
78. //關閉文件對象
79. file.close();
80. //將存儲測試數據的List轉換為一個Object的二維數組
81. Object[][] results=new Object[records.size()][];
82. //設置二位數組每行的值，每行是一個Object對象
83. for(int i=0;i<records.size();i++){
84. results[i]=records.get(i);
85. }
86. return results;
87. }
88. }

運行結果：

1. PASSED: testSearch("老九門", "演員", "趙麗穎")
2. PASSED: testSearch("X站警天啟", "導演", "布萊恩·辛格")
3. PASSED: testSearch("誅仙青雲誌", "編劇", "張戩")
5. ===============================================
6. Default test
7. Tests run: 3, Failures: 0, Skips: 0
8. ===============================================

測試數據的CSV文件內容：

搜索關鍵詞1,搜索關鍵詞2,搜索結果
老九門,演員,趙麗穎
X站警天啟,導演,布萊恩·辛格
誅仙青雲誌,編劇,張戩

註意：使用寫字板程序編輯CSV文件內容，在保存文件時要將文件存儲為UTF-8編碼格式。

數據驅動測試二：使用TestNG和CSV文件進行數據驅動