Java自動化測試中多數(shù)據(jù)源的切換(實例講解)
在做自動化測試時，數(shù)據(jù)驅(qū)動是一個很重要的概念，當數(shù)據(jù)與腳本分離后，面對茫茫多的數(shù)據(jù)，管理數(shù)據(jù)又成了一個大問題，而數(shù)據(jù)源又可能面對多個，就跟在開發(fā)過程中，有時候要連接MYSQL，有時候又要連接SQL SERVER一樣，如何做到快速切換？下面的示例中，我們將從一個數(shù)據(jù)源開始，一步步的演示下去：
一. 用外部文件做數(shù)據(jù)驅(qū)動的基本寫法
1.1 我們在做數(shù)據(jù)驅(qū)動時，把數(shù)據(jù)存儲在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轉(zhuǎn)換成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;
}
}
可以看出，我只要有一個類，能夠提供出一個數(shù)據(jù)類型為：List<Map<String, String>>的數(shù)據(jù)對象，就能夠轉(zhuǎn)換成Object[][]的二維數(shù)組，就能夠提供給測試方法運行了。
1.4 在1.3中出現(xiàn)了一個PropertiesData類，現(xiàn)在來實現(xiàn)這個類
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 以上中有數(shù)據(jù)解析類，有數(shù)據(jù)加載類，有數(shù)據(jù)提供的基礎類，于是我們再結合測試方法，把這三個基礎類給融合在一起，就形成了一個外部文件來做數(shù)據(jù)源的完整示例了：
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"));
}
}
二. 屬性文件換成txt文件的實現(xiàn)
2.1 如果有多個數(shù)據(jù)源，我想用txt來存放數(shù)據(jù)，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測試類后，發(fā)現(xiàn)結果與之前用PropertiesData類出現(xiàn)的結果一模一樣。
三. 用接口來實現(xiàn)
3.1 上面的兩種數(shù)據(jù)源，在把數(shù)據(jù)源里的內(nèi)容給加載出來且加載出來的數(shù)據(jù)類型為：List<Map<String, String>>后，只需要把TestBase類里的數(shù)據(jù)源加載類給替換一個即可，那如此一來，我們可以利用JAVA里面的interface來重構我們的代碼，首先當然得要有一個interface
public interface DataInterface {
public List<Map<String, String>> getTestMethodData();
}
3.2 我們的PropertiesData類與TxtData類當然要實現(xiàn)這個接口了
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里就要有所改變了，即產(chǎn)生數(shù)據(jù)加載的類對象要發(fā)生改變，我們在TestBase里新加一個方法(這是產(chǎn)生類對象的一種新的方式)
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，即可發(fā)現(xiàn)結果仍然是一樣的。所以，這時候只需要改變數(shù)據(jù)加載類的路徑即可了。
四. 將數(shù)據(jù)加載類的路徑可配置化
4.1 這時候，我們就可以想著把數(shù)據(jù)加載類的路徑寫在配置文件中了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類，結果仍然是一樣的。到此為止，我們已實現(xiàn)了去更改配置文件里面的內(nèi)容，來選擇加載數(shù)據(jù)源。
五. 多數(shù)據(jù)源的切換
5.1 如果一個測試類里有兩個測試方法，那么在配置文件里配置好數(shù)據(jù)源后，就表示這兩個測試方法都將會加載同樣的數(shù)據(jù)源，但如果我們希望一個測試方法用屬性文件的數(shù)據(jù)源，另一個方法用TXT的數(shù)據(jù)源，這個如何辦？也就是需要實現(xiàn)在全局配置化后，局部可再次選擇數(shù)據(jù)源。我將會利用到JAVA里的注解，來實現(xiàn)。所以我們先定義一個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里多加一個測試方法，以示區(qū)別
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"));
}
}
上面的測試類中，兩個測試方法，一個用了全局的配置數(shù)據(jù)源值，一個用了注解數(shù)據(jù)源值。大家可以運行的看看結果。
六. 工程結構圖：
以上這篇Java自動化測試中多數(shù)據(jù)源的切換(實例講解)就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
本篇文章給大家分享了JAVA中實現(xiàn)終止線程池中正在運行的定時任務的具體步驟和方法，有需要的朋友跟著學習下。2018-05-05
如何更規(guī)范化編寫Java 代碼的重要性想必毋需多言，其中最重要的幾點當屬提高代碼性能、使代碼遠離Bug、令代碼更優(yōu)雅，2021-07-07
下面小編就為大家分享一篇Java獲取json數(shù)組對象的實例講解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-03-03
Filter?過濾器和Interceptor?攔截器是SpringBoot?的?Web?項目開發(fā)中長用到的，本文主要來和大家討論一下?Filter?與?Interceptor?的做法及它們之間的區(qū)別，需要的可以參考下2023-07-07
java中ExecutorService創(chuàng)建方法總結
在本篇文章里小編給大家整理了一篇關于java中ExecutorService創(chuàng)建方法總結，有興趣的朋友們可以參考下。2021-01-01
Spring Boot 啟動、停止、重啟、狀態(tài)腳本
今天給大家分享Spring Boot 項目腳本（啟動、停止、重啟、狀態(tài)），通過示例代碼給大家介紹的非常詳細，需要的朋友參考下吧2021-06-06
為什么Spring和IDEA都不推薦使用 @Autowired 注解
本文主要介紹了為什么Spring和IDEA都不推薦使用 @Autowired 注解，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2022-04-04
這篇文章主要介紹了支持SpEL表達式的自定義日志注解@SysLog，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-02-02
從問題出發(fā),逐步講透在SpringMVC中使用RequestContextHolder對象在多線程情況下無法獲取請求的真實原因,本文將和大家一起深入剖析SpringMVC多線程下無法獲取請求的原因和解決,需要的朋友可以參考下2024-04-04
SpringBoot集成ip2region實現(xiàn)ip白名單的代碼示例
ip2region v2.0 - 是一個離線IP地址定位庫和IP定位數(shù)據(jù)管理框架,10微秒級別的查詢效率,提供了眾多主流編程語言的 xdb 數(shù)據(jù)生成和查詢客戶端實現(xiàn),本文介紹了SpringBoot集成ip2region實現(xiàn)ip白名單的代碼工程,需要的朋友可以參考下2024-08-08