Java自动化测试中多数据源如何切换
这篇文章主要介绍了Java自动化测试中多数据源如何切换,具有一定借鉴价值,感兴趣的朋友可以参考下,希望大家阅读完这篇文章之后大有收获,下面让小编带着大家一起了解一下。
创新互联建站主要从事成都网站设计、网站建设、网页设计、企业做网站、公司建网站等业务。立足成都服务乐陵,十多年网站建设经验,价格优惠、服务专业,欢迎来电咨询建站服务:18980820575
一. 用外部文件做数据驱动的基本写法
1.1 我们在做数据驱动时,把数据存储在JAVA的属性文件中:data.properties
username=test password=123456
1.2 解析properties文件
public class PropertiesHandler { private static Properties loadPropertiesFile(String filePath){ Properties p = new Properties(); InputStream in = null; try { in = new FileInputStream(new File(filePath)); p.load(in); } catch (FileNotFoundException e) { e.printStackTrace(); } catch (IOException e) { e.printStackTrace(); }finally{ try { if(in != null){ in.close(); } } catch (IOException e) { e.printStackTrace(); } } return p; } /** * 将property转换成Map * @param key * @return */ @SuppressWarnings({ "rawtypes", "unchecked" }) public static MapgetPropertyData(String filePath){ try{ return new HashMap ((Map)PropertiesHandler.loadPropertiesFile(filePath)); }catch(Exception e){ e.printStackTrace(); } return new HashMap (); } public static void main(String[] args) { System.out.println(PropertiesHandler.getPropertyData("file/data.properties")); } }
1.3 写一个TestBase类,里面用来存放TestNg的DataProvider
public class TestBase { @DataProvider public Object[][] dataProvider(){ return this.getTestData(); } private Object[][] getTestData(){ PropertiesData testData = new PropertiesData(); List
可以看出,我只要有一个类,能够提供出一个数据类型为:List
1.4 在1.3中出现了一个PropertiesData类,现在来实现这个类
public class PropertiesData { public List
1.5 以上中有数据解析类,有数据加载类,有数据提供的基础类,于是我们再结合测试方法,把这三个基础类给融合在一起,就形成了一个外部文件来做数据源的完整示例了:
public class TestDemo extends TestBase{ @Test(dataProvider="dataProvider") public void testDemo(Mapparam){ System.out.println(param.get("username")); System.out.println(param.get("password")); } }
二. 属性文件换成txt文件的实现
2.1 如果有多个数据源,我想用txt来存放数据,txt里面存放一个json串:data.txt
{ "username":"test", "password":"123456" }
2.2 读出这个txt文件
public class FileUtils { public static String readFile(String fileName) { InputStream is = null; StringBuffer sb = new StringBuffer(); try { is = new FileInputStream(fileName); byte[] byteBuffer = new byte[is.available()]; int read = 0; while((read = is.read(byteBuffer)) != -1){ sb.append(new String(byteBuffer, 0, read)); } } catch (FileNotFoundException e) { e.printStackTrace(); } catch (IOException e) { e.printStackTrace(); }finally{ try { if(is!=null){ is.close(); } } catch (IOException e) { e.printStackTrace(); } } return sb.toString(); } public static void main(String[] args) { System.out.println(FileUtils.readFile("file/data.txt")); } }
2.3 将读取出来的JSON串进行解析(这里需要用到一个JAR包,gson.jar)
public class TxtData { public List> getTestMethodData(){ List > list = new ArrayList >(); String data = FileUtils.readFile("file/data.txt"); Gson gson = new Gson(); Map dataMap = gson.fromJson(data, new TypeToken >(){}.getType()); list.add(dataMap); return list; } }
2.4 将TxtData类给用上,即将TestBase类里的用到PropertiesData类的地方换成TxtData类即可
private Object[][] getTestData(){ TxtData testData = new TxtData(); List> listData = testData.getTestMethodData(); Object[][] object = new Object[listData.size()][]; for (int i = 0; i < listData.size(); i++) { object[i] = new Object[]{listData.get(i)}; } return object; }
2.5 运行TestDemo测试类后,发现结果与之前用PropertiesData类出现的结果一模一样。
三. 用接口来实现
3.1 上面的两种数据源,在把数据源里的内容给加载出来且加载出来的数据类型为:List
public interface DataInterface { public List> getTestMethodData(); }
3.2 我们的PropertiesData类与TxtData类当然要实现这个接口了
public class PropertiesData implements DataInterface{ public List> getTestMethodData(){ List > list = new ArrayList >(); list.add(PropertiesHandler.getPropertyData("file/data.properties")); return list; } }
public class TxtData implements DataInterface{ public List> getTestMethodData(){ List > list = new ArrayList >(); String data = FileUtils.readFile("file/data.txt"); Gson gson = new Gson(); Map dataMap = gson.fromJson(data, new TypeToken >(){}.getType()); list.add(dataMap); return list; } }
3.3 然后在TestBase里就要有所改变了,即产生数据加载的类对象要发生改变,我们在TestBase里新加一个方法(这是产生类对象的一种新的方式)
private DataInterface getDataInstance(String key){ DataInterface data = null; try { data = (DataInterface) Class.forName(key).newInstance(); } catch (InstantiationException | IllegalAccessException | ClassNotFoundException e) { e.printStackTrace(); } return data; }
3.4 TestBase类里的getTestData()方法就要重新的改变一下了
private Object[][] getTestData(){ DataInterface testData = this.getDataInstance("com.test.testdata.PropertiesData"); List> listData = testData.getTestMethodData(); Object[][] object = new Object[listData.size()][]; for (int i = 0; i < listData.size(); i++) { object[i] = new Object[]{listData.get(i)}; } return object; }
private Object[][] getTestData(){ DataInterface testData = this.getDataInstance("com.test.testdata.TxtData"); List> listData = testData.getTestMethodData(); Object[][] object = new Object[listData.size()][]; for (int i = 0; i < listData.size(); i++) { object[i] = new Object[]{listData.get(i)}; } return object; }
3.5 再次运行TestDemo,即可发现结果仍然是一样的。所以,这时候只需要改变数据加载类的路径即可了。
四. 将数据加载类的路径可配置化
4.1 这时候,我们就可以想着把数据加载类的路径写在配置文件中了config.properties
DataSource=com.test.testdata.TxtData
4.2 加载config文件
public class Config { public static String DATA_SOURCE; static{ Mapmap = PropertiesHandler.getPropertyData("config/config.properties"); DATA_SOURCE = map.get("DataSource"); } }
4.3 将TestBase里的getTestData()方法再改进一下:
private Object[][] getTestData(){ DataInterface testData = this.getDataInstance(Config.DATA_SOURCE); List> listData = testData.getTestMethodData(); Object[][] object = new Object[listData.size()][]; for (int i = 0; i < listData.size(); i++) { object[i] = new Object[]{listData.get(i)}; } return object; }
4.4 再次运行TestDemo类,结果仍然是一样的。到此为止,我们已实现了去更改配置文件里面的内容,来选择加载数据源。
五. 多数据源的切换
5.1 如果一个测试类里有两个测试方法,那么在配置文件里配置好数据源后,就表示这两个测试方法都将会加载同样的数据源,但如果我们希望一个测试方法用属性文件的数据源,另一个方法用TXT的数据源,这个如何办?也就是需要实现在全局配置化后,局部可再次选择数据源。我将会利用到JAVA里的注解,来实现。所以我们先定义一个DataSource的注解
@Target(ElementType.FIELD) @Retention(RetentionPolicy.RUNTIME) public @interface DataSource { String value(); }
5.2 解析该注解
public class DataSources { public static String getDataSource(Method method){ DataSource ds = method.getAnnotation(DataSource.class); if(ds != null){ return ds.value(); } return null; } }
5.3 该注解的使用
@DataSource("com.test.testdata.PropertiesData") @Test(dataProvider="dataProvider") public void testDemo(Mapparam){ System.out.println(param.get("username")); System.out.println(param.get("password")); }
5.4 TestBase类里的getTestData()方法再次的更改,要利用上这个注解解析出来的值
private Object[][] getTestData(Method method){ String sourceKey = DataSources.getDataSource(method); if(sourceKey==null){ sourceKey = Config.DATA_SOURCE; } DataInterface testData = this.getDataInstance(sourceKey); List> listData = testData.getTestMethodData(); Object[][] object = new Object[listData.size()][]; for (int i = 0; i < listData.size(); i++) { object[i] = new Object[]{listData.get(i)}; } return object; }
这段代码可以看到,如果测试方法标注DataSource,则会以标注的注解值为准,否则则会以全局配置的值为准。
5.5 在TestDemo里多加一个测试方法,以示区别
public class TestDemo extends TestBase{ @DataSource("com.test.testdata.PropertiesData") @Test(dataProvider="dataProvider") public void testDemo(Mapparam){ System.out.println(param.get("username")); System.out.println(param.get("password")); } @Test(dataProvider="dataProvider") public void testDemo1(Map param){ System.out.println(param.get("username")); System.out.println(param.get("password")); } }
上面的测试类中,两个测试方法,一个用了全局的配置数据源值,一个用了注解数据源值。大家可以运行的看看结果。
六. 工程结构图:
感谢你能够认真阅读完这篇文章,希望小编分享的“Java自动化测试中多数据源如何切换”这篇文章对大家有帮助,同时也希望大家多多支持创新互联,关注创新互联行业资讯频道,更多相关知识等着你来学习!
标题名称:Java自动化测试中多数据源如何切换
标题来源:http://scyanting.com/article/jdjeeh.html