TestNG+Selenium Webdriver 数据(Excel)驱动的方法_testng selenium + excel-优快云博客

本文介绍了一种使用Excel作为数据源进行测试驱动的方法。通过自定义Java类实现迭代器接口，可以逐行读取Excel文件中的测试数据，并将其传递给TestNG测试用例。这种方法适用于需要大量测试数据的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.下载 jxl.jar 复制到测试项目的 lib 下，在项目中新建数据驱动类

ExcelData.Java

[java]view plaincopy 
   
 package com.annie;  
   
 import java.io.File;  
 import java.util.Iterator;  
 import java.util.Map;  
 import java.util.TreeMap;  
 import java.util.regex.Matcher;  
 import jxl.*;  
   
 public class ExcelData implements Iterator<Object[]> {  
     private Workbook book = null;  
     private Sheet sheet = null;  
     private int rowNum = 0;//行数   
     private int curRowNo = 0;//当前行数   
     private int columnNum = 0;//列数   
     private String[] columnnName;//列名   
 /*在TestNG中，由@DataProvider（dataProvider="name"）修饰的方法读取Exel时，调用此类的构造方法（此方法会得到列名并将当前行移到下一行）执行完后， 
 *转到TestNG自己的方法中去，然后由他们调用此类实现的hasNext()、next() 方法； 
 *得到一行数据，然后返回给由@Test（dataProvider="name"）修饰的方法，如此反复到数据读完为止。 
 * @param filepath Excel文件名 
 * @param casename用例名 
  */  
  public ExcelData(String filepath, String casename) {  
         try {  
             File directory = new File(".");  
             String ss = "open.anniewang.newexcel.";  
             book = Workbook.getWorkbook(new File(directory.getCanonicalPath()  
                     + "\\resources\\"  
                     + ss.replaceAll("\\.", Matcher.quoteReplacement("\\"))  
                     + filepath + ".xls"));  
             this.sheet = book.getSheet(casename);  
             this.rowNum = sheet.getRows();  
   
             Cell[] c = sheet.getRow(0);  
             this.columnNum = c.length;  
             columnnName = new String[c.length];  
             for (int i = 0; i < c.length; i++) {  
                 columnnName[i] = c[i].getContents().toString();  
             }  
             this.curRowNo++;  
   
         } catch (Exception e) {  
             e.printStackTrace();  
         }  
     }  
   
     @Override  
     public boolean hasNext() {  
 /** 
 *方法功能：是否有下一条数据 
 *如果行数为0即空sheet或者 当前行数大于总行数 
 *就关闭对excel的操作返回false，否则返回true 
 */  
  if (this.rowNum == 0 || this.curRowNo >= this.rowNum) {  
             try {  
                 book.close();  
             } catch (Exception e) {  
                 e.printStackTrace();  
             }  
             return false;  
         } else  
             return true;  
     }  
   
     @Override  
     public Object[] next() {  
 /* 方法功能：得到并返回下一行数据 
 * 使用for将一行的数据放入TreeMap中（TreeMap默认按照Key值升序排列，HashMap没有排序） 
 *然后将Map装入Object[]并返回，且将curRowNo当前行下移 
 */  
  Cell[] c = sheet.getRow(this.curRowNo);  
         Map<String, String> s = new TreeMap<String, String>();  
         for (int i = 0; i < this.columnNum; i++) {  
             String temp = "";  
             try {  
                 temp = c[i].getContents().toString();  
             } catch (ArrayIndexOutOfBoundsException ex) {  
                 temp = "";  
             }  
             s.put(this.columnnName[i], temp);  
         }  
   
         Object r[] = new Object[1];  
         r[0] = s;  
         this.curRowNo++;  
         return r;  
     }  
   
     @Override  
     public void remove() {  
         throw new UnsupportedOperationException("remove unsupported.");  
     }  
 }  

[java]view plaincopy 
   
 package com.annie;  
   
 import java.io.File;  
 import java.util.Iterator;  
 import java.util.Map;  
 import java.util.TreeMap;  
 import java.util.regex.Matcher;  
 import jxl.*;  
   
 public class ExcelData implements Iterator<Object[]> {  
     private Workbook book = null;  
     private Sheet sheet = null;  
     private int rowNum = 0;//行数  
     private int curRowNo = 0;//当前行数  
     private int columnNum = 0;//列数  
     private String[] columnnName;//列名  
 /*在TestNG中，由@DataProvider（dataProvider="name"）修饰的方法读取Exel时，调用此类的构造方法（此方法会得到列名并将当前行移到下一行）执行完后， 
 *转到TestNG自己的方法中去，然后由他们调用此类实现的hasNext()、next() 方法； 
 *得到一行数据，然后返回给由@Test（dataProvider="name"）修饰的方法，如此反复到数据读完为止。 
 * @param filepath Excel文件名 
 * @param casename用例名 
  */  
  public ExcelData(String filepath, String casename) {  
         try {  
             File directory = new File(".");  
             String ss = "open.anniewang.newexcel.";  
             book = Workbook.getWorkbook(new File(directory.getCanonicalPath()  
                     + "\\resources\\"  
                     + ss.replaceAll("\\.", Matcher.quoteReplacement("\\"))  
                     + filepath + ".xls"));  
             this.sheet = book.getSheet(casename);  
             this.rowNum = sheet.getRows();  
   
             Cell[] c = sheet.getRow(0);  
             this.columnNum = c.length;  
             columnnName = new String[c.length];  
             for (int i = 0; i < c.length; i++) {  
                 columnnName[i] = c[i].getContents().toString();  
             }  
             this.curRowNo++;  
   
         } catch (Exception e) {  
             e.printStackTrace();  
         }  
     }  
   
     @Override  
     public boolean hasNext() {  
 /** 
 *方法功能：是否有下一条数据 
 *如果行数为0即空sheet或者 当前行数大于总行数 
 *就关闭对excel的操作返回false，否则返回true 
 */  
  if (this.rowNum == 0 || this.curRowNo >= this.rowNum) {  
             try {  
                 book.close();  
             } catch (Exception e) {  
                 e.printStackTrace();  
             }  
             return false;  
         } else  
             return true;  
     }  
   
     @Override  
     public Object[] next() {  
 /* 方法功能：得到并返回下一行数据 
 * 使用for将一行的数据放入TreeMap中（TreeMap默认按照Key值升序排列，HashMap没有排序） 
 *然后将Map装入Object[]并返回，且将curRowNo当前行下移 
 */  
  Cell[] c = sheet.getRow(this.curRowNo);  
         Map<String, String> s = new TreeMap<String, String>();  
         for (int i = 0; i < this.columnNum; i++) {  
             String temp = "";  
             try {  
                 temp = c[i].getContents().toString();  
             } catch (ArrayIndexOutOfBoundsException ex) {  
                 temp = "";  
             }  
             s.put(this.columnnName[i], temp);  
         }  
   
         Object r[] = new Object[1];  
         r[0] = s;  
         this.curRowNo++;  
         return r;  
     }  
   
     @Override  
     public void remove() {  
         throw new UnsupportedOperationException("remove unsupported.");  
     }  
 }  

EXECL数据驱动：ExcelTest.xls

注意：此处的要用office2003扩展名为xls（office 2007 的excel 扩展名为xlsx），否则会报I/O 输入输出流的错误。

测试类调用：TheExcelTest.java

[java]view plaincopy 
   
 package com.annie;  
   
 import java.util.Iterator;  
 import java.util.Map;  
 import java.util.Set;  
   
 import org.testng.annotations.DataProvider;  
 import org.testng.annotations.Test;  
   
 public class TheExcelTest {  
       
     @Test(dataProvider = "db1")  
     public void ts(Map<String, String> data) throws Exception{  
         this.prmap(data);  
         System.out.println("=====over=====");  
         System.out.println("");  
     }  
       
     @DataProvider(name = "db1")  
     public Iterator<Object[]> data() throws Exception{  
         return (Iterator<Object[]>)new ExcelData("ExcelTest","testB");  
     }  
       
     public  void prmap(Map<String,String>arr){  
         Set<String> set=arr.keySet();  
         Iterator<String> it=set.iterator();  
         while(it.hasNext()){  
             String s=(String)it.next();  
              System.out.println(arr.get(s));  
         }  
     }  
   
 }  

[java]view plaincopy 
   
 package com.annie;  
   
 import java.util.Iterator;  
 import java.util.Map;  
 import java.util.Set;  
   
 import org.testng.annotations.DataProvider;  
 import org.testng.annotations.Test;  
   
 public class TheExcelTest {  
       
     @Test(dataProvider = "db1")  
     public void ts(Map<String, String> data) throws Exception{  
         this.prmap(data);  
         System.out.println("=====over=====");  
         System.out.println("");  
     }  
       
     @DataProvider(name = "db1")  
     public Iterator<Object[]> data() throws Exception{  
         return (Iterator<Object[]>)new ExcelData("ExcelTest","testB");  
     }  
       
     public  void prmap(Map<String,String>arr){  
         Set<String> set=arr.keySet();  
         Iterator<String> it=set.iterator();  
         while(it.hasNext()){  
             String s=(String)it.next();  
              System.out.println(arr.get(s));  
         }  
     }  
   
 }  