Mockito的使用 @InjectMocks、@Spy、@Mock

项目中，有些函数需要处理某个服务的返回结果，而在对函数单元测试的时候，又不能启动那些服务，这里就可以利用Mockito工具，其中有如下三种注解：

@InjectMocks：创建一个实例，简单的说是这个Mock可以调用真实代码的方法，其余用@Mock（或@Spy）注解创建的mock将被注入到用该实例中。
@Mock：对函数的调用均执行mock（即虚假函数），不执行真正部分。
@Spy：对函数的调用均执行真正部分。

Mockito中的Mock和Spy都可用于拦截那些尚未实现或不期望被真实调用的对象和方法，并为其设置自定义行为。二者的区别在于Mock不真实调用，Spy会真实调用。

具体的工作中会遇到的具体案例：
问题1：
实例对象的注入对象和注入对象含有相同的属性。举例说明：实例对象ClassA中含有注入对象ClassB、ClassC，实例对象ClassB中也含有ClassC。
问题2：
实例对象中含有太多的注入类，并且大部分的代码走真实调用方式，只想要数据库操作的部分代码走Mock调用的方式。

对于上面两个问题其实要解决的根本问题就是：如何解决Mock注入冲突，同时注入两个甚至多个对象中？

首先介绍一下Mockito给我们提供的一些工具类：通过AopTestUtils对切面对象进行mock

上面的博文中提供给了我们很好的一个思路，我们可以通过切面方式把每一个Mock对象都放到实例对象中。
以下是我的解决方案；

首先：将实例对象注入，同时使用Autowired和@InjectMocks注解

再者，将Mock对象通过反射写入到实例对象中

最后，规定好返回参数就可以进行流程的测试了。

SpringBootTest 注入 mock Mybatis mapper的例子：

@SpringBootTest
@AutoConfigureMockMvc
class ABCTest {
	
    // 注入真实的ABCService
	@Autowired
	@InjectMocks
	ABCService abcService;
	
	@Test
	void ABCtest_001() throws Exception {
        // 通过AopTestUtils获得Springboot注入的ABCService对象
		ABCService service = AopTestUtils.getTargetObject(abcService);
		ABCMapper abcMapperMock = mock(ABCMapper.class);
        // 通过ReflectionTestUtils 反射往ABCService注入mock对象
		ReflectionTestUtils.setField(service, "abcMapper", abcMapperMock);
		doThrow(new RuntimeException()).when(abcMapperMock).deleteByPrimaryKey(any());
		
}

单元测试之@Mock与@InjectMocks的例子

@Mock与@InjectMocks一般搭配组合使用，是单元测试必不可少的注解

@Mock：需要模拟的类，我们需要模拟哪些类，就用它修饰哪些类的变量，常用于第三方服务service
@InjectMocks：要测试的类，使用@Mock修饰的对象，就是我们测试哪个类，就用它修饰对应的变量，会整合使用@Mock修饰的对象

@Service
public class ThirdService {

    public Object getThirdUser(String userId) {
        return new Object();
    }
}

@Service
public class UserService {

    @Autowired
    private ThirdService thirdService;

    public Object getUser(String userId) {
        return thirdService.getThirdUser(userId);
    }
}

//测试UserService
@SpringBootTest
class UserServiceTest {

    //需要模拟的类(因为UserService中使用了这个类)
    @Mock
    private ThirdService thirdService;

    //要测试的类，使用@Mock修饰的对象
    //这时候userService对象中持有的thirdService变量就是模拟的对象了
    @InjectMocks
    private UserService userService;

    @Test
    public void testGetUser() {
        //设定行为返回数据，我们可以设定模拟对象的行为，当然也可以不设定
        //any代表任意参数，返回a对象
        Object a = new Object();
        when(thirdService.getThirdUser(any())).thenReturn(a);

        //执行被测试的方法，这时候内部调用的ThirdService就是模拟对象了，行为结果就是上面设置的
        Object abc = userService.getUser("abc");

        //验证结果
        assertEquals(a,abc);
    }

}

92IT

Just love IT.

Mockito的使用 @InjectMocks、@Spy、@Mock