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

梦幻小丑灯

1. testng的功能很强大，利用@DataProvider可以做数据驱动，数据源文件可以是EXCEL，XML，YAML，甚至可以是TXT文本。在这以XML为例：
   
2. 
   
3. 备注：@DataProvider的返回值类型只能是Object[][]与Iterator<Object>[]
   
4. 
   
5. TestData.xml：
   
6. 
   
7. 复制代码
   
8. <?xml version="1.0" encoding="UTF-8"?>
   
9. <data>
   
10.     <testmethod1>
    
11.         <input>1</input>
    
12.         <button>2</button>
    
13.     </testmethod1>
    
14.     <testmethod1>
    
15.         <input>3</input>
    
16.         <button>4</button>
    
17.     </testmethod1>
    
18.     <testmethod2>
    
19.         <input>3</input>
    
20.         <button>4</button>
    
21.     </testmethod2>
    
22.     <testmethod3>
    
23.         <input>3</input>
    
24.         <button>4</button>
    
25.     </testmethod3>
    
26.     <testmethod4>
    
27.         <input>3</input>
    
28.         <button>4</button>
    
29.     </testmethod4>
    
30. </data>
    
31. 复制代码
    
32. 处用DOM4J解析XML，ParserXml.java文件：
    
33. 
    
34. 复制代码
    
35. package com.test;
    
36. 
    
37. import java.io.File;
    
38. import java.util.ArrayList;
    
39. import java.util.HashMap;
    
40. import java.util.Iterator;
    
41. import java.util.List;
    
42. import java.util.Map;
    
43. 
    
44. import org.dom4j.Document;
    
45. import org.dom4j.DocumentException;
    
46. import org.dom4j.Element;
    
47. import org.dom4j.io.SAXReader;
    
48. 
    
49. public class ParserXml {
    
50. 
    
51.     public List parser3Xml(String fileName) {
    
52.         File inputXml = new File(fileName);   
    
53.         List list=new ArrayList();               
    
54.         int count = 1;
    
55.         SAXReader saxReader = new SAXReader();
    
56.         try {
    
57.             Document document = saxReader.read(inputXml);
    
58.             Element employees = document.getRootElement();
    
59.             for (Iterator i = employees.elementIterator(); i.hasNext();) {
    
60.                 Element employee = (Element) i.next();
    
61.                 Map map = new HashMap();
    
62.                 Map tempMap = new HashMap();
    
63.                 for (Iterator j = employee.elementIterator(); j.hasNext();) {
    
64.                     Element node = (Element) j.next();                    
    
65.                     tempMap.put(node.getName(), node.getText());                    
    
66.                 }
    
67.                 map.put(employee.getName(), tempMap);
    
68.                 list.add(map);
    
69.             }
    
70.         } catch (DocumentException e) {
    
71.             System.out.println(e.getMessage());
    
72.         }
    
73.         return list;
    
74.     }   
    
75.      
    
76. 
    
77. }
    
78. 复制代码
    
79. 然后把解析出来的list转换成Object[][]类型的数据，且结合在@DataProvider中。
    
80. 
    
81. TestData.java文件：
    
82. 
    
83. 复制代码
    
84. package com.test;
    
85. 
    
86. import java.io.File;
    
87. import java.lang.reflect.Method;
    
88. import java.util.ArrayList;
    
89. import java.util.HashMap;
    
90. import java.util.Iterator;
    
91. import java.util.List;
    
92. import java.util.Map;
    
93. 
    
94. import org.testng.annotations.DataProvider;
    
95. 
    
96. public class TestData {        
    
97.    
    
98.     private List l;
    
99.    
    
100.     public TestData() {   
     
101.         this.getXmlData();        
     
102.     }
     
103.    
     
104.     public void getXmlData(){
     
105.         ParserXml p = new ParserXml();
     
106.         l = p.parser3Xml(new File("src/com/test/TestData.xml").getAbsolutePath());
     
107.     }
     
108. 
     
109.     @DataProvider
     
110.     public Object[][] providerMethod(Method method){        
     
111.         List<Map<String, String>> result = new ArrayList<Map<String, String>>();        
     
112.         for (int i = 0; i < l.size(); i++) {
     
113.             Map m = (Map) l.get(i);   
     
114.             if(m.containsKey(method.getName())){                           
     
115.                 Map<String, String> dm = (Map<String, String>) m.get(method.getName());
     
116.                 result.add(dm);   
     
117.             }
     
118.         }  
     
119.         Object[][] files = new Object[result.size()][];
     
120.         for(int i=0; i<result.size(); i++){
     
121.             files[i] = new Object[]{result.get(i)};
     
122.         }        
     
123.         return files;
     
124.     }
     
125.    
     
126. 
     
127. }
     
128. 复制代码
     
129. 再通过测试文件来测试一下：
     
130. 
     
131. TestDataProvider.java文件：
     
132. 
     
133. 复制代码
     
134. package com.test;
     
135. 
     
136. import java.util.Map;
     
137. 
     
138. import org.testng.annotations.*;
     
139. 
     
140. public class TestDataProvider extends TestData {
     
141. 
     
142.     @Test(dataProvider="providerMethod")
     
143.     public void testmethod1(Map<?, ?> param){
     
144.         System.out.println("method1 received:"+param.get("input"));
     
145.     }
     
146.      
     
147.     @Test(dataProvider="providerMethod")
     
148.     public void testmethod2(Map<?, ?> param){
     
149.         System.out.println("method2 received:"+param.get("input"));
     
150.     }
     
151.      
     
152.     @Test(dataProvider="providerMethod")
     
153.     public void testmethod3(Map<?, ?> param){
     
154.         System.out.println("method3 received:"+param.get("input"));
     
155.     }
     
156.    
     
157.     @Test
     
158.     public void testmethod4(){
     
159.         System.out.println("method4 received:4");
     
160.     }
     
161. 
     
162. }
     
163. 复制代码
     
164. 我们再回过头来分析一下XML文件,有两个testmethod1结点，testmethod2，testmethod3，testmethod4结点各一个，在TestDataProvider.java文件中，定义了testmethod1，testmethod2，testmethod3，testmethod4四个测试函数，且testmethod4没有用到dataProvider，所以运行结果最后应该是testmethod1运行两遍，testmethod2，testmethod3，testmethod4各运行一遍，结果如下：
     
165. 
     
166. 复制代码
     
167. method1 received:1
     
168. method1 received:3
     
169. method2 received:3
     
170. method3 received:3
     
171. method4 received:4
     
172. PASSED: testmethod1({input=1, button=2})
     
173. PASSED: testmethod1({input=3, button=4})
     
174. PASSED: testmethod2({input=3, button=4})
     
175. PASSED: testmethod3({input=3, button=4})
     
176. PASSED: testmethod4
     
177. ===============================================
     
178.     Default test
     
179.     Tests run: 5, Failures: 0, Skips: 0
     
180. ===============================================
     
181. 也就是说通过这种方式，只需要把测试函数先写好，然后在XML文件中定义好数据就行了，就能控制函数是否运行，运行次数且运行的数据。
     
182. OK，Let's try.....

复制代码