使用Mockito,如何拦截void方法上的回调对象?

时间:2010-08-27 06:11:36

标签: java unit-testing callback jaas mockito

我正在使用mockito来测试遗留的JAAS / LDAP登录模块。

javax.security.auth.callback.CallbackHandler接口定义了函数:

void handle(javax.security.auth.callback.Callback[] callbacks)

我希望callbacks包含NameCallback,这是需要操作以通过测试的对象。

有没有办法有效地模拟这个,或者我会更好地使用CallbackHandler的简化实现?

1 个答案:

答案 0 :(得分:35)

对于返回void的函数,请使用doAnswer()

doAnswer(...).when(mockedObject).handle(any(Callback[].class));

执行拦截的Answer必须作为doAnswer的参数输入,例如作为一个匿名类:

new Answer() {
  public Object answer(InvocationOnMock invocation) {
      Object[] args = invocation.getArguments();
      Mock mock = invocation.getMock();
      return null;
  }}

在这种情况下,args将是数组Callback[]