java - 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() 之前运行，并且它仍然可以正常工作。