如果方法的参数实在无法预先确定,我们可以使用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); }