使用Mockito进行单元测试
一、使用JUnit + Mockito进行单元测试
Java中的Mock工具有很多,比如Mockito, jMock, EasyMock等,在这里我们只说下Mockito,因为其他的我也没用过:-D
下面省去怎么引入mockito的步骤,希望可以用不到1h的时间,让大家可以使用Mockito写出期望的单测。
二、实战mockito
mockito
是一款开源的测试框架,可以方便地通过mock(模拟)方式进行测试验证。可以很方便地跟JUnit结合起来用。为了更好地使用mockito,我们先来了解下mockito中的几个核心概念。
2.1 mock
mock,就是伪造一个待测试的对象,可以随意摆布,让它干啥就干啥。
在mockito
中,可以通过TestClass mockObject = mock(TestClass.class)
来创建一个伪造的对象。
如果你调用mockObject中的方法,则什么事情都不会做,如果对应方法有返回值的话,那就返回返回类型在JAVA中的默认值。比如如果方法的返回值类型是boolean则返回false,如果返回值类型是String则返回null, List返回[]等
2.2 stub桩
桩(Stub),就是把要调用的方法模拟掉,让方法按照我们的期望行动。一般桩的目的,有三种
- doNothing : 啥也不做
- doReturn : 返回具体值
- doThrow: 抛出异常
对于doReturn,有两种实现方法,
- when(mockObject.invokeMethod(...params)).thenReturn(...) : 会真正执行invokeMethod内的逻辑,只是返回值会为设定的返回值。
- doReturn(...).when(mockObject).invokeMethod(...params) : 不会真正执行invokeMethod,只返回设定值。
doThrow逻辑跟doReturn一样。
2.2 mock V.S. spy
我们在使用mockito过程中,可以以mock的方式,也可以以spy的方式把实际的动作给mock掉。如
TestClass object = mock(TestClass.class); // mock
TestClass object = spy(TestClass.class); // spy
这两种方式有什么区别呢? 简而言之
mock就是对象是完完全全假的对象,其中所有的方法都会以桩的方式来执行。
spy,又称部分mock,如果设置了桩,则使用桩的返回值;如果未设置桩,则调用对象真实的方法
当然你也可以指定mock对象具体方法的行为,在2.2节已经讲过。
2.3 使用mock class, 导致@Autowired的bean注入为Null
在spring的项目中,我们通常会在类中通过注解的方式注入对象,比如
public class PersonService {
@Autowired
private PersonDao personDao;
public String getPersonName(Long id) {
return personDao.getPersonName(id);
}
}
如果你在mockito的单测中,直接写PersonService personService = mock(PersonService.class);
那在真正执行getPersonName(id)这个时候,对于personDao这个对象,是null。
原因就是单测对象被Mock掉之后,其中注入的对象spring是感知不到的。这个就跟我们自己通过new TestClass()方式无法注入@Autowired对象一个逻辑。
解决方案就是利用@InjectMocks
进行注入,注意在单测跑之前需要调用MockitoAnnotations.initMocks(this);
具体可以参考: https://dzone.com/articles/use-mockito-mock-autowired
2.4 利用verify做验证
对于Mock的对象,由于我们就是为了把实际的操作给mock掉,那mockito怎么验证测试是否启动成功? 在mockito中,可以记录每次调用的方法以及对应的参数,之后用verify进行验证。
比如,verify(bar, times(1)).someMethod();
来验证bar这个对象,调用了1次someMethod方法。