单元测试之mock使用

一、简介

 一般程序中A类的m1方法调用B类的m2方法，而B类的m2方法又调用了C类的m3方法以此类推等等，而其中的某个方法的一些数据又需要调用其它服务或者查询数据库，一般单元测试只针对某个功能进行测试，但是如上面的情况在做单元测试时受程序结构、环境等条件限制就会变得非常复杂。mock可以模拟对象返回方式来解决与该单元功能不相关的依赖关系，即模拟B类的m2方法返回结果来进行A类的m1方法单元测试，排除受到B类C类等其它不相关因素的影响。

二、使用mock做单元测试的优点

1、效率高

就是跑Java代码，不需要启用Spring及连接数据库。

2、TDD（测试驱动开发）

即先编写单元测试用例，根据单元测试用例再编写程序代码。

3、并行开发

团队有多个开发人员时在把数据格式约定好后可使用mock模拟返回结果而不需等待功能开发完成后才能联调。

4、解决环境依赖问题

比如当网络不通、服务无法访问时也能进行单元测试。

三、mock实现原理

使用Stub（桩）技术动态的替换原程序功能。即在程序运行时将原本程序要调用class字节码的逻辑替换为mock代码。

四、mock使用

1、导入Maven依赖包

<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-core</artifactId>
    <version>2.8.9</version>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-module-junit4</artifactId>
    <version>1.7.4</version>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-api-mockito2</artifactId>
    <version>1.7.4</version>
    <scope>test</scope>
</dependency>

2、编写测试代码

2.1、模块接口调用代码

2.2、模块业务处理代码

 2.3、mock单元测试代码

/**
 * create with Daniel
 * Date: 2022/4/12
 **/
@RunWith(MockitoJUnitRunner.class)
public class MockTest {
    //注入业务对象
    @InjectMocks
    private Service service;
    //使用Mock对象
    @Mock
    private FeignService feignService;
 
    /**
    * 测试方法
    * @author: Daniel
    * @date: 2022/4/12
    * @return void
    */
    @Test
    public void tester(){
        //初始化动作
        MockitoAnnotations.initMocks(this);
        Mockito.when(feignService.doApi(Mockito.anyString())).thenReturn("我是Mock对象");
        String result = service.doSomething("name");
        System.out.println(result);
    }
}

运行结果

从上图可以发现在Service调用Feign方法时返回的不是Feign的功能代码逻辑，而是返回测试用例中Mock返回的代码，这样可有效的解决单元测试用例对环境依赖的问题。

2.4、mock的其它用法

        //方法多次调用返回不同的值
        Mockito.when(feignService.doApi(Mockito.anyString())).thenReturn("第一次调用").thenReturn("第二次调用");
        System.out.println(service.doSomething("name"));
        System.out.println(service.doSomething("name"));
        //Mock没有返回值方法
        Mockito.doNothing().when(feignService).noReturn();
        //模块异常情况
        Mockito.when(feignService.doApi(Mockito.anyString())).thenThrow(new RuntimeException("我是Mock异常"));
        try{
            service.doSomething("name");
        }catch (Exception e){
            System.out.println(e.getMessage());
        }

运行结果

3、PowerMock使用 

PowerMock主要用于模拟静态方法的调用。

3.1、增加工具测试类

3.2、Mock测试类调整

 在类上加以下二个注解

@RunWith(PowerMockRunner.class)
@PrepareForTest({MockUtils.class})

添加测试代码

@Test
    public void testStatic(){
        PowerMockito.mockStatic(MockUtils.class);
        PowerMockito.when(MockUtils.check(Mockito.any())).thenReturn(true);
        System.out.println(MockUtils.check(null));
        System.out.println(MockUtils.check(1));
        System.out.println(MockUtils.check(new Object()));
    }

运行结果

由些可见，无论传什么参数都永远返回True。