Testng-在Java中循环数据提供程序或类？

如果可能的话，我想在我的计划布局上寻求一些帮助

目前，我拥有以下功能，效果良好：

@dataprovider连接到mysql数据库，然后将300个结果存储到一个数组中 @所有300个条目的测试循环和所有测试通过

我需要的，我正在努力解决的是：

@test完成后，我需要返回数据提供者（带有一个变量），然后再次运行测试

如果您想象一个目录结构，我会查询数据库中以a开头的所有类别，测试结果，然后重新查询数据库中以B开头的类别，并重新测试结果

不知道该怎么做，到目前为止，我已经尝试在@dataprovider中添加一个循环，因此该数组有300个a类别、300个B类别等。但这给了我数组的各种问题

有什么想法可以实现这一点，或者我需要为每个数据查询单独设置一个类？

---

Mike

数据提供程序可以返回对象[]数组。你有没有试过这样的方法：

@DataProvider(name="categoryloader")
public Object[][] createData(){
    return new Object[][]{
            {"CatgoryA",new String[300]},
            {"CatgoryB",new String[300]},
    };
}