JavaSelenium处理Excel数据驱动做测试

package com.gloryroad.testcase;



 import java.io.File;

 import java.io.FileInputStream;

 import java.io.FileNotFoundException;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.List;



 import org.apache.poi.hssf.usermodel.HSSFWorkbook;

 import org.apache.poi.ss.usermodel.Row;

 import org.apache.poi.ss.usermodel.Sheet;

 import org.apache.poi.ss.usermodel.Workbook;

 import org.apache.poi.xssf.usermodel.XSSFWorkbook;

 import org.openqa.selenium.By;

 import org.openqa.selenium.WebDriver;

 import org.openqa.selenium.firefox.FirefoxDriver;

 import org.testng.annotations.AfterMethod;

 import org.testng.annotations.BeforeMethod;

 import org.testng.annotations.DataProvider;

 import org.testng.annotations.Test;



 public class TestDataByExcelFile {

public WebDriver driver;

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

@BeforeMethod

public void setUp(){

driver=new FirefoxDriver();

driver.get(url);


}



@AfterMethod

public void tearDown(){

driver.close();

}


@DataProvider(name="testData")

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

{


return getData("C:\\JavaScript\\SeleniumProj\\resuorce", "3.xlsx", "Sheet1");


}


@Test(dataProvider="testData")

public void searchTest(String word1,String word2,String word3){

System.out.println(word1+word2+word3);

driver.findElement(By.id("kw")).sendKeys(word1+" "+word2);



}

/**

* 获取数据

* @throws IOException 

*/



public static Object[][] getData(String filePath,String fileName,String sheetName) throws IOException{

//制作File对象

File file=new File(filePath+"\\"+fileName);

FileInputStream input=new FileInputStream(file);

Workbook workbook=null;

//

String fileExtensionName=fileName.substring(fileName.indexOf("."));

if (fileExtensionName.equals(".xlsx")) {

workbook=new XSSFWorkbook(input);


}else if (fileExtensionName.equals(".xls")) {

workbook=new HSSFWorkbook(input);


}


//获取sheet

Sheet sheet=workbook.getSheet(sheetName);

//获取总行数

int row=sheet.getLastRowNum()-sheet.getFirstRowNum();


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

for (int i = 1; i < row+1; i++) {

Row rowValue=sheet.getRow(i);

//声明三个数组用来存储三个数据


String[] fileds=new String[rowValue.getLastCellNum()];


for (int j = 0; j < rowValue.getLastCellNum(); j++) {


fileds[j]=rowValue.getCell(j).getStringCellValue();


}


records.add(fileds);

}


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

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

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

}


return results;

}


 }