Mockito 如何 mock 返回值为 void 的方法 | 隔叶黄莺 Yanbin Blog

2017-01-16 | 阅读(27,177)

最初接触 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

public interface UserDao {
  void save(User user);
}

public interface UserDao {

void save(User user);

}

类 UserService

public class UserService {
  private UserDao userDao;

  public UserService(UserDao userDao) {
    this.userDao = userDao;
  }

  public void saveUser(User user) {
    userDao.save(user);
  }
}

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) 的测试方法

@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);
}

@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 异常, 它的代码实现如下：

//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");
  }
}

//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");

}

要测试上面那个异常处理过程，我们可以写下面的测试

@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);
}

@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 的

@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());
}

@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 时还是很有用的，如下面的测试

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);
  
    ......
}

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();

M	T	W	T	F	S	S
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