Mockito 如何 mock 返回值为 void 的方法
转载:https://unmi.cc/mockito-how-to-mock-void-method/#more-7748
最初接触 Mockito 还思考并尝试过如何用它来 mock 返回值为 void 的方法,然而 Google 查找到的一般都会说用 doThrow()
的办法
doThrow(new RuntimeException()).when(mockObject).methodWithVoidReturn();
因为无法使用常规的 when(mockObject.foo()).thenReturn(...)
的方法。
当时我就纳闷,为何我想 mock 一个返回值为 void 的方法,却是在模拟抛出一个异常,现在想来如果一个返回值为 void 的方法,为何要去 mock 这个方法呢?
回想一个我们要 mock 一个方法的意图是什么:
- 在特定输入参数的情况下期待需要的输出结果(返回值)
- 在方法抛出某种类型异常调用者作出的反应
对于 void 返回值的方法,如果要验证有没有被调用过几次可以在事后用 verify()
方法去断言。所以基本上对于 void 返回值的方法一般可不用去 mock 它,只需用 verify() 去验证,或者就是像前面一样模拟出现异常时的情况。
所以本文并不像是去直接回答标题所示的问题: Mockito 如何 mock 返回值为 void 的方法,而是如何应对 mock 对象的 void 方法
一般不用对 void 方法打桩, 事后 verify 就行
测试代码针对 mock 对象的 void 方法调用本来就没有什么效果,所以一般也无须用 doNothing()
, 况且 void 提供不了返回值作进一步 mock,只需要在事后用 verify()
进行验证一下。
例如有下面的代码
类 UserDao
1
2
3
|
public interface UserDao {
void save(User user);
}
|
类 UserService
1
2
3
4
5
6
7
8
9
10
11
|
public class UserService {
private UserDao userDao;
public UserService(UserDao userDao) {
this.userDao = userDao;
}
public void saveUser(User user) {
userDao.save(user);
}
}
|
要写一个针对 UserService.saveUser(User user)
的测试方法
1
2
3
4
5
6
7
8
9
10
|
@Test
public void shouldCallUserDaoSaveMethod() {
UserDao userDao = Mockito.mock(UserDao.class);
UserService userService = new UserService(userDao);
User user = new User(1, "Yanbin");
userService.saveUser(user);
verify(userDao, times(1)).save(user);
}
|
模拟 void 方法抛出异常时调用者的响应
这就是为什么 Google 上那么多答案告诉我们要用到 doThrow()
的原因,假如我们的 UserService.saveUser(User user)
方法在 userDao
报出 DataAccessException 时转换为 ApplicationException
异常, 它的代码实现如下:
1
2
3
4
5
6
7
8
9
|
//UserService 的 saveUser(User user) 方法修改如下
public void saveUser(User user) {
try (
userDao.save(user);
catch(DataAccessException dae) {
throw new ApplicationException(dae, "can't save user due to issue reported from database");
}
}
|
要测试上面那个异常处理过程,我们可以写下面的测试
1
2
3
4
5
6
7
8
9
10
|
@Test(expected = ApplicationException.class)
public void raiseApplicationExceptionIfDataAccessExceptionOccursFromUserDao() {
UserDao userDao = Mockito.mock(UserDao.class);
UserService userService = new UserService(userDao);
User user = new User(1, "Yanbin");
doThrow(new DataAccessException()).when(userDao).save(user);
userService.saveUser(user);
}
|
上面的 doThrow()
行写成下面的方式也是一样的
doNothing().doThrow(new DataAccessException()).when(userDao).save(user);
Mockito 在调用 mock 对象的 void 方法时本来默认就是 doNothing
. 那什么时候需要用到 doNothing()
呢?就是下面的情况
部分 mock 时 void 方法可 doNothing
除了 @Mock
或 Mockito.mock(Class<T>)
来创建 mock 对象,还可以使用 spy(realObject)
或 spy(Class)
来获得 mock 对象,这两种方式得到的 mock 对象是不同的。前者所有的方法都被 mock, 后者(spy) 未打桩的方法会调用被 spy 对象的实际实现。
看下面的例子如何对 spy 对象的方法进行 mock 的
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
@Test
public void partialMockSpiedObject() {
List<String> names = new ArrayList<>();
List<String> spy = spy(names);
doNothing().when(spy).add(anyInt(), anyString());
spy.add("Yanbin");
spy.add(0, "Unmi");
assertEquals(1, spy.size());
assertEquals("Yanbin", spy.get(0));
assertEquals(0, names.size());
}
|
上面的测试可成功通过,我们用 doNothing().when(spy).add(anyInt(), anyString())
让带下标的方式添加元素什么也不做,而未被打桩的方法会调用 names
的实际实现。因此我们可以看到上面的断言结果。
doNothing()
还真不是 doNothing()
, 在搭配 spy 作部分 mock 时还是很有用的,如下面的测试
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
public class Handler {
private UserDao userDao;
void initialize() {
userDao = new UserDao();
}
public void handle(Event event) {
if(event != null) {
initialize();
}
......
}
}
//test code
@Test
public void testHandleEvent() {
Handler handler = spy(new Handler()); //spy 会是部分 mock, 只有 stub 了方法才会被 mock, 其余调用实际方法
//放心的用反射去修改 userDao 的值,因为不会被 initialize() 方法覆盖
UserDao mockedUserDao = Mockito.mock(UserDao);
setFieldValueOfUserDaoInReflection(handler, mockedUserDao);
doNothing().when(handler).initialize(); //因为它 doNothing, 保证了 handler 仍然持有 mockedUserDao
when(mockedUserDao.findUsers()).thenReturn(users);
......
}
|
另外如果还有其他情况可考虑 doCallRealMethod() 或 doAnser()
doCallRealMethod().when(mockObject).voidMethod();
doAnswer(answer).when(mockObject).voidMethod();
相关链接:
注(2017-09-19):如果用 spy
进行部分 mock 的时候,和常规方法 mock 略有不同,比如要先申请 doReturn(...)
1
2
3
4
5
6
7
8
|
UserService service = new UserService();
UserService spyService = spy(service);
doReturn("Yanbin").when(spyService).findNameById(123);
// spy 部分 mock 不能下面那样
// when(spyService.findNameById(123)).thenReturn("Yanbin");
// 否则 when(..) 打桩的时候就会执行实际的方法 service.findNameById(123),这是我们不希望的
|
再次说明,spy 打桩的方法被 Mock, 未打桩的会调用实际方法。