单元测试之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依赖包


<span style="background-color:#f8f8f8"><span style="color:#333333"><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></span></span>

2、编写测试代码

2.1、模块接口调用代码

2.2、模块业务处理代码

2.3、mock单元测试代码


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

运行结果

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

2.4、mock的其它用法


<span style="background-color:#f8f8f8"><span style="color:#333333">        //方法多次调用返回不同的值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());}</span></span>

运行结果

3、PowerMock使用

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

3.1、增加工具测试类

3.2、Mock测试类调整

在类上加以下二个注解


<span style="background-color:#f8f8f8"><span style="color:#333333">@RunWith(PowerMockRunner.class)@PrepareForTest({MockUtils.class})</span></span>

添加测试代码


<span style="background-color:#f8f8f8"><span style="color:#333333">@Testpublic 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()));}</span></span>

运行结果