1. 概述

JUnit和TestNG无疑是Java生态系统中两个最流行的单元测试框架。虽然JUnit启发了TestNG本身，但它提供了其独特的功能，并且与JUnit不同，它适用于功能和更高级别的测试。本文我们将通过介绍它们的功能和常见用例来讨论和比较这些框架。

2. 测试设置

在编写测试用例时，我们通常需要在测试执行之前执行一些配置或初始化指令，并在测试完成后执行一些清理。让我们在两个框架中评估这些。

JUnit 在每个方法和类之前和之后提供两个级别的初始化和清理。我们在方法级别有@BeforeEach、@AfterEach注释，在类级别有@BeforeAll和@AfterAll：

public class SummationServiceTest {    private static List<Integer> numbers;    @BeforeAll    public static void initialize() {        numbers = new ArrayList<>();    }    @AfterAll    public static void tearDown() {        numbers = null;    }    @BeforeEach    public void runBeforeEachTest() {        numbers.add(1);        numbers.add(2);        numbers.add(3);    }    @AfterEach    public void runAfterEachTest() {        numbers.clear();    }    @Test    public void givenNumbers_sumEquals_thenCorrect() {        int sum = numbers.stream().reduce(0, Integer::sum);        assertEquals(6, sum);    }}

请注意，此示例使用 JUnit 5。在之前的 JUnit 4 版本中，我们需要使用等效于 @BeforeEach 和 @AfterEach 的@Before和@After注释。 同样，@BeforeAll和@AfterAll是JUnit 4的@BeforeClass和@AfterClass的替代品。

与JUnit类似，TestNG也提供方法和类级别的初始化和清理。虽然@BeforeClass和@AfterClass在类级别保持不变，但方法级别的注释是@BeforeMethod和@AfterMethod：

@BeforeClasspublic void initialize() {    numbers = new ArrayList<>();}@AfterClasspublic void tearDown() {    numbers = null;}@BeforeMethodpublic void runBeforeEachTest() {    numbers.add(1);    numbers.add(2);    numbers.add(3);}@AfterMethodpublic void runAfterEachTest() {    numbers.clear();}

TestNG还提供@BeforeSuite，@AfterSuite，@BeforeGroup和@AfterGroup注释，用于套件和组级别的配置：

@BeforeGroups("positive_tests")public void runBeforeEachGroup() {    numbers.add(1);    numbers.add(2);    numbers.add(3);}@AfterGroups("negative_tests")public void runAfterEachGroup() {    numbers.clear(); }

此外，如果我们需要在 TestNG XML 配置文件中的 <test> 标记中包含的测试用例之前或之后进行任何配置，我们可以使用 @BeforeTest 和 @AfterTest：

<test name="test setup">    <classes>        <class name="SummationServiceTest">            <methods>                <include name="givenNumbers_sumEquals_thenCorrect" />            </methods>        </class>    </classes></test>

这两个框架都支持忽略测试用例，尽管它们的做法完全不同。JUnit 提供了@Ignore注释：

@Ignore@Testpublic void givenNumbers_sumEquals_thenCorrect() {    int sum = numbers.stream().reduce(0, Integer::sum);    Assert.assertEquals(6, sum);}

而 TestNG 使用带有参数“enabled”的@Test，布尔值为 true 或 false：

@Test(enabled=false)public void givenNumbers_sumEquals_thenCorrect() {    int sum = numbers.stream.reduce(0, Integer::sum);    Assert.assertEquals(6, sum);}

在 JUnit 和 TestNG 中都可以将测试作为一个集合一起运行，但它们以不同的方式进行。

我们可以使用 @Suite、@SelectPackages 和@SelectClasses注释对测试用例进行分组，并在 JUnit 5 中将它们作为一个套件运行。套件是测试用例的集合，我们可以将它们组合在一起并作为单个测试运行。如果我们想将不同包的测试用例分组以在套件中一起运行，我们需要@SelectPackages注释：

@Suite@SelectPackages({ "org.baeldung.java.suite.childpackage1", "org.baeldung.java.suite.childpackage2" })public class SelectPackagesSuiteUnitTest {}

如果我们希望特定的测试类一起运行，JUnit 5 通过@SelectClasses提供了灵活性：

@Suite@SelectClasses({Class1UnitTest.class, Class2UnitTest.class})public class SelectClassesSuiteUnitTest {}

以前使用 JUnit 4，我们使用@RunWith和@Suite注释实现了对多个测试进行分组和运行：

@RunWith(Suite.class)@Suite.SuiteClasses({ RegistrationTest.class, SignInTest.class })public class SuiteTest {}

在 TestNG 中，我们可以使用 XML 文件对测试进行分组：

<suite name="suite">    <test name="test suite">        <classes>            <class name="com.baeldung.RegistrationTest" />            <class name="com.baeldung.SignInTest" />        </classes>    </test></suite>

这表示注册测试和登录测试将一起运行。除了对类进行分组之外，TestNG还可以使用@Test（groups=“groupName”）注释对方法进行分组：

@Test(groups = "regression")public void givenNegativeNumber_sumLessthanZero_thenCorrect() {    int sum = numbers.stream().reduce(0, Integer::sum);    Assert.assertTrue(sum < 0);}

让我们使用 XML 来执行组：

<test name="test groups">    <groups>        <run>            <include name="regression" />        </run>    </groups>    <classes>        <class           name="com.baeldung.SummationServiceTest" />    </classes></test>

这将执行用组回归标记的测试方法。

5. 测试异常

使用注释测试异常的功能在 JUnit 和 TestNG 中都可用。让我们首先创建一个类，其中包含一个抛出异常的方法：

public class Calculator {    public double divide(double a, double b) {        if (b == 0) {            throw new DivideByZeroException("Divider cannot be equal to zero!");        }        return a/b;    }}

在 JUnit 5 中，我们可以使用 assertThrows API 来测试异常：

@Testpublic void whenDividerIsZero_thenDivideByZeroExceptionIsThrown() {    Calculator calculator = new Calculator();    assertThrows(DivideByZeroException.class, () -> calculator.divide(10, 0));}

在 JUnit 4 中，我们可以通过在测试 API 上使用 @Test(expected = DivideByZeroException.class)来实现这一点。使用 TestNG，我们也可以实现相同的功能：

@Test(expectedExceptions = ArithmeticException.class) public void givenNumber_whenThrowsException_thenCorrect() {     int i = 1 / 0;}

此功能意味着从一段代码中引发的异常，这是测试的一部分。

6. 参数化测试

参数化单元测试有助于在多种条件下测试相同的代码。借助参数化单元测试，我们可以设置一个从某个数据源获取数据的测试方法。主要思想是使单元测试方法可重用，并使用一组不同的输入进行测试。

在 JUnit 5 中，我们的优势是测试方法直接从配置的源使用数据参数。默认情况下，JUnit 5 提供一些源注释，例如：

@ValueSource：我们可以将其与 Short、Byte、Int、Long、Float、Double、Char 和 String 类型的值数组一起使用：

@ParameterizedTest@ValueSource(strings = { "Hello", "World" })void givenString_TestNullOrNot(String word) {    assertNotNull(word);}

@ParameterizedTest@EnumSource(value = PizzaDeliveryStrategy.class, names = {"EXPRESS", "NORMAL"})void givenEnum_TestContainsOrNot(PizzaDeliveryStrategy timeUnit) {    assertTrue(EnumSet.of(PizzaDeliveryStrategy.EXPRESS, PizzaDeliveryStrategy.NORMAL).contains(timeUnit));}

static Stream<String> wordDataProvider() {    return Stream.of("foo", "bar");}@ParameterizedTest@MethodSource("wordDataProvider")void givenMethodSource_TestInputStream(String argument) {    assertNotNull(argument);}

@ParameterizedTest@CsvSource({ "1, Car", "2, House", "3, Train" })void givenCSVSource_TestContent(int id, String word) {	assertNotNull(id);	assertNotNull(word);}

同样，我们还有其他来源，例如@CsvFileSource，如果我们需要从类路径和@ArgumentSource中读取CSV文件以指定自定义的，可重用的ArgumentsProvider。

在 JUnit 4 中，必须用 @RunWith 对测试类进行批注，使其成为参数化类，@Parameter使用 表示单元测试的参数值。

在 TestNG 中，我们可以使用 @Parameter 或@DataProvider注释对测试进行参数化。使用 XML 文件时，使用 @Parameter注释测试方法：

@Test@Parameters({"value", "isEven"})public void   givenNumberFromXML_ifEvenCheckOK_thenCorrect(int value, boolean isEven) {    Assert.assertEquals(isEven, value % 2 == 0);}

并在 XML 文件中提供数据：

<suite name="My test suite">    <test name="numbersXML">        <parameter name="value" value="1"/>        <parameter name="isEven" value="false"/>        <classes>            <class name="baeldung.com.ParametrizedTests"/>        </classes>    </test></suite>

虽然使用 XML 文件中的信息既简单又有用，但在某些情况下，您可能需要提供更复杂的数据。

为此，我们可以使用@DataProvider注释，它允许我们为测试方法映射复杂的参数类型。下面是对基元数据类型使用 @DataProvider 的示例：

@DataProvider(name = "numbers")public static Object[][] evenNumbers() {    return new Object[][]{{1, false}, {2, true}, {4, true}};}@Test(dataProvider = "numbers")public void givenNumberFromDataProvider_ifEvenCheckOK_thenCorrect  (Integer number, boolean expected) {    Assert.assertEquals(expected, number % 2 == 0);}

对象的@DataProvider：

@Test(dataProvider = "numbersObject")public void givenNumberObjectFromDataProvider_ifEvenCheckOK_thenCorrect  (EvenNumber number) {    Assert.assertEquals(number.isEven(), number.getValue() % 2 == 0);}@DataProvider(name = "numbersObject")public Object[][] parameterProvider() {    return new Object[][]{{new EvenNumber(1, false)},      {new EvenNumber(2, true)}, {new EvenNumber(4, true)}};}

同样，可以使用数据提供程序创建和返回要测试的任何特定对象。当与 Spring 等框架集成时，它很有用。

请注意，在 TestNG 中，由于@DataProvider方法不必是静态的，因此我们可以在同一测试类中使用多个数据提供程序方法。

7. 测试超时

超时测试意味着，如果执行未在某个指定时间段内完成，则测试用例应失败。JUnit 和 TestNG 都支持超时测试。在 JUnit 5 中，我们可以将超时测试编写为：

@Testpublic void givenExecution_takeMoreTime_thenFail() throws InterruptedException {    Assertions.assertTimeout(Duration.ofMillis(1000), () -> Thread.sleep(10000));}

在 JUnit 4 和 TestNG 中，我们可以使用 @Test(timeout=1000)进行相同的测试

@Test(timeOut = 1000)public void givenExecution_takeMoreTime_thenFail() {    while (true);}

TestNG 支持依赖测试。这意味着在一组测试方法中，如果初始测试失败，则将跳过所有后续相关测试，而不是像 JUnit 那样标记为失败。

让我们看一个场景，我们需要验证电子邮件，如果成功，将继续登录：

@Testpublic void givenEmail_ifValid_thenTrue() {    boolean valid = email.contains("@");    Assert.assertEquals(valid, true);}@Test(dependsOnMethods = {"givenEmail_ifValid_thenTrue"})public void givenValidEmail_whenLoggedIn_thenTrue() {    LOGGER.info("Email {} valid >> logging in", email);}

在 JUnit 4 或 TestNG 中执行测试方法没有定义的隐式顺序。这些方法只是按照 Java Reflection API 返回的方式调用。从 JUnit 4 开始，它使用更具确定性但不可预测的顺序。

为了获得更多控制，我们将用@FixMethodOrder注释来注释测试类，并提到一个方法排序器：

@FixMethodOrder(MethodSorters.NAME_ASCENDING)public class SortedTests {    @Test    public void a_givenString_whenChangedtoInt_thenTrue() {        assertTrue(          Integer.valueOf("10") instanceof Integer);    }    @Test    public void b_givenInt_whenChangedtoString_thenTrue() {        assertTrue(          String.valueOf(10) instanceof String);    }}

MethodSorters.NAME_ASCENDING参数按方法名称按字典顺序对方法进行排序。除了这个排序器，我们还有MethodSorter.DEFAULT和MethodSorter.JVM。

而 TestNG 还提供了几种控制测试方法执行顺序的方法。我们在@Test注释中提供了优先级参数：

@Test(priority = 1)public void givenString_whenChangedToInt_thenCorrect() {    Assert.assertTrue(      Integer.valueOf("10") instanceof Integer);}@Test(priority = 2)public void givenInt_whenChangedToString_thenCorrect() {    Assert.assertTrue(      String.valueOf(23) instanceof String);}

请注意，优先级基于优先级调用测试方法，但不能保证在调用下一个优先级之前完成一个级别的测试。

有时，在 TestNG 中编写功能测试用例时，我们可能会有一个相互依赖的测试，其中每个测试运行的执行顺序必须相同。为了实现这一点，我们应该使用 dependsOnMethods 参数来 @Test 注释，正如我们在前面的章节中看到的那样。

10. 自定义测试名称

默认情况下，每当我们运行测试时，测试类和测试方法名称都会打印在控制台或 IDE 中。JUnit 5 提供了一个独特的功能，我们可以在其中使用@DisplayName注释提及类和测试方法的自定义描述性名称。

此注释不提供任何测试优势，但它也为非技术人员带来了易于阅读和理解的测试结果：

@ParameterizedTest@ValueSource(strings = { "Hello", "World" })@DisplayName("Test Method to check that the inputs are not nullable")void givenString_TestNullOrNot(String word) {    assertNotNull(word);}

每当我们运行测试时，输出将显示显示名称而不是方法名称。目前，在TestNG中无法提供自定义名称。

11. 结论

JUnit和TestNG都是在Java生态系统中进行测试的现代工具。在本文中，我们快速浏览了使用这两个测试框架中的每一个编写测试的各种方法。