编写一个以Hashmap为集合的数据提供程序类，并在API测试中为多个参数传递它

Question

我试图通过以TestNG DataProvider的形式编写一个可重用的组件来最小化我的测试中的代码行。我需要发送到服务器的测试规范接受一个Map

 @Test(dataProvider = "provideData")
 public void TestMethod(Map<String,Object> map) throws Exception {
RequestSpecification spec = generateCommonReqSpecJsonWithQueryParams(map);
Response res = RestOperationUtils.get(url, spec, null);
}

@DataProvider(name="provideData")
    public static Object[][] getData() throws Exception {
        Map<String, ArrayList<String>> map = new HashMap<>();
        ArrayList<String> a1 = new ArrayList<>();
        a1.add("First Value");
        a1.add("Second Value);
        a1.add("Third Value");
        a1.add("Fourth Value");
        map.put("Test[]", a1);
        map.put("month_start", new ArrayList(Arrays.asList("2019-06-01")));
        map.put("month_end", new ArrayList(Arrays.asList("2019-06-30")));
        map.put("viewers[]", new ArrayList(Arrays.asList("ESPN")));
        ArrayList<String> b1 = new ArrayList<>();
        b1.add("Fifth Value");
        b1.add("Sixth Value");
        b1.add("Seventh Value");
        map.put("Result[]", b1);

由于TestNG要求我们从DataProvider返回Object[][]，以下是我尝试过的不同方法：

方法1：

String[] keys = new String[map.size()];
        ArrayList<ArrayList<String>> values = new ArrayList<>();
        int index = 0;
        for (Map.Entry<String, ArrayList<String>> mapEntry : map.entrySet()) {
            keys[index] = mapEntry.getKey();
            values.add(index, (ArrayList<String>) mapEntry.getValue());
            //   x[index] = mapEntry.getValue();
            index++;
        }
        Object[][] result = new Object[values.size()][];
        index = 0;
        int index2;
        for (List<String> list : values) {
            result[index] = new Object[list.size()];
            index2 = 0;
            for (String item : list) {
                result[index][index2] = item;
                index2++;
            }
            index++;
        }
return result ;

方法2：

     Object[][] arr = new Object[map.size()][2];
        Set entries = map.entrySet();
        Iterator entriesIterator = entries.iterator();
        int i = 0;
        while(entriesIterator.hasNext()){
            Map.Entry mapping = (Map.Entry) entriesIterator.next();
            arr[i][0] = mapping.getKey();
            arr[i][1] = mapping.getValue();
            i++;
        }
        return arr;

方法3：

只需返回以下内容：

return new Object[][] {{map}};

方法1：使用方法1，它给出了5个预期参数，但由于我需要map在我的规范中作为queryParameters传递，我发现在我的测试方法中很难使用它来从DataProvider类中读取它们。

方法2：它给我返回2个参数，并使用Map

方法3：我不理解它为什么/如何工作，但调试后发现它是TestNg库的特例，否则我们需要使用方法1/2将hashmap转换为对象[]。

如果此查询需要任何其他信息，请告诉我。




Answer 1

方法3对我来说似乎完全有效。对象[][]只是保存所有测试用例的一种方式，其中每个对象[]索引都只是测试用例。然后，每个测试用例都应该匹配测试方法期望的参数的数量和类型。

选择Object[][]来保存测试用例，因为java中的所有对象要么从Object扩展，要么在原语的情况下可以自动装箱到其确实从Object扩展的对象形式中。

然后，TestNG将处理将数据提供者连接到其每个测试，以及为每个测试应用和转换测试用例参数。

例如：

@Test(dataProvider="getTestCases")
public void test(List<Integer> list, double d){
    // test code
}

你可能会想到：

@DataProvider
public Object[][] getTestCases(){
    return new Object[][] {
        {Arrays.asList(1, 2, 3), 1.0},
        {Arrays.asList(4, 5, 6), 2.0}
    };
}

其中{Arrays.asList（1,2,3），1.0}将是测试用例1，{Arrays.asList（4,5,6），2.0}将是测试用例2。

编辑：

为了解决清理数据提供程序的代码更改问题，Holger介绍了以下内容：

@DataProvider(name="provideData")
public static Object[][] getData() {
    Map<String, List<String>> map = new HashMap<>();
    map.put("Test[]", Arrays.asList("First Value", "Second Value", "Third Value", "Fourth Value"));
    map.put("month_start", Arrays.asList("2019-06-01"));
    map.put("month_end", Arrays.asList("2019-06-30"));
    map.put("viewers[]", Arrays.asList("ESPN"));
    map.put("Result[]", Arrays.asList("Fifth Value", "Sixth Value", "Seventh Value"));
    return new Object[][]{{map}};
}

至于方法1和2不适用于您的原因，它与数据提供程序类型/返回测试用例的数量不匹配有关。您的测试需要向其提供地图

public void TestMethod(Map<String,Object> map) 

它需要一个Map类型的参数

我建议将测试改为接受Map

在测试中，简单性是首选的，因为如果你给测试本身增加了太多的复杂性。测试可能比它正在测试的东西更容易出错。因此，简单的map返回就足够了。