JMockit 中捕获 mock 对象方法的调用参数

三个月前写过一篇 Mockito 中捕获 mock 对象方法的调用参数，一般项目中 Mockito 不决求助于 JMockit, 同样的在 JMockit 也需对捕获被 Mock 的方法调用参数。当我们用 new Expectations(){{}} 打桩并在后面断言了返回值，那就无需捕获参数来断言，匹配到了方法调用即证明传入的参数也是对的，如下面的代码所示

public class UserServiceTest {

    @Mocked
    private UserDao userDao;

    @Test
    public void couldCallUserDaoToAddUser() {
        new Expectations(){{
           userDao.findById(123);
           result = "Yanbin";
        }};

        UserService userService = new UserService(userDao);
        String user = userService.findBy(123);

        assertThat(user).isEqualTo("Yanbin");  //这里断言成功也就证明了 userDao.findById(123) 方法被调用，参数必须是 123
    }
}

但如果是未打桩的方法，或打桩是用的模糊参数(withInstanceOf(String.class)), 或是无返回值的方法就要事后对是否调用了某个方法以及传入什么参数的情况进行断言。

再或者是中间调用曲折，调用参数无法直接提供的话，这时候就需要对调用参数进行捕获再断言。下面是一个完整的例子，立即来检验一下 JMockit 是如何操作的

首先是待测试的 UserService 类

和我们要 Mock 的 UserDao  接口，Mock 的优越性这里表现在我们根本没有接口实现，但并不妨碍我们写出完整的 UserService 的单元测试

接着就是我们捕获参数的测试代码

关键就是在上面的

userDao.add(userId = withCapture(), username = withCapture());

JMockit 恰到好处的用到 Java 的赋值语句返回等号右边的结果，相较而言 Mockito 捕获参数的方式就稍显笨拙，以下是 Mockito 的方式

ArgumentCaptor<User> argumentCaptor = ArgumentCaptor.forClass(User.class);
verify(userDao, times(1)).save(argumentCaptor.capture());

具体可参考前一篇文章 Mockito 中捕获 mock 对象方法的调用参数, 尤其是对于带泛型的参数的捕获 Mockito 一定要在声明带 @Captor 注解实例级的变量。而 JMockit 不管多复杂类型参数的捕获都毫无区别。

注意到 JMockit 在捕获参数是用的 withCapture(...) 方法，它有三个变体

上面刚刚用过的 withCapture() 简单了，就是直接把实际方法接受的值赋给一个变量。

withCapture(List<T> valueHolderForMultipleInvocations) 是在被 Mock 方法多次调用时，它会捕获每次调用的参数放到一个 List 中去。譬如在 UserService 中有一个方法用于批量删除多个用户

现在要捕获多次调用的参数就得

和 Mockito 的捕获后调用 argumentCaptor.getValues() 来取得多次调用的参数类似。

最后一个方法 withCapture(T constructorVerification) 是 Mockito 所不能的，因为 Mockito 是不能 Mock 构造函数的，而 JMockit 是无所不能的。官方文档对此的解释是

捕获在单元测试执行中新创建的实例，这些实例必须是由对构造函数调用时参数匹配时创建的。

还是有点矇，主要还是没怎么感觉到这个方法有什么实际意义，还未到某时非用不可以状况，作以直接贴一下官方的示例代码

主要还是掌握下前面两个捕获参数的方法

1. <T> T withCapture()   捕获单调用的参数值，以返回值的形式给下捕获到的值，所以可用 userDao.add(userId = withCapture(), ...) 的形式
     如果参数本身是列表也是用这种形式：

   List<Person> people;
   dao.save(people = withCapture());
   assertEquals(2, people.size()); 

2. <T> T withCapture(List<T> valueHolderForMultipleInvocations)  捕获多次调用的参数放到一个列表中，它的返回值只会保留最后一次的调用参数，需要获得全部捕获的值必须传递一个 List<T> 给 withCapture 方法。这类似于 Mockito 在捕获多次调用参数后，captor.getValues() 获得所有，而 captor.getValue() 只是最后一次的调用参数值。

参考：Capturing invocation arguments for verification