本文首发于 https://jaychen.cc
作者 jaychen

写一篇小文，介绍一下 Java 下单元测试工具 TestNG 的使用，代码在 IDEA 环境在编写。

单元测试，顾名思义，对系统中原子性的功能进行测试，一般情况下是单元测试是针对某个功能函数的测试。编写单元测试是系统开发中重要的一环，也是一项科学优雅的装 X 方式。而且，编写单元测试代码并不是一件很麻烦的事情，只要稍微学习就可以掌握这项技能。

TestNG 使用

快速体验

在开始之前，需要引入 TestNG 库，使用 maven 直接引入，在 pom.xml 添加依赖

<dependency>
    <groupId>org.testng</groupId>
    <artifactId>testng</artifactId>
    <version>6.8</version>
    <scope>test</scope>
</dependency>

使用 IDEA 新建一个项目，目录结构如下:

├── pom.xml
├── src
│   ├── main
│   │   ├── java
│   │   └── resources
│   └── test

新建一个 Demo.java 类，包含如下简单代码

public class Demo {
    public int add(int a, int b) {
        return a + b;
    }

    public int sub(int a, int b) {
        return a - b;
    }
}

在 IDEA 下使用快捷键 Ctrl + Shift + t 为其生成测试类

这里可以看到，我们选择了 TestNG 作为单元测试库，IDEA 自动为我们生成了单元测试类的类名，其命名规则为：被测试类+Test。最后勾选要进行测试的方法，这里我只选择 add 方法。

确定之后，会在 src/test/java 目录下生成对应的类文件，可以看到已经生成了 DemoTest.java 文件，其内容如下

public class DemoTest {
    @Test
    public void testAdd() throws Exception {

    }
}

此时，我们就可以开始编写测试代码。这里，单元测试的目的是为了测试 Demo#add 这个函数的功能是否准确正常，所以我们在 testAdd 中编写代码

@Test
public void testAdd() throws Exception {
    Demo d = new Demo();
    assertEquals(7, d.add(3, 4));
}

这里，使用了 assertEquals 进行断言，这句话相当于说：d.add(3,4) 的结果应该是 7，你帮我执行下 add 看看是不是返回 7。好了，一个单元测试的用例完成，之后就可以直接运行该测试方法，可以看到输出如下：

[TestNG] Running:
===============================================
Default Suite
Total tests run: 1, Failures: 0, Skips: 0
===============================================

表明这个测试通过，函数功能没错。如果我们把代码改成 assertEquals(d.add(3, 4), 8);，那么就会出现如下提示

Expected :8
Actual   :7
 <Click to see difference>


    at org.testng.Assert.fail(Assert.java:94)
    at org.testng.Assert.failNotEquals(Assert.java:494)
    ...........
===============================================
Default Suite
Total tests run: 1, Failures: 1, Skips: 0
===============================================

表明 add 方法的返回结果和期望的不同，方法可能存在 bug。

这里要注意一个问题，上面我们对 add 进行一次测试通过，不代表 add 方法就不存在 bug。assertEquals(d.add(3, 4), 7); 只是一个测试用例。这里要理清一个概念：为 add 函数编写了一个单元测试函数 testAdd，之后我们需要使用多个测试用例来测试 add 函数是否存在 bug。为了证明 add 没有 bug，需要考虑所有可能的情况，包括输入为0，输入的 a，b 参数为负数等等尽量的覆盖所有可能性。所以一个严谨的测试应该如下：

@Test
public void testAdd() throws Exception {
    Demo d = new Demo();
    assertEquals(d.add(3, 4), 7);
    assertEquals(d.add(-3, 4), 1);
    assertEquals(d.add(-3, -4), -7);
    assertEquals(d.add(0, 4), 4);
    assertEquals(d.add(0, 0), 0);
}

高级用法

上面的例子只是简单用法，旨在让初学者可以快速上手了解 TestNG 的用法，下面介绍一些高级用法来帮助我们更好的进行单元测试。

`@BeforeClass/@AfterClass` 和 `@BeforeMethod/@AfterMethod`

除了 @Test 注解，TestNG 还有两对常用的注解：@BeforeClass/@AfterClass 和 @BeforeMethod/@AfterMethod。这些注解的关系如下图：

从上图可以看出，@BeforeMethod/@AfterMethod 是在 @Test 注解函数执行之前/之后执行的钩子函数。在执行每一个 @Test 注解函数执行之前/之后都会执行 @BeforeMethod/@AfterMethod 注解函数。

@BeforeClass/@AfterClass 的作用和 @BeforeMethod/@AfterMethod 类似，不同的是，@BeforeClass/@AfterClass 是在初始化类的时候执行，这就意味着 @BeforeClass/@AfterClass 只会执行一次，而 @BeforeMethod/@AfterMethod 执行次数和 @Test 注解函数个数一样。

public class DemoTest {

    @BeforeClass
    public static void beforeClass() {
        System.out.println("before test....");
    }

    @BeforeMethod
    public void beforeTest() {
        System.out.println("before  test...");
    }

    @Test
    public void testAdd() {

        int res = new Dao().add(1, 2);
        Assert.assertEquals(res, 3);
    }

    @Test
    public void testSub() {
        int res = new Dao().sub(1, 2);
        Assert.assertEquals(res, -1);
    }

    @AfterMethod
    public void afterTest() {
        System.out.println("after  test....");
    }

    @AfterClass
    public static void afterClass() {
        System.out.println("after class....");
    }
}

执行上面的代码，可以看到 beforeClass/afterClass 只执行一次，而 beforeMethod/afterMethod 执行了两次。

这里还需要提一点：@BeforeClass/@AfterClass 注解的函数必须使用 static 修饰。

除了使用 assertEquals 断言函数测试结果之外，TestNG 还提供了一些额外的测试情况。

超时测试

在 @Test 注解中添加 timeOut 参数就可以进行超时测试，@Test(timeOut=10) 表示测试方法的运行时间应该低于 10ms，如果超时者测试失败。超时测试对于网络连接类的测试相当有用。超时测试具体用法如下

@Test(timeOut = 1)
public void testSub() {
    int i =0;
    while (i < 1000000000) {
        i++;
    }
}

异常测试

异常测试用于测试方法是否有抛出异常，通过 @Test(expected=NullPointerException.class) 来指定方法必须抛出 NullPointerException，如果没有抛出异常或者抛出其他异常则测试失败。

@Test(expectedExceptions = NullPointerException.class)
public void testSub() {
    throw new  NullPointerException();
}

依赖测试

有时候需要测试方法按照一个特定的顺序被调用，这个时候需要使用 @Test 注解的 dependsOnMethods 参数来指定依赖方法和方法的执行顺序

// test1 执行之前会先执行 test2， test3
@Test(dependsOnMethods = {"test2","test3"})
public void test1(){

}
@Test
public void test2(){

}
@Test
public void test3(){

}

好了，TestNg 的使用就到这里了，其实单元测试并不是一件麻烦的事情，花上一些时间学习一下很快就很上手。过了 TestNg 下次说下 Mockito 的使用。