在做自動化測試時，數(shù)據(jù)驅(qū)動是一個很重要的概念，當(dāng)數(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
|
1
2
|
username=testpassword=123456 |
1.2 解析properties文件
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
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
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
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ù)組，就能夠提供給測試方法運(yùn)行了。
1.4 在1.3中出現(xiàn)了一個PropertiesData類，現(xiàn)在來實現(xiàn)這個類
|
1
2
3
4
5
6
7
8
9
|
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ù)源的完整示例了：
|
1
2
3
4
5
6
7
8
9
|
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
|
1
2
3
4
|
{ "username":"test", "password":"123456"} |
2.2 讀出這個txt文件
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
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)
|
1
2
3
4
5
6
7
8
9
10
11
12
|
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類即可
|
1
2
3
4
5
6
7
8
9
|
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 運(yùn)行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
|
1
2
3
|
public interface DataInterface { public List<Map<String, String>> getTestMethodData();} |
3.2 我們的PropertiesData類與TxtData類當(dāng)然要實現(xiàn)這個接口了
|
1
2
3
4
5
6
7
8
9
|
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; } } |
|
1
2
3
4
5
6
7
8
9
10
11
12
|
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)生類對象的一種新的方式)
|
1
2
3
4
5
6
7
8
9
|
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()方法就要重新的改變一下了
|
1
2
3
4
5
6
7
8
9
|
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;} |
|
1
2
3
4
5
6
7
8
9
|
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 再次運(yùn)行TestDemo，即可發(fā)現(xiàn)結(jié)果仍然是一樣的。所以，這時候只需要改變數(shù)據(jù)加載類的路徑即可了。
四. 將數(shù)據(jù)加載類的路徑可配置化
4.1 這時候，我們就可以想著把數(shù)據(jù)加載類的路徑寫在配置文件中了config.properties
|
1
|
DataSource=com.test.testdata.TxtData |
4.2 加載config文件
|
1
2
3
4
5
6
7
8
9
10
|
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)一下：
|
1
2
3
4
5
6
7
8
9
|
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 再次運(yùn)行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的注解
|
1
2
3
4
5
|
@Target(ElementType.FIELD)@Retention(RetentionPolicy.RUNTIME)public @interface DataSource { String value();} |
5.2 解析該注解
|
1
2
3
4
5
6
7
8
9
10
11
|
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 該注解的使用
|
1
2
3
4
5
6
|
@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()方法再次的更改，要利用上這個注解解析出來的值
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
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ū)別
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
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ù)源值。大家可以運(yùn)行的看看結(jié)果。
六. 工程結(jié)構(gòu)圖：
以上這篇Java自動化測試中多數(shù)據(jù)源的切換(實例講解)就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持服務(wù)器之家。
原文鏈接：http://www.cnblogs.com/zhangfei/p/5229434.html
