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

    1

    2

    3

    4

    3

    4

    3

    4

    3

    4

packagecom.test;

importjava.io.File;

importjava.util.ArrayList;

importjava.util.HashMap;

importjava.util.Iterator;

importjava.util.List;

importjava.util.Map;

importorg.dom4j.Document;

importorg.dom4j.DocumentException;

importorg.dom4j.Element;

importorg.dom4j.io.SAXReader;

publicclassParserXml {

  publicList parser3Xml(String fileName) {

    File inputXml =newFile(fileName); 

    List list=newArrayList();       

    intcount =1;

    SAXReader saxReader =newSAXReader();

    try{

      Document document = saxReader.read(inputXml);

      Element employees = document.getRootElement();

      for(Iterator i = employees.elementIterator(); i.hasNext();) {

        Element employee = (Element) i.next();

        Map map =newHashMap();

        Map tempMap =newHashMap();

        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());

    }

    returnlist;

  } 

}

packagecom.test;

importjava.io.File;

importjava.lang.reflect.Method;

importjava.util.ArrayList;

importjava.util.HashMap;

importjava.util.Iterator;

importjava.util.List;

importjava.util.Map;

importorg.testng.annotations.DataProvider;

publicclassTestData {   

  privateList l;

  publicTestData() { 

    this.getXmlData();   

  }

  publicvoidgetXmlData(){

    ParserXml p =newParserXml();

    l = p.parser3Xml(newFile("src/com/test/TestData.xml").getAbsolutePath());

  }

  @DataProvider

  publicObject[][] providerMethod(Method method){   

    List> result =newArrayList>();   

    for(inti =0; i < l.size(); i++) {

      Map m = (Map) l.get(i); 

      if(m.containsKey(method.getName())){             

        Mapdm = (Map) m.get(method.getName());

        result.add(dm); 

      }

    }

    Object[][] files =newObject[result.size()][];

    for(inti=0; i

      files[i] =newObject[]{result.get(i)};

    }   

    returnfiles;

  }

}

packagecom.test;

importjava.util.Map;

importorg.testng.annotations.*;

publicclassTestDataProviderextendsTestData {

  @Test(dataProvider="providerMethod")

  publicvoidtestmethod1(Map param){

    System.out.println("method1 received:"+param.get("input"));

  }

  @Test(dataProvider="providerMethod")

  publicvoidtestmethod2(Map param){

    System.out.println("method2 received:"+param.get("input"));

  }

  @Test(dataProvider="providerMethod")

  publicvoidtestmethod3(Map param){

    System.out.println("method3 received:"+param.get("input"));

  }

  @Test

  publicvoidtestmethod4(){

    System.out.println("method4 received:4");

  }

}

method1 received:1

method1 received:3

method2 received:3

method3 received:3

method4 received:4

PASSED: 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

===============================================