Mockito-提升单元测试效率利器

前言

以前在开发进行到写单元测试阶段的时候，发现要测试的方法里面是包含依赖的：外部接口RPC调用、DB调用。在某些状况下，部分依赖不稳定或者没法在测试环境调用时，会致使用例偶尔执行失败。

另一点，不少用例都是在测试用例的开头写了@SpringRunTest的注解，致使跑用例的时候会启动整个Spring容器，这样一来，运行测试用例就很是慢了。当在一些比较大的项目运行用例时，甚至达到了每次启动容器须要5-6分钟的时长，渐渐就有点受不了这种操做，每改一行代码内心都焦急，由于若是错了的话又要再等5-6分钟才能看到效果了。后来请教同事和上网搜索，找到了一种比较快且安全的方案，使用Mock框架--Mockito，学习并实践了一段时间，总结一下使用方法。

Mockito

Mockito是当前最流行的单元测试Mock框架。

什么是Mock

Mock的字面意思就是模仿，虚拟，在单元测试中，使用Mock能够虚拟出一个外部依赖对象。

对于在单元测试中一些不容易构造或者不容易获取的对象（如外部服务），用一个Mock对象来建立，能够下降测试的复杂度，只关心当前单元测试的方法。

为何用Mock

单元测试的目的就是为了验证一个代码单元的正确性，真正要验证的只是某个输入对应的输出的正确与否。若是把外部依赖服务引入进来，就会增长原来单元的复杂度，且在该单元中隐形地掺杂了其余功能的内容。

使用Mock对象进行单元测试，开发能够只关心要测试单元的代码。

使用示例

先看看代码示例，假设有如下的场景：

• 验证获取用户信息接口：包含用户ID、用户昵称、是否vip
• 是否vip须要外部服务VIPService获取，经过RPC调用，测试环境若是机器性能较差或者网络很差会致使用例不稳定
• 编写单元测试判断用户VIP信息返回是否正确

需求是判断获取用户信息接口返回的格式是否正确，与vip接口的返回值无关，只要透传vip接口返回的字段便可，测试代码以下：

@RunWith(PowerMockRunner.class)
public class UserServiceTest {

    @InjectMocks
    private UserService userService;

    @Mock
    private VIPService vipService;

    @Test
    public void getUserInfo() {
        Mockito.when(vipService.isVip(Mockito.anyString())).thenReturn(true);

        Result result = userService.getUserInfo("123");

        Assert.assertEquals(true, result.getData().get("isVip"));
    }
}
复制代码

解释下上面代码用到的几个注解

@Mock：建立一个Mock

@InjectMocks：Mock一个实例，其他用Mock注解建立的mock将被注入到该实例中。

Mockito.when(...).thenReturn(...)：Mock方法，若是知足when里面的条件，返回thenReturn指定的结果。

在这段代码里，使用@Mock注解建立了一个VipService实例，使用@InjectMock建立了UserService，Mock建立的vipService实例会被注入到UserService的实例中，在写测试用例的时候就能够模拟vipService的行为。

Mockito.when(vipService.isVip(Mockito.anyString())).thenReturn(true); 这段代码表示无论传任何参数给vipService.isVip方法，该方法都会返回true，这样，就不影响获取用户信息接口的正常测试，也可使用断言验证返回的数据。

遇到过的场景

以上是使用Mockito实践最简单的示例，在生产环境使用过程当中，会有各类各样的需求须要知足，下面列一下笔者遇到过的场景。

mock异常

这种场景是，方法里面声明了可能会抛出A异常，而A异常有多种可能性，不一样的异常对应不一样的message，为了验证抛出某种A异常后的功能，就须要模拟方法抛出指定message的A异常。

使用方式是定义一个Rule注解的属性，在使用时，设置thrown抛出的异常类型和所带的message。简要代码以下：

class AException extends RuntimeException {
    private final int code;

    public AException(int code, String msg) {
        super(msg);
        this.code = code;
    }
}

@RunWith(PowerMockRunner.class)
public class MockExceptionTest {

    @Rule
    public ExpectedException thrown = ExpectedException.none();

    @Test
    public void mockException() {

        thrown.expect(AException.class);
        thrown.expectMessage("expected message");

        // test code
    }

}
复制代码

mock空方法

mock一个空方法，比较简单，就是调用doNothing().when()...。

mock静态方法

若是要Mock静态方法，首先在类的开头增长注解：@PrepareForTest({ClassNameA.class})。

在须要Mock类方法的以前，增长代码：PowerMockito.mockStatic(ClassNameA.class);，而后就能够愉快的Mock了。简要代码以下：

class ClassNameA {

    public static int methodA() {
        // code

        return ret;
    }

}

@RunWith(PowerMockRunner.class)
@PrepareForTest({ClassNameA.class})
public class MockStaticClassTest {

    @Test
    public void mockStaticMethod() {
        PowerMockito.mockStatic(ClassNameA.class);

        Mockito.when(ClassNameA.methodA()).thenReturn(1);
        // test code
    }

}
复制代码

部分mock

对于某些场景，在一个单元测试里，须要某个方法Mock，某个方法走正常逻辑，这种操做就一点要启动容器，目前还没找到合适的方法能够进行这种操做，若是有更好的方法麻烦指点指点。笔者目前的作法是将原来的方法再拆分，拆分为更小的单元，让各自能够进行Mock，在集成测试时才真正执行所有代码。

以上是笔者在平常开发中遇到的场景

总结

单元测试是针对代码逻辑最小单元进行正确性检验的校验工做，写好单元测试，对于发现代码bug、保障系统稳定性以及重构而言都是很是必要的一项工做，能够提早发现一些隐藏问题。

JUnit最佳实践这篇文章提到，Mock全部外部服务和状态：

Mock out all external services and state Otherwise, behavior in those external services overlaps multiple tests, and state data means that different unit tests can influence each other’s outcome. You’ve definitely taken a wrong turn if you have to run your tests in a specific order, or if they only work when your database or network connection is active.

Also, this is important because you would not love to debug the test cases which are actually failing due to bugs in some external system.

因此，仍是尽量使用Mock来进行有外部服务的单元测试。

原创文章，文笔有限，才疏学浅，文中如有不正之处，万望告知。

若是本文对你有帮助，麻烦点个赞，谢谢

更多精彩内容，请关注我的公众号。

参考文章 JUnit Best Practices Guide