13-14. @InjectMocks、@Captor、ArgumentCaptor、captor.capture()、captor.getValue()

@InjectMocks修饰的对象,会创建一个真实对象,会调用真实方法(若方法中再调用对象中的其他方法时,也是调用真实方法)

package lesson13_14;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import java.util.List;
import static org.hamcrest.CoreMatchers.equalTo;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@RunWith(MockitoJUnitRunner.class)
public class ArgumentCaptorTest {

    @Mock
    private List<String> list;
    @Mock
    private UserDao userDao;
    @Captor
    private ArgumentCaptor<User> captorAnnotation;

    /**
     * @InjectMocks作用:
     * 1.构造注入:
     *   通过UserService的构造函数将Mock的userDao注入UserService,等价于UserService userService = new UserService(userDao);
     * 2.会调用修饰对象的真实方法
     * */
    @InjectMocks
    UserService userService;

    @Test
    public void test() {
        ArgumentCaptor<String> captor = ArgumentCaptor.forClass(String.class);
        when(list.add("test")).thenReturn(true);
        list.add("test");

        verify(list).add(captor.capture());
        assertThat(captor.getValue(), equalTo("test"));
    }

    @Test
    public void test2() {
        User user = new User(1);
        when(userDao.delUser(user)).thenReturn(true);
        assertThat(userService.delUser(user), equalTo(true));
        assertThat(user.getType(), equalTo("D"));
    }

    @Test
    public void test3() {
        ArgumentCaptor<User> captor = ArgumentCaptor.forClass(User.class);

        User user = new User(1);
        when(userDao.delUser(user)).thenReturn(true);

        boolean result = userService.delUser(user);
        assertThat(result, equalTo(true));
        verify(userDao).delUser(captor.capture());
        assertThat(captor.getValue().getType(), equalTo("D"));
    }

    @Test
    public void testAnnotation() {
        User user = new User(1);
        when(userDao.delUser(user)).thenReturn(true);

        boolean result = userService.delUser(user);
        assertThat(result, equalTo(true));
        verify(userDao).delUser(captorAnnotation.capture());
        assertThat(captorAnnotation.getValue().getType(), equalTo("D"));
    }
}
package lesson13_14;

public class UserService {
    private final UserDao userDao;

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

    public boolean delUser(User user){
        user.setType("D");
        return userDao.delUser(user);
    }
}
package lesson13_14;

public class UserDao {
    public boolean delUser(User user) {
        throw new RuntimeException();
    }
}
package lesson13_14;

public class User {
    private final Integer id;
    private String type;

    public User(Integer id) {
        this.id = id;
    }

    public Integer getId() {
        return id;
    }

    public String getType() {
        return type;
    }

    public void setType(String type) {
        this.type = type;
    }

    @Override
    public String toString() {
        return "User{" +
                "id=" + id +
                ", type='" + type + '\'' +
                '}';
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值