JAVA自动化测试中多数据源的切换

在做自动化测试时，数据驱动是一个很重要的概念，当数据与脚本分离后，面对茫茫多的数据，管理数据又成了一个大问题，而数据源又可能面对多个，就跟在开发过程中，有时候要连接MYSQL，有时候又要连接SQL SERVER一样，如何做到快速切换？下面的示例中，我们将从一个数据源开始，一步步的演示下去：

一. 用外部文件做数据驱动的基本写法

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 Map<String, String> getPropertyData(String filePath){
        try{
        	return new HashMap<String, String>((Map)PropertiesHandler.loadPropertiesFile(filePath));
        }catch(Exception e){
            e.printStackTrace();
        }
        return new HashMap<String, String>();
    }

	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<Map<String, String>> 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;
	}

}

可以看出，我只要有一个类，能够提供出一个数据类型为：List<Map<String, String>>的数据对象，就能够转换成Object[][]的二维数组，就能够提供给测试方法运行了。

1.4　在1.3中出现了一个PropertiesData类，现在来实现这个类

public class PropertiesData {

	public List<Map<String, String>> getTestMethodData(){
		List<Map<String, String>> list = new ArrayList<Map<String, String>>();
		list.add(PropertiesHandler.getPropertyData("file/data.properties"));
		return list;
	}

}

1.5　以上中有数据解析类，有数据加载类，有数据提供的基础类，于是我们再结合测试方法，把这三个基础类给融合在一起，就形成了一个外部文件来做数据源的完整示例了：

public class TestDemo extends TestBase{

	@Test(dataProvider="dataProvider")
	public void testDemo(Map<String, String> param){
		System.out.println(param.get("username"));
		System.out.println(param.get("password"));
	}

}

1.6　以上的运行结果输出为：

二.　属性文件换成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<Map<String, String>> getTestMethodData(){
		List<Map<String, String>> list = new ArrayList<Map<String, String>>();
		String data = FileUtils.readFile("file/data.txt");
		Gson gson = new Gson();
		Map<String, String> dataMap = gson.fromJson(data, new TypeToken<Map<String, String>>(){}.getType());
		list.add(dataMap);
		return list;
	}

}

2.4　将TxtData类给用上，即将TestBase类里的用到PropertiesData类的地方换成TxtData类即可

	private Object[][] getTestData(){
		TxtData testData = new TxtData();
		List<Map<String, String>> 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<Map<String, String>>后，只需要把TestBase类里的数据源加载类给替换一个即可，那如此一来，我们可以利用JAVA里面的interface来重构我们的代码，首先当然得要有一个interface

public interface DataInterface {
	public List<Map<String, String>> getTestMethodData();
}

3.2　我们的PropertiesData类与TxtData类当然要实现这个接口了

public class PropertiesData implements DataInterface{

	public List<Map<String, String>> getTestMethodData(){
		List<Map<String, String>> list = new ArrayList<Map<String, String>>();
		list.add(PropertiesHandler.getPropertyData("file/data.properties"));
		return list;
	}

}

public class TxtData implements DataInterface{

	public List<Map<String, String>> getTestMethodData(){
		List<Map<String, String>> list = new ArrayList<Map<String, String>>();
		String data = FileUtils.readFile("file/data.txt");
		Gson gson = new Gson();
		Map<String, String> dataMap = gson.fromJson(data, new TypeToken<Map<String, String>>(){}.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<Map<String, String>> 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<Map<String, String>> 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{
		Map<String, String> map = 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<Map<String, String>> 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(Map<String, String> param){
		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<Map<String, String>> 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(Map<String, String> param){
		System.out.println(param.get("username"));
		System.out.println(param.get("password"));
	}

	@Test(dataProvider="dataProvider")
	public void testDemo1(Map<String, String> param){
		System.out.println(param.get("username"));
		System.out.println(param.get("password"));
	}

}

上面的测试类中，两个测试方法，一个用了全局的配置数据源值，一个用了注解数据源值。大家可以运行的看看结果。

六.　工程结构图：

至于源码，大家自行的拷贝粘贴吧，也当作是一种知识的巩固。

时间： 2024-10-10 10:36:32

JAVA自动化测试中多数据源的切换的相关文章

spring框架中多数据源创建加载并且实现动态切换的配置实例代码

原文:spring框架中多数据源创建加载并且实现动态切换的配置实例代码 源代码下载地址:http://www.zuidaima.com/share/1774074130205696.htm 在我们的项目中遇到这样一个问题:我们的项目需要连接多个数据库,而且不同的客户在每次访问中根据需要会去访问不同的数据库.我们以往在spring和hibernate框架中总是配置一个数据源,因而sessionFactory的dataSource属性总是指向这个数据源并且恒定不变,所有DAO在使用sessionFa

自动化测试中自动化切换网络----解决方案

自动化测试中自动化切换网络----解决方案 思路 通过安装一个控制网络的APP,在测试脚本中,通过shell命令来控制实现 使用方式 关闭WIFI,切换到4G网络(测试手机默认流量是开着的) adb  shell am force-stop com.steinwurf.adbjoinwifi adb  shell am start -n com.steinwurf.adbjoinwifi/com.steinwurf.adbjoinwifi.MainActivity -e ssid 4G 切换到W

基于Struts2 Spring ibatis Oracle10g架构 多数据源动态切换实例

一.概述 基于Spring动态配置多数据源,在大型的应用中对数据进行切分,并且采用多个数据库实例进行管理,这样可以有效的提高系统的水平伸缩性,而这样的解决方案就会补同于常见的单一数据实例的方案,这就要程序在运行时根据当时的请求以及系统状态来动态的决定将数据存储在哪个数据库实例中,以及从哪个数据库提取数据. Spring配置多个数据源的方式和具体使用过程,Spring对于多数据源,以数据库表为参照,大体上可以分为两大类情况: 1.表级上的跨数据库,即对于不同的数据库却有不相同的表(表名和表结构完全

Spring3.3 整合 Hibernate3、MyBatis3.2 配置多数据源/动态切换数据源 方法(转)

一.开篇 这里整合分别采用了Hibernate和MyBatis两大持久层框架,Hibernate主要完成增删改功能和一些单一的对象查询功能,MyBatis主要负责查询功能.所以在出来数据库方言的时候基本上没有什么问题,但唯一可能出现问题的就是在hibernate做添加操作生成主键策略的时候.因为我们都知道hibernate的数据库本地方言会针对不同的数据库采用不同的主键生成策略. 所以针对这一问题不得不采用自定义的主键生成策略,自己写一个主键生成器的表来维护主键生成方式或以及使用其他的方式来生成

基于spring的aop实现多数据源动态切换

https://lanjingling.github.io/2016/02/15/spring-aop-dynamicdatasource/ 基于spring的aop实现多数据源动态切换 发表于 2016-02-15   |   分类于 spring  | 一.多数据源动态切换原理 项目中我们经常会遇到多数据源的问题,尤其是数据同步或定时任务等项目更是如此:又例如:读写分离数据库配置的系统. 1.多数据源设置: 1)静态数据源切换:一般情况下,我们可以配置多个数据源,然后为每个数据源写一套对应的

spring+springMVC+Mybatis架构下采用AbstractRoutingDataSource、atomikos、JTA实现多数据源灵活切换以及分布式事务管理

背景: 1.系统采用SSM架构.需要在10多个MYSQL数据库之间进行切换并对数据进行操作,上篇博文<springMVC+Mybatis(使用AbstractRoutingDataSource实现多数据源切换时)事务管理未生效的解决办法> 2.第一步先通过AbstractRoutingDataSource实现了多数据源的灵活切换,但是后来发现事务不起作用: 3.发现问题是因为重复扫描service包导致第二次扫入容器的BEAN没有纳入事务管理,因此在springMVC的配置文件中排除了对Ser

Spring Boot 多数据源 自动切换

实现案例场景: 某系统除了需要从自己的主要数据库上读取和管理数据外,还有一部分业务涉及到其他多个数据库,要求可以在任何方法上可以灵活指定具体要操作的数据库.为了在开发中以最简单的方法使用,本文基于注解和AOP的方法实现,在spring boot框架的项目中,添加本文实现的代码类后,只需要配置好数据源就可以直接通过注解使用,简单方便.一配置二使用 1. 启动类注册动态数据源 2. 配置文件中配置多个数据源 3. 在需要的方法上使用注解指定数据源1.在启动类添加 @Import({DynamicDa

spring+mybatis多数据源动态切换

spring mvc+mybatis+多数据源切换 选取oracle,mysql作为例子切换数据源.oracle为默认数据源,在测试的action中,进行mysql和oracle的动态切换. web.xml <context-param> <param-name>webAppRootKey</param-name> <param-value>trac</param-value> </context-param> <!-- Spr

spring中的数据源

http://yonguo.iteye.com/blog/115221 1)Spring在第三方依赖包中包含了两个数据源的实现类包,DBCP数据源 , C3P0数据源,可以在spring中直接配置使用: 2)获取JNDI数据源 :如果应用配置在高性能的应用服务器(如WebLogic或Websphere等)上,我们可能更希望使用应用服务器本身提供的数据源; 应用服务器的数据源,使用JNDI开放调用者使用 3) spring使用多数据源,其实就是使用动态数据源,进行切换 http://blog.cs