import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
 * @description 利用CSV文件做数据驱动测试
 * @author rongrong
 * @version 1.0
 * @date 2020/6/27 13:22
 */
public class TestCSVData {


    /**
     * 读取csv文件
     *
     * @param filePath
     * @return返回list集合
     */
    public List<Map<String, String>> getCSVData(String filePath) {
        List<Map<String, String>> list = new ArrayList<Map<String, String>>();
        ;
        FileInputStream fileInputStream = null;
        try {
            fileInputStream = new FileInputStream(filePath);
            InputStreamReader inputStreamReader = new InputStreamReader(fileInputStream);
            BufferedReader br = new BufferedReader(inputStreamReader);
            //通过for循环遍历，将读取的每一行数据使用逗号进行分割并保存到map中
            for (String line = br.readLine(); line != null; line = br.readLine()) {
                Map<String, String> map = new HashMap<String, String>();
                String key = line.split(",")[0];
                String value = line.split(",")[1];
                map.put("userName", key);
                map.put("passWord", value);
                //再将map对象放到list集合中
                list.add(map);
            }
            br.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return list;
    }

    @DataProvider
    public Object[][] testCSVData() {
        //读取csv文件数据，调用getCSVData()方法来获得测试的数据
        List<Map<String, String>> result = getCSVData("d:\\data.csv");
        Object[][] files = new Object[result.size()][];
        //通过for循环遍历list，每一行的数据都会通过构造函数来进行初始化赋值
        for (int i = 0; i < result.size(); i++) {
            files[i] = new Object[]{result.get(i)};
        }
        return files;
    }


    @Test(dataProvider = "testCSVData")
    public void testCSVData(Map<String, String> param) {
        System.out.println(param.get("userName") + "\t" + param.get("passWord"));
    }
}

package com.data.demo;

import org.testng.annotations.Test;

import java.util.Map;
/**
 * @description 读取mysql数据库
 * @author rongrong
 * @version 1.0
 * @date 2020/6/27 14:22
 */
public class TestDbData extends DbDataHeleper {

    @Test(dataProvider = "dbDataMethod")
    public void testmethod1(Map<?, ?> param) {
        System.out.println(param.get("username") + "\t" + param.get("passWord") + "\t" + param.get("remark"));
    }
}

package com.data.demo;

import org.testng.annotations.DataProvider;
 
import java.sql.*;
import java.util.*;
/**
 * @description 读取mysql数据库
 * @author rongrong
 * @version 1.0
 * @date 2020/6/27 14:22
 */
public class DbDataHeleper {
 
    static Connection conn = null;
 
    public static String driverClassName = "com.mysql.jdbc.Driver";
    public static String url = "jdbc:mysql://127.0.0.1:3306/demo?serverTimezone=UTC&useSSL=false&allowPublicKeyRetrieval=true&zeroDateTimeBehavior=convertToNull";
    public static String username = "root";
    public static String password = "root";
 
 
    /**
     * 执行sql
     *
     * @param jdbcUrl 数据库配置连接
     * @param sql     sql语句
     * @return
     */
    public static List<Map<String, String>> getDataList(String jdbcUrl, String sql) {
        List<Map<String, String>> paramList = new ArrayList<Map<String, String>>();
        Map<String, String> param = new HashMap<String, String>();
        Statement stmt = null;
        try {
            // 注册 JDBC 驱动
            Class.forName(driverClassName);
            // 打开链接
            conn = DriverManager.getConnection(jdbcUrl, username, password);
            // 执行查询
            stmt = conn.createStatement();
            ResultSet rs = null;
            rs = stmt.executeQuery(sql);
            String columns[] = {"username", "passWord", "remark"};
            // 展开结果集数据库
            while (rs.next()) {
                Map<String, String> map = new LinkedHashMap<String, String>();
                for (int i = 0; i < columns.length; i++) {
                    String cellData = rs.getString(columns[i]);
                    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
    public Object[][] dbDataMethod() {
        String sql = "SELECT * FROM `account`;";
        List<Map<String, String>> result = getDataList(url, sql);
        Object[][] files = new Object[result.size()][];
        for (int i = 0; i < result.size(); i++) {
            files[i] = new Object[]{result.get(i)};
        }
        return files;
    }
}