Mockito 单元测试打桩神器

    如上图所示，测试金字塔将测试分为三类

    • 单元测试: 由开发者自行编写，应当覆盖80%以上的场景。对微服务架构而言，主要是在单个微服务内部，对复杂业务逻辑，编写单元测试。
    • 集成测试: 由测试人员编写，强调系统整体联动，多偏向业务可用性验证。如下单流程是否畅通，库存扣减是否成功。它的覆盖场景一般之占10%。
    • 功能测试: 一些由单元测试、集成测试不好做的，通过功能测试完成，一般来说，这类不好自动化的，需要手动进行测试。由于测试成功很高，这类一般只覆盖5%的场景。

    测试金字塔也向我们揭示了一个实时：单元测试是整个测试环节的根基，如果单元测试做不好，上层的集成测试、功能测试都会无从谈起。

    遗憾的是，多数开发者都不具备编写单元测试的良好习惯，甚至缺乏编写单元的动力。

    除了缺乏软件质量保障的意识外，”嫌麻烦”也是这类开发者面对单元测试的口头禅。

    本节将介绍Mockito，这是一个单元测试的利器。Mockito的出现，让我们可以更加轻松地编写单元测试。

    在介绍Mockito之前，先来解释下，我们为什么不推荐使用Spring Boot启动单元测试框架。

    在实际工作中，我也尝试过上述方法，但效果却并不太好，主要原因是：

    • 启动Spring Boot环境速度很慢，至少要3秒，而一般的单元测试都是毫秒级别。
    • 依赖管理需要手动声明，随着业务不断升级，经常忘记维护单元测试中的依赖，导致单元测试无法正常执行。

    基于上述原因，我强烈不推荐在单元测试中启动Spring Boot环境。

    对于服务之间存在依赖关系的场景，建议直接使用Mockito的打桩(Mock)进行。

    希望在仔细的阅读本节后，你也会爱上单元测试:-)

    Mockito

    在软件测试中，Mock指的是效仿、模仿。Mockito就是为了解决测试中的Mock问题而诞生的，它可以很好的解决单元测试中，由于不同类耦合而带来的难以测试的问题。

    还是以上面Spring Boot环境为例子。假设我们要测试A类，而类A又调用了B类和C类。此时可能有两种选择：

    1. 手动构造B和C。
    2. 通过启动Spring环境，自动地注入B和C。

    现在有了Mockito后，我们有了另外的思路：无需构造B和C，而是通过Mockito，”Mock”出B和C(构造符合接口但没有实现的类)，由于我们要测试的是A类中的逻辑，只要检查A调用B和C的时机、次数、参数是否正确，就可以了。

    首先是ServiceA和它的实现:

    3. import com.coder4.lmsia.abc.server.service.intf.ServiceA;
    4. import com.coder4.lmsia.abc.server.service.intf.ServiceB;
    5. import org.springframework.beans.factory.annotation.Autowired;
    6. import org.springframework.stereotype.Service;
    8. /**
    9.  * @author coder4
    10.  */
    11. @Service
    12. public class ServiceAImpl implements ServiceA {
    14.     @Autowired
    15.     private ServiceB serviceB;
    17.     @Override
    18.     public int methodA(int a, int b) {
    19.         if (a <= 10 && b <= 10) {
    20.             return a + b;
    21.         } else {
    22.             return serviceB.methodB(a, b);
    23.         }
    24.     }

    然后是服务B和它的实现:

    1. package com.coder4.lmsia.abc.server.service.impl;
    3. import com.coder4.lmsia.abc.server.service.intf.ServiceB;
    4. import org.springframework.stereotype.Service;
    6. /**
    7.  * @author coder4
    8.  */
    9. @Service
    10. public class ServiceBImpl implements ServiceB {
    12.     @Override
    13.     public int methodB(int a, int b) {
    14.         return a * b;
    15.     }
    16. }

    我们总结一下功能：

    • 在服务A中，若参数a和b都小于10，则返回求和结果，否则交给服务B处理。
    • 在服务B中，直接返回参数a和b的乘积结果。

    在编写单元测试前，先要引用对应的包，lmabc-server/build.gradle：

    这里要指出的是，mockito本身还是需要单元测试框架才能运行的，我们这里用的是最常见的JUnit。

    然后看一下单元测试

    1. package com.coder4.lmsia.abc.server;
    3. import com.coder4.lmsia.abc.server.service.impl.ServiceAImpl;
    4. import com.coder4.lmsia.abc.server.service.intf.ServiceA;
    5. import com.coder4.lmsia.abc.server.service.intf.ServiceB;
    6. import org.junit.Assert;
    7. import org.junit.Before;
    8. import org.junit.Test;
    9. import org.mockito.Mockito;
    10. import org.mockito.internal.util.reflection.Whitebox;
    12. import static org.hamcrest.CoreMatchers.is;
    14. /**
    15.  * @author coder4
    16. public class ServiceATest {
    18.     private ServiceA serviceA;
    20.     private ServiceB serviceB;
    22.     @Before
    23.     public void before() {
    24.         serviceA = new ServiceAImpl();
    25.         serviceB = Mockito.mock(ServiceB.class);
    26.         Whitebox.setInternalState(serviceA, "serviceB", serviceB);
    27.     }
    29.     @Test
    30.     public void testBelow10() {
    31.         Assert.assertThat(serviceA.methodA(1, 1), is(2));
    33.         Mockito.verifyZeroInteractions(serviceB);
    35.     }
    37.     @Test
    38.     public void testAbove10() {
    39.         serviceA.methodA(100, 1);
    41.         Mockito.verify(serviceB).methodB(100, 1);
    43.     }
    45. }

    我们分步解释一下：

    • before: 初始化ServiceA，因为我们要测试这个，所以必须手动初始化。而ServiceB我们在A的测试并不关注，直接Mock一个，并通过Whitebox注入到服务A中。
    • testBelow10: 前面服务实现已经介绍过，当参数a和b都小于10的场景，是在ServiceA中直接求和。所以这里我们验证两个的和，然后验证下是不是没有”碰过”服务B(verifyZeroInteractions)
    • testAbove10: 当任何一个参数大于10时候，实际会走服务B。所以我们验证下是否调用了服务B，且参数恰好是传给A的就好。

    中提供了更多有趣的例子，等待你的发掘。