首页 > 编程 > Java > 正文

java 与testng利用XML做数据源的数据驱动示例详解

2019-11-26 13:17:02
字体:
来源:转载
供稿:网友

java 与testng利用XML做数据源的数据驱动示例详解

testng的功能很强大,利用@DataProvider可以做数据驱动,数据源文件可以是EXCEL,XML,YAML,甚至可以是TXT文本。在这以XML为例:

备注:@DataProvider的返回值类型只能是Object[][]与Iterator<Object>[]

TestData.xml:

<?xml version="1.0" encoding="UTF-8"?><data>  <testmethod1>    <input>1</input>    <button>2</button>  </testmethod1>  <testmethod1>    <input>3</input>    <button>4</button>  </testmethod1>  <testmethod2>    <input>3</input>    <button>4</button>  </testmethod2>  <testmethod3>    <input>3</input>    <button>4</button>  </testmethod3>  <testmethod4>    <input>3</input>    <button>4</button>  </testmethod4></data>

处用DOM4J解析XML,ParserXml.java文件:

package com.test;import java.io.File;import java.util.ArrayList;import java.util.HashMap;import java.util.Iterator;import java.util.List;import java.util.Map;import org.dom4j.Document;import org.dom4j.DocumentException;import org.dom4j.Element;import org.dom4j.io.SAXReader;public class ParserXml {  public List parser3Xml(String fileName) {    File inputXml = new File(fileName);      List list=new ArrayList();            int count = 1;    SAXReader saxReader = new SAXReader();    try {      Document document = saxReader.read(inputXml);      Element employees = document.getRootElement();      for (Iterator i = employees.elementIterator(); i.hasNext();) {        Element employee = (Element) i.next();        Map map = new HashMap();        Map tempMap = new HashMap();        for (Iterator j = employee.elementIterator(); j.hasNext();) {          Element node = (Element) j.next();                    tempMap.put(node.getName(), node.getText());                  }        map.put(employee.getName(), tempMap);        list.add(map);      }    } catch (DocumentException e) {      System.out.println(e.getMessage());    }    return list;  }     }

然后把解析出来的list转换成Object[][]类型的数据,且结合在@DataProvider中。

TestData.java文件:

package com.test;import java.io.File;import java.lang.reflect.Method;import java.util.ArrayList;import java.util.HashMap;import java.util.Iterator;import java.util.List;import java.util.Map;import org.testng.annotations.DataProvider;public class TestData {        private List l;    public TestData() {      this.getXmlData();      }    public void getXmlData(){    ParserXml p = new ParserXml();    l = p.parser3Xml(new File("src/com/test/TestData.xml").getAbsolutePath());  }  @DataProvider  public Object[][] providerMethod(Method method){        List<Map<String, String>> result = new ArrayList<Map<String, String>>();        for (int i = 0; i < l.size(); i++) {      Map m = (Map) l.get(i);        if(m.containsKey(method.getName())){                      Map<String, String> dm = (Map<String, String>) m.get(method.getName());        result.add(dm);        }    }     Object[][] files = new Object[result.size()][];    for(int i=0; i<result.size(); i++){      files[i] = new Object[]{result.get(i)};    }        return files;  }  }

再通过测试文件来测试一下:

TestDataProvider.java文件:

package com.test;import java.util.Map;import org.testng.annotations.*;public class TestDataProvider extends TestData {  @Test(dataProvider="providerMethod")  public void testmethod1(Map<?, ?> param){    System.out.println("method1 received:"+param.get("input"));  }     @Test(dataProvider="providerMethod")  public void testmethod2(Map<?, ?> param){    System.out.println("method2 received:"+param.get("input"));  }     @Test(dataProvider="providerMethod")  public void testmethod3(Map<?, ?> param){    System.out.println("method3 received:"+param.get("input"));  }    @Test  public void testmethod4(){    System.out.println("method4 received:4");  }}

我们再回过头来分析一下XML文件,有两个testmethod1结点,testmethod2,testmethod3,testmethod4结点各一个,在TestDataProvider.java文件中,定义了testmethod1,testmethod2,testmethod3,testmethod4四个测试函数,且testmethod4没有用到dataProvider,所以运行结果最后应该是testmethod1运行两遍,testmethod2,testmethod3,testmethod4各运行一遍,结果如下:

method1 received:1method1 received:3method2 received:3method3 received:3method4 received:4PASSED: testmethod1({input=1, button=2})PASSED: testmethod1({input=3, button=4})PASSED: testmethod2({input=3, button=4})PASSED: testmethod3({input=3, button=4})PASSED: testmethod4===============================================  Default test  Tests run: 5, Failures: 0, Skips: 0===============================================

也就是说通过这种方式,只需要把测试函数先写好,然后在XML文件中定义好数据就行了,就能控制函数是否运行,运行次数且运行的数据。

OK,Let's try.....

感谢阅读,希望能帮助到大家,谢谢大家对本站的支持!

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表