单元测试单元测试

使用Mockito进行单元测试

2017-08-11  本文已影响297人  rockops

一、使用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),就是把要调用的方法模拟掉,让方法按照我们的期望行动。一般桩的目的,有三种

  1. doNothing : 啥也不做
  2. doReturn : 返回具体值
  3. doThrow: 抛出异常

对于doReturn,有两种实现方法,

  1. when(mockObject.invokeMethod(...params)).thenReturn(...) : 会真正执行invokeMethod内的逻辑,只是返回值会为设定的返回值。
  2. 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方法。

上一篇下一篇

猜你喜欢

热点阅读