在做自動(dòng)化測(cè)試時(shí)，數(shù)據(jù)驅(qū)動(dòng)是一個(gè)很重要的概念，當(dāng)數(shù)據(jù)與腳本分離后，面對(duì)茫茫多的數(shù)據(jù)，管理數(shù)據(jù)又成了一個(gè)大問(wèn)題，而數(shù)據(jù)源又可能面對(duì)多個(gè)，就跟在開(kāi)發(fā)過(guò)程中，有時(shí)候要連接MYSQL，有時(shí)候又要連接SQL SERVER一樣，如何做到快速切換？下面的示例中，我們將從一個(gè)數(shù)據(jù)源開(kāi)始，一步步的演示下去：
一. 用外部文件做數(shù)據(jù)驅(qū)動(dòng)的基本寫(xiě)法
1.1 我們?cè)谧鰯?shù)據(jù)驅(qū)動(dòng)時(shí)，把數(shù)據(jù)存儲(chǔ)在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 寫(xiě)一個(gè)TestBase類，里面用來(lái)存放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; } } |
可以看出，我只要有一個(gè)類，能夠提供出一個(gè)數(shù)據(jù)類型為：List<Map<String, String>>的數(shù)據(jù)對(duì)象，就能夠轉(zhuǎn)換成Object[][]的二維數(shù)組，就能夠提供給測(cè)試方法運(yùn)行了。
1.4 在1.3中出現(xiàn)了一個(gè)PropertiesData類，現(xiàn)在來(lái)實(shí)現(xiàn)這個(gè)類
|
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ǔ)類，于是我們?cè)俳Y(jié)合測(cè)試方法，把這三個(gè)基礎(chǔ)類給融合在一起，就形成了一個(gè)外部文件來(lái)做數(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文件的實(shí)現(xiàn)
2.1 如果有多個(gè)數(shù)據(jù)源，我想用txt來(lái)存放數(shù)據(jù)，txt里面存放一個(gè)json串：data.txt
|
1
2
3
4
|
{ "username":"test", "password":"123456"} |
2.2 讀出這個(gè)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 將讀取出來(lái)的JSON串進(jìn)行解析(這里需要用到一個(gè)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測(cè)試類后，發(fā)現(xiàn)結(jié)果與之前用PropertiesData類出現(xiàn)的結(jié)果一模一樣。
三. 用接口來(lái)實(shí)現(xiàn)
3.1 上面的兩種數(shù)據(jù)源，在把數(shù)據(jù)源里的內(nèi)容給加載出來(lái)且加載出來(lái)的數(shù)據(jù)類型為：List<Map<String, String>>后，只需要把TestBase類里的數(shù)據(jù)源加載類給替換一個(gè)即可，那如此一來(lái)，我們可以利用JAVA里面的interface來(lái)重構(gòu)我們的代碼，首先當(dāng)然得要有一個(gè)interface
|
1
2
3
|
public interface DataInterface { public List<Map<String, String>> getTestMethodData();} |
3.2 我們的PropertiesData類與TxtData類當(dāng)然要實(shí)現(xiàn)這個(gè)接口了
|
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ù)加載的類對(duì)象要發(fā)生改變，我們?cè)赥estBase里新加一個(gè)方法(這是產(chǎn)生類對(duì)象的一種新的方式)
|
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í)候只需要改變數(shù)據(jù)加載類的路徑即可了。
四. 將數(shù)據(jù)加載類的路徑可配置化
4.1 這時(shí)候，我們就可以想著把數(shù)據(jù)加載類的路徑寫(xiě)在配置文件中了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é)果仍然是一樣的。到此為止，我們已實(shí)現(xiàn)了去更改配置文件里面的內(nèi)容，來(lái)選擇加載數(shù)據(jù)源。
五. 多數(shù)據(jù)源的切換
5.1 如果一個(gè)測(cè)試類里有兩個(gè)測(cè)試方法，那么在配置文件里配置好數(shù)據(jù)源后，就表示這兩個(gè)測(cè)試方法都將會(huì)加載同樣的數(shù)據(jù)源，但如果我們希望一個(gè)測(cè)試方法用屬性文件的數(shù)據(jù)源，另一個(gè)方法用TXT的數(shù)據(jù)源，這個(gè)如何辦？也就是需要實(shí)現(xiàn)在全局配置化后，局部可再次選擇數(shù)據(jù)源。我將會(huì)利用到JAVA里的注解，來(lái)實(shí)現(xiàn)。所以我們先定義一個(gè)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()方法再次的更改，要利用上這個(gè)注解解析出來(lái)的值
|
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;} |
這段代碼可以看到，如果測(cè)試方法標(biāo)注DataSource,則會(huì)以標(biāo)注的注解值為準(zhǔn)，否則則會(huì)以全局配置的值為準(zhǔn)。
5.5 在TestDemo里多加一個(gè)測(cè)試方法，以示區(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")); } } |
上面的測(cè)試類中，兩個(gè)測(cè)試方法，一個(gè)用了全局的配置數(shù)據(jù)源值，一個(gè)用了注解數(shù)據(jù)源值。大家可以運(yùn)行的看看結(jié)果。
六. 工程結(jié)構(gòu)圖：
以上這篇Java自動(dòng)化測(cè)試中多數(shù)據(jù)源的切換(實(shí)例講解)就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持服務(wù)器之家。
原文鏈接：http://www.cnblogs.com/zhangfei/p/5229434.html
