java结合testng，利用XML做数据源的数据驱动示例

原文地址：http://www.cnblogs.com/zhangfei/archive/2012/10/10/2718989.html

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

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

TestData.xml：

[html] view plain copy

1. <?xml version="1.0" encoding="UTF-8"?>  
2. <data>  
3.     <testmethod1>  
4.         <input>1</input>  
5.         <button>2</button>  
6.     </testmethod1>  
7.     <testmethod1>  
8.         <input>3</input>  
9.         <button>4</button>  
10.     </testmethod1>  
11.     <testmethod2>  
12.         <input>3</input>  
13.         <button>4</button>  
14.     </testmethod2>  
15.     <testmethod3>  
16.         <input>3</input>  
17.         <button>4</button>  
18.     </testmethod3>  
19.     <testmethod4>  
20.         <input>3</input>  
21.         <button>4</button>  
22.     </testmethod4>  
23. </data>  

处用DOM4J解析XML，ParserXml.java文件：

[java] view plain copy

1. package com.test;  
2.   
3. import java.io.File;  
4. import java.util.ArrayList;  
5. import java.util.HashMap;  
6. import java.util.Iterator;  
7. import java.util.List;  
8. import java.util.Map;  
9.   
10. import org.dom4j.Document;  
11. import org.dom4j.DocumentException;  
12. import org.dom4j.Element;  
13. import org.dom4j.io.SAXReader;  
14.   
15. public class ParserXml {  
16.   
17.     public List parser3Xml(String fileName) {  
18.         File inputXml = new File(fileName);      
19.         List list=new ArrayList();                  
20.         int count = 1;  
21.         SAXReader saxReader = new SAXReader();  
22.         try {  
23.             Document document = saxReader.read(inputXml);  
24.             Element employees = document.getRootElement();  
25.             for (Iterator i = employees.elementIterator(); i.hasNext();) {  
26.                 Element employee = (Element) i.next();  
27.                 Map map = new HashMap();  
28.                 Map tempMap = new HashMap();  
29.                 for (Iterator j = employee.elementIterator(); j.hasNext();) {  
30.                     Element node = (Element) j.next();                      
31.                     tempMap.put(node.getName(), node.getText());                      
32.                 }  
33.                 map.put(employee.getName(), tempMap);  
34.                 list.add(map);  
35.             }  
36.         } catch (DocumentException e) {  
37.             System.out.println(e.getMessage());  
38.         }  
39.         return list;  
40.     }      
41.        
42.   
43. }  

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

TestData.java文件

[java] view plain copy

1. package com.test;  
2.   
3. import java.io.File;  
4. import java.lang.reflect.Method;  
5. import java.util.ArrayList;  
6. import java.util.HashMap;  
7. import java.util.Iterator;  
8. import java.util.List;  
9. import java.util.Map;  
10.   
11. import org.testng.annotations.DataProvider;  
12.   
13. public class TestData {          
14.       
15.     private List l;  
16.       
17.     public TestData() {      
18.         this.getXmlData();          
19.     }  
20.       
21.     public void getXmlData(){  
22.         ParserXml p = new ParserXml();  
23.         l = p.parser3Xml(new File("src/com/test/TestData.xml").getAbsolutePath());  
24.     }  
25.   
26.     @DataProvider  
27.     public Object[][] providerMethod(Method method){          
28.         List<Map<String, String>> result = new ArrayList<Map<String, String>>();          
29.         for (int i = 0; i < l.size(); i++) {  
30.             Map m = (Map) l.get(i);      
31.             if(m.containsKey(method.getName())){                              
32.                 Map<String, String> dm = (Map<String, String>) m.get(method.getName());  
33.                 result.add(dm);      
34.             }  
35.         }    
36.         Object[][] files = new Object[result.size()][];  
37.         for(int i=0; i<result.size(); i++){  
38.             files[i] = new Object[]{result.get(i)};  
39.         }          
40.         return files;  
41.     }  
42.       
43.   
44. }  

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

TestDataProvider.java文件：

[java] view plain copy

1. package com.test;  
2.   
3. import java.util.Map;  
4.   
5. import org.testng.annotations.*;  
6.   
7. public class TestDataProvider extends TestData {  
8.   
9.     @Test(dataProvider="providerMethod")  
10.     public void testmethod1(Map<?, ?> param){  
11.         System.out.println("method1 received:"+param.get("input"));  
12.     }  
13.        
14.     @Test(dataProvider="providerMethod")  
15.     public void testmethod2(Map<?, ?> param){  
16.         System.out.println("method2 received:"+param.get("input"));  
17.     }  
18.        
19.     @Test(dataProvider="providerMethod")  
20.     public void testmethod3(Map<?, ?> param){  
21.         System.out.println("method3 received:"+param.get("input"));  
22.     }  
23.       
24.     @Test  
25.     public void testmethod4(){  
26.         System.out.println("method4 received:4");  
27.     }  
28.   
29. }  

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

[java] view plain copy

1. method1 received:1  
2. method1 received:3  
3. method2 received:3  
4. method3 received:3  
5. method4 received:4  
6. PASSED: testmethod1({input=1, button=2})  
7. PASSED: testmethod1({input=3, button=4})  
8. PASSED: testmethod2({input=3, button=4})  
9. PASSED: testmethod3({input=3, button=4})  
10. PASSED: testmethod4  
11. ===============================================  
12.     Default test  
13.     Tests run: 5, Failures: 0, Skips: 0  
14. ===============================================  
15. 也就是说通过这种方式，只需要把测试函数先写好，然后在XML文件中定义好数据就行了，就能控制函数是否运行，运行次数且运行的数据。