Junit參數化測試方法（五）——測試數據來源於數據庫和excel

        很多時候我們需要測試數據庫中的數值是否正確，期望值則存在excel中，本文將對這個問題展開描述。

一、思路

       利用feed4junit讀取excel中的期望值，在測試方法中連接數據庫，讀取數據庫中的實測值。如下示意：

二、示例

1、要查詢的數據表結構如下：

2、期望值excel：

3、代碼：

法一：將數據庫操作類分離出來（缺點：數據庫連接關閉執行多次）

1）test：

package JunitTestSample0001;

import org.databene.benerator.anno.Database;
import org.databene.benerator.anno.Source;
import org.databene.feed4junit.Feeder;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;

import java.sql.*;
@RunWith(Feeder.class)
/*此法失敗，需要使用配置文件
@Database(
        id = "DEEJdb",   
        url = "jdbc:oracle:thin:@zs-PC:1521:ytdf",   
        driver = "oracle.jdbc.driver.OracleDriver",   
        user = "deejuser",   
        password = "deejuser") 
*/
@Database(id = "DEEJdb", environment = "DEEJdb")  
public  class JunitTest0001 {
    /*
    public static String actpakagsn;
    @Test
    @Source(id = "DEEJdb", selector = "select packagsn from product")
    public void actualdata(String packagsn) {
        System.out.println("-------------");
        Assert.assertEquals(packagsn,"11");
    }*/
    @Test
    @Source("testData2.xlsx")
    public void expectdata(String BeachID,String AreaID,String WaterTempEva,String WaveHeightEva,String WeatherEva,String TempEva,String WindEva){
        ConnectOracle conn=new ConnectOracle();
        ResultSet result= conn.ConnectOraclenew(BeachID);
        try {
            while (result.next()){
                // 當結果集不為空時
            // System.out.println(result.getRow()+":"+result.getString(1));
            Assert.assertEquals(AreaID,result.getString(2));
            Assert.assertEquals(WaterTempEva,result.getString(3));
            Assert.assertEquals(WaveHeightEva,result.getString(4));
            Assert.assertEquals(WeatherEva,result.getString(5));
            Assert.assertEquals(TempEva,result.getString(6));
            Assert.assertEquals(WindEva,result.getString(7));
            }            
                
        } catch (SQLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } 
        conn.closed();
    }    
}

2）數據庫操作類：

package JunitTestSample0001;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

public class ConnectOracle {
    private Connection con= null;// 創建一個數據庫連接;
    private ResultSet result = null;// 創建一個結果集對象
    private Statement stmt = null;
    // private String user = "sys as sysdba";
    private String user = "deejuser";
    private String password = "deejuser";
    private String className = "oracle.jdbc.driver.OracleDriver";
    private String url = "jdbc:oracle:thin:@zs-PC:1521:ytdf";

    public ResultSet ConnectOraclenew(String BeachID) {    
        try
         {
             Class.forName(className);// 加載Oracle驅動程序
             System.out.println("開始嘗試連接數據庫！");
             result=getCon(BeachID);           
         }
         catch (Exception e)
         {     
            e.printStackTrace();
         }
        return result;
    }

    /** 創建數據庫連接 */
    public ResultSet getCon(String BeachID) {
        ResultSet rs=null;
        try {
            con = DriverManager.getConnection(url, user, password);
            System.out.println("創建數據庫連接成功！");
            stmt = con
                    .createStatement(ResultSet.TYPE_SCROLL_SENSITIVE,
                            ResultSet.CONCUR_UPDATABLE);
            String sql = "select * from evaresult_beach where BeachID="+BeachID;
             rs= stmt.executeQuery(sql);            
        } catch (SQLException e) {
            // System.out.print(con);
            // System.out.println("創建數據庫連接失敗！");
            con = null;
            e.printStackTrace();            
        }
        return rs;
    }
    public void closed() {
        try {
             // 逐一將上面的幾個對象關閉，因為不關閉的話會影響性能、並且占用資源
            // 注意關閉的順序，最后使用的最先關閉
            if (result != null)
                result.close();
            if(stmt!=null)
                stmt.close();      
            if (con != null) {
                con.close();
            System.out.println("數據庫連接已關閉！");
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}

法二：將數據庫連接與關閉寫在@BeforeClass @AfterClass中（優點：數據庫連接關閉只執行一次）

package oceanmonitoringtest;

import org.databene.benerator.anno.Source;
import org.databene.feed4junit.Feeder;
import org.junit.AfterClass;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.runner.RunWith;

import java.sql.*;

@RunWith(Feeder.class) 
public  class Evaresult_beach2 {
    private static Connection con= null;// 創建一個數據庫連接;
    private static ResultSet result = null;// 創建一個結果集對象
    private static Statement stmt = null;
    
    @BeforeClass
    public static void setUp() throws Exception {
        System.out.println("開始嘗試連接數據庫！");
        Class.forName("oracle.jdbc.driver.OracleDriver");// 加載Oracle驅動程序
        con = DriverManager.getConnection( "jdbc:oracle:thin:@zs-PC:1521:ytdf", "deejuser", "deejuser");
        System.out.println("創建數據庫連接成功！");
      stmt = con.createStatement(ResultSet.TYPE_SCROLL_SENSITIVE,
                        ResultSet.CONCUR_UPDATABLE);              
    }
    @Test
    @Source("testData2.xlsx")
    public void beachevaresult(String BeachID,String AreaID,String WaterTempEva,String WaveHeightEva,String WeatherEva,String TempEva,String WindEva) throws Exception{
         String sql = "select * from evaresult_beach where AreaID="+AreaID;
          result= stmt.executeQuery(sql);
         while (result.next()){
                // 當結果集不為空時
            Assert.assertEquals(AreaID,result.getString(2));
            Assert.assertEquals(WaterTempEva,result.getString(3));
            Assert.assertEquals(WaveHeightEva,result.getString(4));
            Assert.assertEquals(WeatherEva,result.getString(5));
            Assert.assertEquals(TempEva,result.getString(6));
            Assert.assertEquals(WindEva,result.getString(7));
            }
    }
    
    @AfterClass
     public static void tearDown() throws Exception {
             // 逐一將上面的幾個對象關閉，因為不關閉的話會影響性能、並且占用資源
           // 注意關閉的順序，最后使用的最先關閉
           if (result != null)
               result.close();
           if(stmt!=null)
               stmt.close();      
            if (con != null)
                con.close();
            System.out.println("數據庫連接已關閉！");            
    }
}

4、執行結果