Java自動化測試中多數(shù)據(jù)源的切換(實例講解)

更新時間：2017年10月07日 10:57:46 作者：張飛_

下面小編就為大家?guī)硪黄狫ava自動化測試中多數(shù)據(jù)源的切換(實例講解)。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

在做自動化測試時，數(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ù)提供的基礎(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串進行解析(這里需要用到一個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)我們的代碼，首先當然得要有一個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)結(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()方法再改進一下：

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;
}

這段代碼可以看到，如果測試方法標注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ù)源值。大家可以運行的看看結(jié)果。

六.　工程結(jié)構(gòu)圖：