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

最新推荐文章于 2024-03-18 17:34:04 发布

原创最新推荐文章于 2024-03-18 17:34:04 发布 · 626 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #java #jquery #数据库 #junit

Mockito 专栏收录该内容

12 篇文章

订阅专栏

本文介绍如何使用JUnit和Mockito进行单元测试，演示了@InjectMocks、@Mock和@Captor等注解的应用，并通过具体例子展示了如何模拟对象行为及验证方法调用。

@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 + '\'' +
                '}';
    }
}