TestNG如何在每个测试用例中使用DataProvider之前重新初始化它？

Question

我最近在进行api测试时遇到有关使用testNG的dataProvider的问题。 我的问题是，在每次将数据提供者提供给测试用例之前，应如何重新初始化它？

@DataProvider(name = "testDataProvider")
public static Object[][] testDataProvider() {
    Map<String, String> params = new HashMap<>(2);
    params.put("a", "foo");
    params.put("b", "bar");
    return new Object[][]{
        {params}
    };
}

@Test(dataProvider = "testDataProvider")
public void testCase01(Map<String, String> params) {
    params.put("a", "oof");
    assertEquals(params.get("a"), "oof");
}

@Test(dataProvider = "testDataProvider")
public void testCase02(Map<String, String> params) {
    assertEquals(params.get("a"), "foo"); // this won't pass
}

我希望两个测试都通过。 但是，当前，如果testCase01在testCase02之前运行，则testCase02不会通过，因为我已经覆盖了params.get("a") 。 如何确保DataProvider始终在其定义中返回硬编码值？

Answer 1

如果您使用的是最新版本的TestNG（今天为7.0.0-beta7 ），这应该不是问题。

这是一个演示此的示例。

import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertFalse;
import static org.testng.Assert.assertTrue;

import java.io.File;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;
import org.testng.TestNG;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

public class Sample {

  @Test
  public void versionConfirm() {
    String file = TestNG.class.getProtectionDomain().getCodeSource().getLocation().getFile();
    String actualVersion = new File(file).getParentFile().getName();
    assertEquals(actualVersion, "7.0.0-beta7");
  }

  @DataProvider(name = "testDataProvider")
  public static Object[][] testDataProvider(Method method) {
    Map<String, String> params = new HashMap<>(2);
    params.put("a", "foo");
    params.put("b", "bar");
    params.put(method.getName(), Integer.toString(params.hashCode()));
    return new Object[][] {{params}};
  }

  @Test(dataProvider = "testDataProvider")
  public void testCase01(Map<String, String> params) {
    params.put("a", "oof");
    assertEquals(params.get("a"), "oof");
    assertTrue(params.containsKey("testCase01"));
    assertFalse(params.containsKey("testCase02"));
  }

  @Test(dataProvider = "testDataProvider", dependsOnMethods = "testCase01")
  public void testCase02(Map<String, String> params) {
    assertEquals(params.get("a"), "foo"); // this won't pass
    assertFalse(params.containsKey("testCase01"));
    assertTrue(params.containsKey("testCase02"));
  }
}

如您所见，在这里我通过使用dependsOnMethods强制testCase01()在testCase02()之前运行，但它仍然可以正常工作。