您的位置: 首页  >  文章  >  TestNG+Selenium Webdriver 数据(Excel)驱动的方法

TestNG+Selenium Webdriver 数据(Excel)驱动的方法

分类: 文章 • 2023-01-21 00:14:56

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

ExcelData.Java

[java] view plaincopy
  1. package com.annie;  
  2.   
  3. import java.io.File;  
  4. import java.util.Iterator;  
  5. import java.util.Map;  
  6. import java.util.TreeMap;  
  7. import java.util.regex.Matcher;  
  8. import jxl.*;  
  9.   
  10. public class ExcelData implements Iterator<Object[]> {  
  11.     private Workbook book = null;  
  12.     private Sheet sheet = null;  
  13.     private int rowNum = 0;//行数   
  14.     private int curRowNo = 0;//当前行数   
  15.     private int columnNum = 0;//列数   
  16.     private String[] columnnName;//列名   
  17. /*在TestNG中，由@DataProvider（dataProvider="name"）修饰的方法读取Exel时，调用此类的构造方法（此方法会得到列名并将当前行移到下一行）执行完后， 
  18. *转到TestNG自己的方法中去，然后由他们调用此类实现的hasNext()、next() 方法； 
  19. *得到一行数据，然后返回给由@Test（dataProvider="name"）修饰的方法，如此反复到数据读完为止。 
  20. * @param filepath Excel文件名 
  21. * @param casename用例名 
  22.  */  
  23.  public ExcelData(String filepath, String casename) {  
  24.         try {  
  25.             File directory = new File(".");  
  26.             String ss = "open.anniewang.newexcel.";  
  27.             book = Workbook.getWorkbook(new File(directory.getCanonicalPath()  
  28.                     + "\\resources\\"  
  29.                     + ss.replaceAll("\\.", Matcher.quoteReplacement("\\"))  
  30.                     + filepath + ".xls"));  
  31.             this.sheet = book.getSheet(casename);  
  32.             this.rowNum = sheet.getRows();  
  33.   
  34.             Cell[] c = sheet.getRow(0);  
  35.             this.columnNum = c.length;  
  36.             columnnName = new String[c.length];  
  37.             for (int i = 0; i < c.length; i++) {  
  38.                 columnnName[i] = c[i].getContents().toString();  
  39.             }  
  40.             this.curRowNo++;  
  41.   
  42.         } catch (Exception e) {  
  43.             e.printStackTrace();  
  44.         }  
  45.     }  
  46.   
  47.     @Override  
  48.     public boolean hasNext() {  
  49. /** 
  50. *方法功能：是否有下一条数据 
  51. *如果行数为0即空sheet或者 当前行数大于总行数 
  52. *就关闭对excel的操作返回false，否则返回true 
  53. */  
  54.  if (this.rowNum == 0 || this.curRowNo >= this.rowNum) {  
  55.             try {  
  56.                 book.close();  
  57.             } catch (Exception e) {  
  58.                 e.printStackTrace();  
  59.             }  
  60.             return false;  
  61.         } else  
  62.             return true;  
  63.     }  
  64.   
  65.     @Override  
  66.     public Object[] next() {  
  67. /* 方法功能：得到并返回下一行数据 
  68. * 使用for将一行的数据放入TreeMap中（TreeMap默认按照Key值升序排列，HashMap没有排序） 
  69. *然后将Map装入Object[]并返回，且将curRowNo当前行下移 
  70. */  
  71.  Cell[] c = sheet.getRow(this.curRowNo);  
  72.         Map<String, String> s = new TreeMap<String, String>();  
  73.         for (int i = 0; i < this.columnNum; i++) {  
  74.             String temp = "";  
  75.             try {  
  76.                 temp = c[i].getContents().toString();  
  77.             } catch (ArrayIndexOutOfBoundsException ex) {  
  78.                 temp = "";  
  79.             }  
  80.             s.put(this.columnnName[i], temp);  
  81.         }  
  82.   
  83.         Object r[] = new Object[1];  
  84.         r[0] = s;  
  85.         this.curRowNo++;  
  86.         return r;  
  87.     }  
  88.   
  89.     @Override  
  90.     public void remove() {  
  91.         throw new UnsupportedOperationException("remove unsupported.");  
  92.     }  
  93. }  
[java] view plain copy
  1. package com.annie;  
  2.   
  3. import java.io.File;  
  4. import java.util.Iterator;  
  5. import java.util.Map;  
  6. import java.util.TreeMap;  
  7. import java.util.regex.Matcher;  
  8. import jxl.*;  
  9.   
  10. public class ExcelData implements Iterator<Object[]> {  
  11.     private Workbook book = null;  
  12.     private Sheet sheet = null;  
  13.     private int rowNum = 0;//行数  
  14.     private int curRowNo = 0;//当前行数  
  15.     private int columnNum = 0;//列数  
  16.     private String[] columnnName;//列名  
  17. /*在TestNG中，由@DataProvider（dataProvider="name"）修饰的方法读取Exel时，调用此类的构造方法（此方法会得到列名并将当前行移到下一行）执行完后， 
  18. *转到TestNG自己的方法中去，然后由他们调用此类实现的hasNext()、next() 方法； 
  19. *得到一行数据，然后返回给由@Test（dataProvider="name"）修饰的方法，如此反复到数据读完为止。 
  20. * @param filepath Excel文件名 
  21. * @param casename用例名 
  22.  */  
  23.  public ExcelData(String filepath, String casename) {  
  24.         try {  
  25.             File directory = new File(".");  
  26.             String ss = "open.anniewang.newexcel.";  
  27.             book = Workbook.getWorkbook(new File(directory.getCanonicalPath()  
  28.                     + "\\resources\\"  
  29.                     + ss.replaceAll("\\.", Matcher.quoteReplacement("\\"))  
  30.                     + filepath + ".xls"));  
  31.             this.sheet = book.getSheet(casename);  
  32.             this.rowNum = sheet.getRows();  
  33.   
  34.             Cell[] c = sheet.getRow(0);  
  35.             this.columnNum = c.length;  
  36.             columnnName = new String[c.length];  
  37.             for (int i = 0; i < c.length; i++) {  
  38.                 columnnName[i] = c[i].getContents().toString();  
  39.             }  
  40.             this.curRowNo++;  
  41.   
  42.         } catch (Exception e) {  
  43.             e.printStackTrace();  
  44.         }  
  45.     }  
  46.   
  47.     @Override  
  48.     public boolean hasNext() {  
  49. /** 
  50. *方法功能：是否有下一条数据 
  51. *如果行数为0即空sheet或者 当前行数大于总行数 
  52. *就关闭对excel的操作返回false，否则返回true 
  53. */  
  54.  if (this.rowNum == 0 || this.curRowNo >= this.rowNum) {  
  55.             try {  
  56.                 book.close();  
  57.             } catch (Exception e) {  
  58.                 e.printStackTrace();  
  59.             }  
  60.             return false;  
  61.         } else  
  62.             return true;  
  63.     }  
  64.   
  65.     @Override  
  66.     public Object[] next() {  
  67. /* 方法功能：得到并返回下一行数据 
  68. * 使用for将一行的数据放入TreeMap中（TreeMap默认按照Key值升序排列，HashMap没有排序） 
  69. *然后将Map装入Object[]并返回，且将curRowNo当前行下移 
  70. */  
  71.  Cell[] c = sheet.getRow(this.curRowNo);  
  72.         Map<String, String> s = new TreeMap<String, String>();  
  73.         for (int i = 0; i < this.columnNum; i++) {  
  74.             String temp = "";  
  75.             try {  
  76.                 temp = c[i].getContents().toString();  
  77.             } catch (ArrayIndexOutOfBoundsException ex) {  
  78.                 temp = "";  
  79.             }  
  80.             s.put(this.columnnName[i], temp);  
  81.         }  
  82.   
  83.         Object r[] = new Object[1];  
  84.         r[0] = s;  
  85.         this.curRowNo++;  
  86.         return r;  
  87.     }  
  88.   
  89.     @Override  
  90.     public void remove() {  
  91.         throw new UnsupportedOperationException("remove unsupported.");  
  92.     }  
  93. }  

EXECL数据驱动：ExcelTest.xls

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

TestNG+Selenium Webdriver 数据(Excel)驱动的方法


测试类调用：TheExcelTest.java

[java] view plaincopy
  1. package com.annie;  
  2.   
  3. import java.util.Iterator;  
  4. import java.util.Map;  
  5. import java.util.Set;  
  6.   
  7. import org.testng.annotations.DataProvider;  
  8. import org.testng.annotations.Test;  
  9.   
  10. public class TheExcelTest {  
  11.       
  12.     @Test(dataProvider = "db1")  
  13.     public void ts(Map<String, String> data) throws Exception{  
  14.         this.prmap(data);  
  15.         System.out.println("=====over=====");  
  16.         System.out.println("");  
  17.     }  
  18.       
  19.     @DataProvider(name = "db1")  
  20.     public Iterator<Object[]> data() throws Exception{  
  21.         return (Iterator<Object[]>)new ExcelData("ExcelTest","testB");  
  22.     }  
  23.       
  24.     public  void prmap(Map<String,String>arr){  
  25.         Set<String> set=arr.keySet();  
  26.         Iterator<String> it=set.iterator();  
  27.         while(it.hasNext()){  
  28.             String s=(String)it.next();  
  29.              System.out.println(arr.get(s));  
  30.         }  
  31.     }  
  32.   
  33. }  
[java] view plain copy
  1. package com.annie;  
  2.   
  3. import java.util.Iterator;  
  4. import java.util.Map;  
  5. import java.util.Set;  
  6.   
  7. import org.testng.annotations.DataProvider;  
  8. import org.testng.annotations.Test;  
  9.   
  10. public class TheExcelTest {  
  11.       
  12.     @Test(dataProvider = "db1")  
  13.     public void ts(Map<String, String> data) throws Exception{  
  14.         this.prmap(data);  
  15.         System.out.println("=====over=====");  
  16.         System.out.println("");  
  17.     }  
  18.       
  19.     @DataProvider(name = "db1")  
  20.     public Iterator<Object[]> data() throws Exception{  
  21.         return (Iterator<Object[]>)new ExcelData("ExcelTest","testB");  
  22.     }  
  23.       
  24.     public  void prmap(Map<String,String>arr){  
  25.         Set<String> set=arr.keySet();  
  26.         Iterator<String> it=set.iterator();  
  27.         while(it.hasNext()){  
  28.             String s=(String)it.next();  
  29.              System.out.println(arr.get(s));  
  30.         }  
  31.     }  
  32.   
  33. }  

右键：RUN-as- TestNG

TestNG+Selenium Webdriver 数据(Excel)驱动的方法

或者运行 RUN -as-ANT build

TestNG+Selenium Webdriver 数据(Excel)驱动的方法

路径下报告：

TestNG+Selenium Webdriver 数据(Excel)驱动的方法


TestNG+Selenium Webdriver 数据(Excel)驱动的方法

相关推荐