JAVA自動化測試中多數據源的切換

時間  2019-11-06
標籤 java 自動化 測試 數據 切換 欄目 Java 简体版
原文   原文鏈接

在作自動化測試時，數據驅動是一個很重要的概念，當數據與腳本分離後，面對茫茫多的數據，管理數據又成了一個大問題，而數據源又可能面對多個，就跟在開發過程當中，有時候要鏈接MYSQL，有時候又要鏈接SQL SERVER同樣，如何作到快速切換？下面的示例中，咱們將從一個數據源開始，一步步的演示下去：java

一. 用外部文件作數據驅動的基本寫法json

1.1　咱們在作數據驅動時，把數據存儲在JAVA的屬性文件中：data.properties數組

username=test
password=123456

1.2　解析properties文件app

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的DataProvideride

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類，如今來實現這個類this

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　以上的運行結果輸出爲：blog

二.　屬性文件換成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"));
	}
	
}

上面的測試類中，兩個測試方法，一個用了全局的配置數據源值，一個用了註解數據源值。你們能夠運行的看看結果。

六.　工程結構圖：

至於源碼，你們自行的拷貝粘貼吧，也看成是一種知識的鞏固。

相關文章
相關標籤/搜索
每日一句
    每一个你不满意的现在，都有一个你没有努力的曾经。
本站公眾號
   歡迎關注本站公眾號,獲取更多信息
聯繫我們 最近搜索 最新文章 沪ICP备13005482号-10 MyBatis教程 SQL 教程 MySQL教程 Java 教程 Thymeleaf 教程 Hibernate教程 Spring教程 Redis教程