如何将SQL查询返回的数据传入TestNG的DataProvider？

TestNG的DataProvider核心要求是返回Object[][]类型的二维数组，你已经完成SQL查询逻辑、拿到了结果列表，直接把列表内容转换成对应格式的二维数组返回即可，无需额外特殊配置。

具体实现步骤如下：

• 第一步：封装已完成的SQL查询逻辑为独立方法，根据查询字段返回对应类型的列表，比如仅查id就返回List<Integer>，查多个字段可返回自定义实体类列表或者List<Object[]>
• 第二步：在@DataProvider修饰的方法中调用上述SQL查询方法，把列表转换为符合要求的二维数组

代码示例 #

单字段（如仅查询id）场景 #

// 你已实现的SQL查询方法，返回id列表
public static List<Integer> queryIdsFromDB() {
    List<Integer> idList = new ArrayList<>();
    // 此处替换为你自己的JDBC/ORM查询逻辑，将查询到的id存入idList
    return idList;
}

@DataProvider(name = "dbIdProvider")
public static Object[][] getDBIdData(){
    List<Integer> idList = queryIdsFromDB();
    // 每个测试用例接收1个id参数，所以二维数组第二个维度长度为1
    Object[][] data = new Object[idList.size()][1];
    for (int i = 0; i < idList.size(); i++) {
        data[i][0] = idList.get(i);
    }
    return data;
}

多字段（如查询id、用户名）场景 #

// 多字段查询返回自定义实体类列表示例
public static List<User> queryUsersFromDB() {
    List<User> userList = new ArrayList<>();
    // 此处替换为你自己的查询逻辑，将id、username等字段封装为User对象存入列表
    return userList;
}

@DataProvider(name = "userDataProvider")
public static Object[][] getUserData(){
    List<User> userList = queryUsersFromDB();
    // 每个测试用例接收2个参数：id、username，所以二维数组第二个维度长度为2
    Object[][] data = new Object[userList.size()][2];
    for (int i = 0; i < userList.size(); i++) {
        User user = userList.get(i);
        data[i][0] = user.getId();
        data[i][1] = user.getUsername();
    }
    return data;
}

注意事项：

1. @DataProvider支持static和非static修饰，建议把SQL查询逻辑和数据转换逻辑拆分，避免代码冗余臃肿
2. 数据量较大时可开启TestNG的DataProvider并行执行特性，提升用例执行效率
3. 建议在SQL查询逻辑中添加异常捕获，避免查询失败直接导致用例初始化报错，可捕获异常后返回空数组或者抛出自定义初始化异常方便定位问题

内容的提问来源于stack exchange，提问作者brithwulf