您好,登錄后才能下訂單哦!
這篇文章主要介紹了Java自動化測試中多數(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ù)提供的基礎(chǔ)類,于是我們再結(jié)合測試方法,把這三個基礎(chǔ)類給融合在一起,就形成了一個外部文件來做數(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串進(jìn)行解析(這里需要用到一個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)結(jié)果與之前用PropertiesData類出現(xiàn)的結(jié)果一模一樣。
三. 用接口來實現(xiàn)
3.1 上面的兩種數(shù)據(jù)源,在把數(shù)據(jù)源里的內(nèi)容給加載出來且加載出來的數(shù)據(jù)類型為:List<Map<String, String>>后,只需要把TestBase類里的數(shù)據(jù)源加載類給替換一個即可,那如此一來,我們可以利用JAVA里面的interface來重構(gòu)我們的代碼,首先當(dāng)然得要有一個interface
public interface DataInterface { public List<Map<String, String>> getTestMethodData(); }
3.2 我們的PropertiesData類與TxtData類當(dāng)然要實現(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)結(jié)果仍然是一樣的。所以,這時候只需要改變數(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()方法再改進(jìn)一下:
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類,結(jié)果仍然是一樣的。到此為止,我們已實現(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; }
這段代碼可以看到,如果測試方法標(biāo)注DataSource,則會以標(biāo)注的注解值為準(zhǔn),否則則會以全局配置的值為準(zhǔn)。
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ù)源值。大家可以運行的看看結(jié)果。
六. 工程結(jié)構(gòu)圖:
感謝你能夠認(rèn)真閱讀完這篇文章,希望小編分享的“Java自動化測試中多數(shù)據(jù)源如何切換”這篇文章對大家有幫助,同時也希望大家多多支持億速云,關(guān)注億速云行業(yè)資訊頻道,更多相關(guān)知識等著你來學(xué)習(xí)!
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進(jìn)行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。