EasyMock Capture的使用

如果方法的参数实在无法预先确定，我们可以使用capture。比如UserService 有这么个方法:

public boolean registerVIP(){

User user = new User();

user.type = "VIP";

return userDao.insertUser(user);

}

 

当我们mock userDao的insertUser(user)方法时，我们无法确定他的参数会是什么样子的，因为参数是我们要测试的registerVIP这个方法new出来给他的，我们可以假定insertUser的方法逻辑的正确的，但是参数是registerVIP传给他的，所以我们得验证参数的正确性。

public void testRegisterVIP(){

userDao = createMock(UserDao.class);

Capture<User> captureUser = new Capture<User>();

expect(userDao.insertUser(capture(captureUser))).andReturn(true).times(2);

replay(userDao);

userService.setUserDao(userDao);

userService.registerVIP();

assertEquals("VIP", captureUser.getValue().type); //验证捕捉到的参数是否正确

assertEquals(true, userService.registerVIP());//验证最后的结果

verify(userDao);

}