【java 菜鸟自动化实践之四】将数据库查询数据，转为TestNG适用的对象二维数组数据

数据库数据：

数据库操作：

import java.sql.*;
import org.apache.log4j.Logger;

public class MysqlConn {	
	private static Logger log = Logger.getLogger(MysqlConn.class);
	private static MysqlConn op;
	private Connection conn;
	private String driver;
	private String url;
	private String userName;
	private String userPassword;

	private MysqlConn() {
	}

	private MysqlConn(String driver, String url, String userName, String userPassword) throws Exception {
		this.driver = driver;
		this.url = url;
		this.userName = userName;
		this.userPassword = userPassword;
		try {
			// 加载mysql驱动器
			Class.forName(driver);
			// 建立数据库连接
			conn = DriverManager.getConnection(url, userName, userPassword);
		} catch (ClassNotFoundException e) {
			System.out.println("加载驱动器失败");
			e.printStackTrace();
		} catch (SQLException e) {
			System.out.println("注册驱动器失败");
			e.printStackTrace();
		}
	}

	// 获取数据库连接
	public void getConnection() throws Exception {
		if (conn != null) {
			Class.forName(this.driver);
			conn = DriverManager.getConnection(this.url, this.userName, this.userPassword);
		}
	}

	// 静态方法，返回mysqlconn实例
	public static MysqlConn instance() {
		if (op == null) {
			op = new MysqlConn();
		}
		return op;
	}

	// 静态方法，返回mysqlconn实例
	public static MysqlConn instance(String driver, String url, String userName, String userPassword) throws Exception {
		if (op == null) {
			op = new MysqlConn(driver, url, userName, userPassword);
		}
		return op;
	}

	// 访问数据库，执行插入、删除、更新操作
	public String execute_Update(String sql) throws Exception {
		getConnection();// 初始化数据库连接
		Statement stmt = this.conn.createStatement();// 创建Statement对象
		if (stmt.executeUpdate(sql) != 1) {
			return "failure";
		}
		return "success";
	}

	// 访问数据库，执行查询操作
	public ResultSet queryx(String sql) throws Exception {
		getConnection();
		Statement stmt = this.conn.createStatement();
		return stmt.executeQuery(sql);
	}

	// 关闭数据库链接
	public void closeDB() {
		try {
			conn.close();
		} catch (SQLException ex) {
			log.error(ex.getMessage());
		}
	}
} 

TestNG代码：

import java.util.HashMap;
import org.testng.annotations.Test;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.DataProvider;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.AfterTest;
import org.testng.annotations.BeforeSuite;
import MysqlConn;

public class run_Interface_testng extends Thread {

	private static  MysqlConn t_conn;
	
	@DataProvider(name = "Data")
	public Object[][] Authentication() throws Exception {
		// 获取数据库数据源用例数据，并返回Object[][]
		Object[][] results = get_sql_data.Get_DB_TestCase_data(t_conn,Loading_IT_sql);
		return results;
	}
	
	@Test(dataProvider = "Data")
	public void run_testcase(HashMap<String, String> data) throws Exception {
		log.info("当前执行的用例是：第" + (data.get("Num")) + "行:" + "test_module：【"
				+ data.get("Test_module") + "】--" + "testcase_name:【"
				+ data.get("Testcase_name") + "】--" + "Testcase_desc:【"
				+ data.get("Testcase_desc") + "】--" + "is_run_module:【"
				+ data.get("is_run_module") + "】--" + "Request_url：【"
				+ data.get("Request_url") + "】--" + "Method:【"
				+ data.get("Method") + "】--" + "Headers:【"
				+ data.get("Headers") + "】--" + "Accept_content_type:【"
				+ data.get("Accept_content_type") + "】--" + "Body:【"
				+ data.get("Body") + "】--" + "Checking_response_data:【"
				+ data.get("Checking_response_data") + "】--"
				+ "Checking_DB_data:【" + data.get("Checking_DB_data") + "】"
				+ "\n");
	
	}

	@BeforeMethod
	public void beforeMethod() {
	}

	@AfterMethod
	public void afterMethod() {
	}

	@BeforeClass
	public void beforeClass() {

	}

	@AfterClass
	public void afterClass() {
	}

	@BeforeTest
	public void beforeTest() throws Exception {
		System.out.println("Start running");
		Loading_IT_sql = "SELECT * from interface_testcase;";
		String T_mysql_driver = com.mysql.jdbc.Driver
		String T_mysql_url = jdbc:mysql://localhost:3306/auto_testcase?useUnicode=true&characterEncoding=UTF-8
		String T_mysql_DB_user = root
		String T_mysql_DB_password = 123456
		t_conn = MysqlConn.instance(T_mysql_driver, T_mysql_url, T_mysql_DB_user, T_mysql_DB_password);		
	}

	@AfterTest
	public void afterTest() {
		System.out.println("End running \n");
		t_conn.closeDB();
	}

	@BeforeSuite
	public void beforeSuite() {
	}
}

/**
 * 
 * 获取数据库数据存入二维数组
 * 
 * @return Object[][]
 * @author xzee
 */
public class Read_sql_data {
	private static Logger log = Logger.getLogger(Read_sql_data.class);
	
    @SuppressWarnings({ "unchecked", "rawtypes"})
	public Object[][] Get_DB_TestCase_data(MysqlConn t_conn,String sql) throws Exception {
    	ArrayList<String> arrkey = new ArrayList<String>();
    	ResultSet rs = t_conn.queryx(sql);
		ResultSetMetaData data = rs.getMetaData();
        // 总列数 
        int columns=data.getColumnCount();
		// 获得首行的列名，作为hashmap的key值
		for (int c = 1; c <= columns; c++) {
			String columnName = data.getColumnName(c);
			arrkey.add(columnName);
		}
        int rowTotalNum = 1;
        while (rs.next()) {
        	//获取总行数
        	rowTotalNum = rs.getRow() + 1;
        }
		HashMap<String, String>[][] map = new HashMap[rowTotalNum - 1][1];
		// 对数组中所有元素hashmap进行初始化
		if (rowTotalNum > 1) {
			for (int i = 0; i < rowTotalNum - 1; i++) {
				map[i][0] = new HashMap();
			}
			// 特别重要，否则取到的全是0。因为执行上面的while(rs.next())后，ResultSet对象的下标已指到0，所以rs需要重新赋值。
	        // API：当生成ResultSet对象的Statement对象关闭、重新执行或用来从多个结果的序列获取下一个结果时，ResultSet对象将自动关闭。
			rs = t_conn.queryx(sql);
			// 输出二维数组
			// 遍历所有的单元格的值添加到hashmap中
			for (int r = 1; rs.next(); r++) {
				for (int c = 1; c <= columns; c++) {
					String rowValue = rs.getString(c);
					map[r - 1][0].put(arrkey.get(c - 1), rowValue);
				}
			}
		} else {
			log.error("数据库中没有数据！");
		}
        return map;
    }	
}