testng+mysql进行数据驱动（转）

共用方法mysql读取

import org.testng.annotations.DataProvider;

import org.testng.annotations.Test;

import java.sql.*;

import java.util.*;

public class JdbcReader {

    static Connection conn = null;

    public static List<Map<String, String>> getDataList(String driverClassName,

            String url, String username, String password, String sql,

            String columns[]) {

        List<Map<String, String>> paramList = new ArrayList<Map<String, String>>();

        Map<String, String> param = new HashMap<>();

        Statement stmt = null;

        try {

            // 注册 JDBC 驱动

            Class.forName(driverClassName);

            // 打开链接

            conn = DriverManager.getConnection(url, username, password);

            // 执行查询

            stmt = conn.createStatement();

            ResultSet rs = null;

            rs = stmt.executeQuery(sql);

            // 展开结果集数据库

            while (rs.next()) {

                Map<String, String> map = new LinkedHashMap<String, String>();

                for (int i = 0; i < columns.length; i++) {

                    String cellData = rs.getString(i + 1);

                    map.put(columns[i], cellData);

                }

                paramList.add(map);

            }

            // 完成后关闭

            rs.close();

            stmt.close();

            conn.close();

        } catch (SQLException se) {

            // 处理 JDBC 错误

            System.out.println("处理 JDBC 错误!");

        } catch (Exception e) {

            // 处理 Class.forName 错误

            System.out.println("处理 Class.forName 错误");

        } finally {

            // 关闭资源

            try {

                if (stmt != null)

                    stmt.close();

                if (conn != null)

                    conn.close();

            } catch (SQLException se) {

                se.printStackTrace();

            }

        }

        return paramList;

    }

}

dataprovider使用

@DataProvider

    public Object[][] dbDataMethod() throws ClassNotFoundException, SQLException {

        String sql = "select * from cs1";

        String columns[] = {"xm", "id"};

        List<Map<String, String>> result = getDataList( sql, columns);

        Object[][] files = new Object[result.size()][];

        for (int i = 0; i < result.size(); i++) {

            files[i] = new Object[]{result.get(i)};

        }

        return files;

    }