基于testNG的通用数据驱动-DataProvider

    上一篇有提过可以通过DataProvider做数据驱动，通过@Test(dataProvider = "abc"),@DataProvider(name = "abc")来获取指定的测试数据，在缺点是测试数据需要保存在代码中，如果想从外部获取参数，需要一个通用的数据驱动。

     DataProvider 提供了一个方法，DataProvider（Method m），通过入参传入的方法，可以知晓该方法要调用哪个数据，从外部获取到数据后，转为二维数组返回即可。直接上代码～～

    首先我们创建一个通用数据驱动类ExtentTestNGDataProvider，当脚本调用@DataProvider(name = "createData")时，将该方法名传入，获取方法名称，在数据库查到该方法对应的数据，最终返回Object [ ][ ] 格式的结果。

public class ExtentTestNGDataProvider { @DataProvider(name = "createData") public static Object[][] createData(Method m) throws Exception { String casename = m.getName(); String sql = "select data_json from tb_data where status = 1 and casename = '"+casename+"'"; List<Map<String, Object>> list = MysqlUtils.exeSqlApi("auto_test", sql); Parameter[] parameter= m.getParameters(); List<Map<String,Object>> listnew = new ArrayList<>(); for(Map<String,Object> map :list){ String data_json = map.get("data_json").toString(); LinkedHashMap<String, Object> mapdata = JSON.parseObject(data_json, LinkedHashMap.class); listnew.add(mapdata); } Object [ ][ ] result=new Object[listnew.size()][parameter.length]; for(int i=0;i<listnew.size();i++) { result[i]=listnew.get(i).values().toArray(); } return result; } }

  数据库：通过status控制测试数据是否生效，通过data_json返回测试数据。data_json内容和代码中原来的测试数据一致。

   

 

 最后测试用例类

@Test(dataProvider = "createData", priority = 2,threadPoolSize = 1, invocationCount = 1)// dataProviderClass = ExtentTestNGDataProvider.class,,threadPoolSize = 1, invocationCount = 1 public void payment_order(String name, String dealer_id, String replacejson, String assertjson) throws Exception { }