我有一个包含用户名和密码的登录页面。我有一个类XcelParserTestNGLogin来创建，更新和加载Excel工作表方法。另一个类Login是TestNG类。我正在使用DataProvider从Excel传递数据。但是我得到了异常数据提供者试图传递4个参数，但该方法需要2 。
这是我的TestNG代码：
public class Login {
private static WebDriver driver;
XcelParserTestNGLogin login1 = new XcelParserTestNGLogin();
Object[][] data1;
/*public Login() throws IOException, InterruptedException {
FileInputStream fis = new FileInputStream("Data//LoginPage.xls");
XcelParserTestNGLogin login1 = new XcelParserTestNGLogin(fis, "Login");
//this.data1 = login1.loadFromSpreadsheet(fis, "Login");
}*/
@BeforeClass
public void test() throws Exception {
System.setProperty("webdriver.chrome.driver",
"C:\\Chrome\\chromedriver_win_26.0.1383.0\\chromedriver.exe");
driver = new ChromeDriver();
driver.get("Any Url");
}
@DataProvider
public Object[][] dp() throws IOException {
//login1.fileName = "Data//Login.xls";
//login1.sheetName = "Sheet1";
FileInputStream fis = new FileInputStream("Data//LoginPage.xls");
String sheetName = "Login";
login1.loadFromSpreadsheet(fis,sheetName);
return login1.getData();
}
@Test(dataProvider = "dp")
public void devLogin(String UserName,String PassWord) throws InterruptedException, IOException {
driver.findElement(By.name("txtUserName")).sendKeys(UserName);
driver.findElement(By.name("txtPwd")).sendKeys(PassWord);
driver.findElement(By.name("btnSignIn")).click();
Thread.sleep(5000);
if (driver.findElement(By.linkText("DashBoard")).isDisplayed()) {
List<String> arrayList = new ArrayList<String>();
arrayList.add("Pass");
HSSFWorkbook workbook = new HSSFWorkbook();
login1.createSheet("Login", workbook, arrayList);
}
else {
try{
Alert alert=driver.switchTo().alert();
String alertText=alert.getText();
Assert.assertEquals("invalid username or password,please try again",alertText);
alert.accept();
}catch(UnhandledAlertException e){
e.printStackTrace();
}
List<String> arrayList = new ArrayList<String>();
arrayList.add("Fail");
HSSFWorkbook workbook = new HSSFWorkbook();
login1.createSheet("Login", workbook, arrayList);
}
}
}
这是我的XcelParserTestNGLogin（）
的代码public class XcelParserTestNGLogin {
private transient Object[][] data;
String fileName,sheetName;
public XcelParserTestNGLogin() {
}
public XcelParserTestNGLogin(InputStream excelInputStream, String sheetName)
throws IOException {
this.data = loadFromSpreadsheet(excelInputStream, sheetName);
}
public Object[][] getData() {
return data;
}
Object[][] loadFromSpreadsheet(InputStream excelFile, String sheetName)
throws IOException {
// TODO Auto-generated method stub
HSSFWorkbook workbook = new HSSFWorkbook(excelFile);
Sheet sheet = workbook.getSheet(sheetName);
int numberOfColumns = countNonEmptyColumns(sheet);
int numberOfRows = sheet.getLastRowNum() + 1;
data = new Object[numberOfRows - 1][numberOfColumns - 1];
for (int rowNum = 1; rowNum < numberOfRows; rowNum++) {
Row row = sheet.getRow(rowNum);
if (isEmpty(row)) {
break;
} else {
for (int column = 1; column < numberOfColumns; column++) {
Cell cell = row.getCell(column);
if (cell == null
|| cell.getCellType() == Cell.CELL_TYPE_BLANK) {
data[rowNum - 1][column - 1] = "";
} else {
data[rowNum - 1][column - 1] = objectFrom(workbook,
cell);
}
}
}
}
return data;
}
private boolean isEmpty(Row row) {
// TODO Auto-generated method stub
Cell firstCell = row.getCell(0);
boolean rowIsEmpty = (firstCell == null)
|| (firstCell.getCellType() == Cell.CELL_TYPE_BLANK);
return rowIsEmpty;
}
/**
* Count the number of columns, using the number of non-empty cells in the
* first row.
*/
private int countNonEmptyColumns(Sheet sheet) {
// TODO Auto-generated method stub
Row firstRow = sheet.getRow(0);
return firstEmptyCellPosition(firstRow);
}
private int firstEmptyCellPosition(Row cells) {
// TODO Auto-generated method stub
int columnCount = 0;
for (Cell cell : cells) {
if (cell.getCellType() == Cell.CELL_TYPE_BLANK) {
break;
}
columnCount++;
}
return columnCount;
}
private Object objectFrom(HSSFWorkbook workbook, Cell cell) {
// TODO Auto-generated method stub
Object cellValue = null;
if (cell.getCellType() == Cell.CELL_TYPE_BLANK) {
cellValue = cell.getRichStringCellValue().getString();
} else if (cell.getCellType() == Cell.CELL_TYPE_NUMERIC) {
cellValue = getNumericCellValue(cell);
} else if (cell.getCellType() == Cell.CELL_TYPE_BOOLEAN) {
cellValue = cell.getBooleanCellValue();
} else if (cell.getCellType() == Cell.CELL_TYPE_FORMULA) {
cellValue = evaluateCellFormula(workbook, cell);
}
return cellValue;
}
private Object getNumericCellValue(final Cell cell) {
Object cellValue;
if (DateUtil.isCellDateFormatted(cell)) {
cellValue = new Date(cell.getDateCellValue().getTime());
} else {
cellValue = cell.getNumericCellValue();
}
return cellValue;
}
private Object evaluateCellFormula(final HSSFWorkbook workbook,
final Cell cell) {
FormulaEvaluator evaluator = workbook.getCreationHelper()
.createFormulaEvaluator();
CellValue cellValue = evaluator.evaluate(cell);
Object result = null;
if (cellValue.getCellType() == Cell.CELL_TYPE_BOOLEAN) {
result = cellValue.getBooleanValue();
} else if (cellValue.getCellType() == Cell.CELL_TYPE_NUMERIC) {
result = cellValue.getNumberValue();
} else if (cellValue.getCellType() == Cell.CELL_TYPE_STRING) {
result = cellValue.getStringValue();
}
return result;
}
public void updateExcel(final InputStream excelFile, String SheetName,
List<String> list) {
HSSFWorkbook workbook = new HSSFWorkbook();
Sheet sheet = null;
if (workbook.getSheetIndex(SheetName) > 0) {
sheet = workbook.getSheet(SheetName);
if (list != null && list.size() != sheet.getLastRowNum()) {
workbook.removeSheetAt(workbook.getSheetIndex(SheetName));
createSheet(SheetName, workbook, list);
} else {
createSheet(SheetName, workbook, list);
}
}
}
void createSheet(String SheetName, HSSFWorkbook workbook, List<String> list) {
// TODO Auto-generated method stub
String[] Heading = {"UserName", "Password",
"Result" };
Sheet sheet = workbook.createSheet(SheetName);
HSSFRow row = null;
HSSFCell cell = null;
row = (HSSFRow) sheet.createRow(0);
for (int cellNum = 0; cellNum < Heading.length; cellNum++) {
cell = row.createCell(cellNum);
cell.setCellValue(Heading[cellNum]);
}
for (int rowNum = 1; rowNum <= list.size(); rowNum++) {
String[] cellVals = {"uname",
"pswd", list.get(rowNum - 1) };
row = (HSSFRow) sheet.createRow(rowNum);
for (int cellNum = 0; cellNum < cellVals.length; cellNum++) {
cell = row.createCell(cellNum);
if (!(cellNum == cellVals.length))
cell.setCellValue(cellVals[cellNum]);
else
cell.setCellValue(true);
}
}
try {
FileOutputStream out = new FileOutputStream("Data//LoginPage.xls");
workbook.write(out);
out.close();
} catch (Exception e) {
e.printStackTrace();
}
}
}
答案 0 :(得分：2)
记住对象数组左边第一个方括号的长度表示调用测试方法的次数。
第二个方括号的长度表示您的测试方法应该有多少个参数。根据您的测试方法的签名，您的数组声明应为obj = new Object[maxRows][2]。
答案 1 :(得分：0)
package testng;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.hssf.usermodel.HSSFHyperlink;
import org.apache.poi.hssf.util.HSSFColor;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.xssf.usermodel.*;
import java.io.*;
import java.util.Calendar;
public class ExcelReader {
public String path;
public FileInputStream fis = null;
public FileOutputStream fileOut =null;
private XSSFWorkbook workbook = null;
private XSSFSheet sheet = null;
private XSSFRow row =null;
private XSSFCell cell = null;
public ExcelReader(String path) {
this.path=path;
try {
fis = new FileInputStream(path);
workbook = new XSSFWorkbook(fis);
sheet = workbook.getSheetAt(0);
fis.close();
} catch (Exception e) {
e.printStackTrace();
}
}
// returns the row count in a sheet
public int getRowCount(String sheetName){
int index = workbook.getSheetIndex(sheetName);
if(index==-1)
return 0;
else{
sheet = workbook.getSheetAt(index);
int number=sheet.getLastRowNum()+1;
return number;
}
}
// returns the data from a cell
public String getCellData(String sheetName,String colName,int rowNum){
try{
if(rowNum <=0)
return "";
int index = workbook.getSheetIndex(sheetName);
int col_Num=-1;
if(index==-1)
return "";
sheet = workbook.getSheetAt(index);
row=sheet.getRow(0);
for(int i=0;i<row.getLastCellNum();i++){
//System.out.println(row.getCell(i).getStringCellValue().trim());
if(row.getCell(i).getStringCellValue().trim().equals(colName.trim()))
col_Num=i;
}
if(col_Num==-1)
return "";
sheet = workbook.getSheetAt(index);
row = sheet.getRow(rowNum-1);
if(row==null)
return "";
cell = row.getCell(col_Num);
if(cell==null)
return "";
if(cell.getCellType()==Cell.CELL_TYPE_STRING)
return cell.getStringCellValue();
else if(cell.getCellType()==Cell.CELL_TYPE_NUMERIC || cell.getCellType()==Cell.CELL_TYPE_FORMULA ){
String cellText = String.valueOf(cell.getNumericCellValue());
if (HSSFDateUtil.isCellDateFormatted(cell)) {
double d = cell.getNumericCellValue();
Calendar cal =Calendar.getInstance();
cal.setTime(HSSFDateUtil.getJavaDate(d));
cellText =
(String.valueOf(cal.get(Calendar.YEAR))).substring(2);
cellText = cal.get(Calendar.DAY_OF_MONTH) + "/" +
cal.get(Calendar.MONTH)+1 + "/" +
cellText;
}
return cellText;
}else if(cell.getCellType()==Cell.CELL_TYPE_BLANK)
return "";
else
return String.valueOf(cell.getBooleanCellValue());
}
catch(Exception e){
e.printStackTrace();
return "row "+rowNum+" or column "+colName +" does not exist in xls";
}
}
// returns the data from a cell
public String getCellData(String sheetName,int colNum,int rowNum){
try{
if(rowNum <=0)
return "";
int index = workbook.getSheetIndex(sheetName);
if(index==-1)
return "";
sheet = workbook.getSheetAt(index);
row = sheet.getRow(rowNum-1);
if(row==null)
return "";
cell = row.getCell(colNum);
if(cell==null)
return "";
if(cell.getCellType()==Cell.CELL_TYPE_STRING)
return cell.getStringCellValue();
else if(cell.getCellType()==Cell.CELL_TYPE_NUMERIC || cell.getCellType()==Cell.CELL_TYPE_FORMULA ){
String cellText = String.valueOf(cell.getNumericCellValue());
if (HSSFDateUtil.isCellDateFormatted(cell)) {
// format in form of M/D/YY
double d = cell.getNumericCellValue();
Calendar cal =Calendar.getInstance();
cal.setTime(HSSFDateUtil.getJavaDate(d));
cellText =
(String.valueOf(cal.get(Calendar.YEAR))).substring(2);
cellText = cal.get(Calendar.MONTH)+1 + "/" +
cal.get(Calendar.DAY_OF_MONTH) + "/" +
cellText;
}
return cellText;
}else if(cell.getCellType()==Cell.CELL_TYPE_BLANK)
return "";
else
return String.valueOf(cell.getBooleanCellValue());
}
catch(Exception e){
e.printStackTrace();
return "row "+rowNum+" or column "+colNum +" does not exist in xls";
}
}
// returns true if data is set successfully else false
public boolean setCellData(String sheetName,String colName,int rowNum, String data){
try{
fis = new FileInputStream(path);
workbook = new XSSFWorkbook(fis);
if(rowNum<=0)
return false;
int index = workbook.getSheetIndex(sheetName);
int colNum=-1;
if(index==-1)
return false;
sheet = workbook.getSheetAt(index);
row=sheet.getRow(0);
for(int i=0;i<row.getLastCellNum();i++){
//System.out.println(row.getCell(i).getStringCellValue().trim());
if(row.getCell(i).getStringCellValue().trim().equals(colName))
colNum=i;
}
if(colNum==-1)
return false;
sheet.autoSizeColumn(colNum);
row = sheet.getRow(rowNum-1);
if (row == null)
row = sheet.createRow(rowNum-1);
cell = row.getCell(colNum);
if (cell == null)
cell = row.createCell(colNum);
cell.setCellValue(data);
fileOut = new FileOutputStream(path);
workbook.write(fileOut);
fileOut.close();
}
catch(Exception e){
e.printStackTrace();
return false;
}
return true;
}// returns true if data is set successfully else false
public boolean setCellData(String sheetName,String colName,int rowNum, String data,String url){
try{
fis = new FileInputStream(path);
workbook = new XSSFWorkbook(fis);
if(rowNum<=0)
return false;
int index = workbook.getSheetIndex(sheetName);
int colNum=-1;
if(index==-1)
return false;
sheet = workbook.getSheetAt(index);
row=sheet.getRow(0);
for(int i=0;i<row.getLastCellNum();i++){
if(row.getCell(i).getStringCellValue().trim().equalsIgnoreCase(colName))
colNum=i;
}
if(colNum==-1)
return false;
sheet.autoSizeColumn(colNum);
row = sheet.getRow(rowNum-1);
if (row == null)
row = sheet.createRow(rowNum-1);
cell = row.getCell(colNum);
if (cell == null)
cell = row.createCell(colNum);
cell.setCellValue(data);
XSSFCreationHelper createHelper = workbook.getCreationHelper();
//cell style for hyperlinks
CellStyle hlink_style = workbook.createCellStyle();
XSSFFont hlink_font = workbook.createFont();
hlink_font.setUnderline(XSSFFont.U_SINGLE);
hlink_font.setColor(IndexedColors.BLUE.getIndex());
hlink_style.setFont(hlink_font);
//hlink_style.setWrapText(true);
XSSFHyperlink link = createHelper.createHyperlink(XSSFHyperlink.LINK_FILE);
link.setAddress(url);
cell.setHyperlink(link);
cell.setCellStyle(hlink_style);
fileOut = new FileOutputStream(path);
workbook.write(fileOut);
fileOut.close();
}
catch(Exception e){
e.printStackTrace();
return false;
}
return true;
}
// returns true if sheet is created successfully else false
public boolean addSheet(String sheetname){
FileOutputStream fileOut;
try {
workbook.createSheet(sheetname);
fileOut = new FileOutputStream(path);
workbook.write(fileOut);
fileOut.close();
} catch (Exception e) {
e.printStackTrace();
return false;
}
return true;
}
// returns true if sheet is removed successfully else false if sheet does not exist
public boolean removeSheet(String sheetName){
int index = workbook.getSheetIndex(sheetName);
if(index==-1)
return false;
FileOutputStream fileOut;
try {
workbook.removeSheetAt(index);
fileOut = new FileOutputStream(path);
workbook.write(fileOut);
fileOut.close();
} catch (Exception e) {
e.printStackTrace();
return false;
}
return true;
}
// returns true if column is created successfully
public boolean addColumn(String sheetName,String colName){
try{
fis = new FileInputStream(path);
workbook = new XSSFWorkbook(fis);
int index = workbook.getSheetIndex(sheetName);
if(index==-1)
return false;
XSSFCellStyle style = workbook.createCellStyle();
style.setFillForegroundColor(HSSFColor.GREY_40_PERCENT.index);
style.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);
sheet=workbook.getSheetAt(index);
row = sheet.getRow(0);
if (row == null)
row = sheet.createRow(0);
if(row.getLastCellNum() == -1)
cell = row.createCell(0);
else
cell = row.createCell(row.getLastCellNum());
cell.setCellValue(colName);
cell.setCellStyle(style);
fileOut = new FileOutputStream(path);
workbook.write(fileOut);
fileOut.close();
}catch(Exception e){
e.printStackTrace();
return false;
}
return true;
}// removes a column and all the contents
public boolean removeColumn(String sheetName, int colNum) {
try{
if(!isSheetExist(sheetName))
return false;
fis = new FileInputStream(path);
workbook = new XSSFWorkbook(fis);
sheet=workbook.getSheet(sheetName);
XSSFCellStyle style = workbook.createCellStyle();
style.setFillForegroundColor(HSSFColor.GREY_40_PERCENT.index);
XSSFCreationHelper createHelper = workbook.getCreationHelper();
style.setFillPattern(HSSFCellStyle.NO_FILL);
for(int i =0;i<getRowCount(sheetName);i++){
row=sheet.getRow(i);
if(row!=null){
cell=row.getCell(colNum);
if(cell!=null){
cell.setCellStyle(style);
row.removeCell(cell);
}
}
}
fileOut = new FileOutputStream(path);
workbook.write(fileOut);
fileOut.close();
}
catch(Exception e){
e.printStackTrace();
return false;
}
return true;
}
// find whether sheets exists
public boolean isSheetExist(String sheetName){
int index = workbook.getSheetIndex(sheetName);
if(index==-1){
index=workbook.getSheetIndex(sheetName.toUpperCase());
if(index==-1)
return false;
else
return true;
}
else
return true;
}
// returns number of columns in a sheet
public int getColumnCount(String sheetName){
// check if sheet exists
if(!isSheetExist(sheetName))
return -1;
sheet = workbook.getSheet(sheetName);
row = sheet.getRow(0);
if(row==null)
return -1;
return row.getLastCellNum();
}
//String sheetName, String testCaseName,String keyword ,String URL,String message
public boolean addHyperLink(String sheetName,String screenShotColName,String testCaseName,int index,String url,String message){
url=url.replace('\\', '/');
if(!isSheetExist(sheetName))
return false;
sheet = workbook.getSheet(sheetName);
for(int i=2;i<=getRowCount(sheetName);i++){
if(getCellData(sheetName, 0, i).equalsIgnoreCase(testCaseName)){
setCellData(sheetName, screenShotColName, i+index, message,url);
break;
}
}
return true;
}
public int getCellRowNum(String sheetName,String colName,String cellValue){
for(int i=2;i<=getRowCount(sheetName);i++){
if(getCellData(sheetName,colName , i).equalsIgnoreCase(cellValue)){
return i;
}
}
return -1;
}
// to run this on stand alone
public static void main(String arg[]) throws IOException{
ExcelReader datatable = null;
datatable = new ExcelReader("C:\\CM3.0\\app\\test\\Framework\\AutomationBvt\\src\\config\\xlfiles\\Controller.xlsx");
for(int col=0 ;col< datatable.getColumnCount("TC5"); col++){
System.out.println(datatable.getCellData("TC5", col, 1));
}
}
}
答案 2 :(得分：-1)
package testng;
import java.util.Hashtable;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
public class testNGParameterizationExcel {
public static ExcelReader excel = null;
@Test(dataProvider = "getData")
public void testData(Hashtable<String, String> data) {
// System.out.println(username + "---------" + password + "------------"
// + is_correct);
System.out.println(data.get("UserName"));
}
@DataProvider
public static Object[][] getData() {
if (excel == null) {
excel = new ExcelReader("C:\\Users\\shaanu\\Desktop\\Java\\Test.xlsx");
}
String sheetName = "new";
int rows = excel.getRowCount(sheetName);
int columns = excel.getColumnCount(sheetName);
Object[][] data = new Object[rows - 1][1];
Hashtable<String, String> table = null;
for (int rowNums = 2; rowNums <= rows; rowNums++) {
table = new Hashtable<String, String>();
for (int colNum = 0; colNum < columns; colNum++) {
// data[rowNums-2][colNum] = excel.getCellData(sheetName,
// colNum, rowNums);
table.put(excel.getCellData(sheetName, colNum, 1), excel.getCellData(sheetName, colNum, rowNums));
data[rowNums - 2][0] = table;
}
}
return data;
}
}