java单元测试(2)mock

在Java单元测试中，Mock是一种常见的测试技术，可以用于模拟对象或方法，以便更好地控制测试环境。以下是Java单元测试中Mock的完整攻略：

1. 引入Mockito依赖

在使用Mockito进行单元测试时，需要先引入Mockito依赖。可以在Maven或Gradle中添加以下依赖：

<!-- Maven -->
<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-core</artifactId>
    <version>3.12.4</version>
    <scope>test</scope>
</dependency>

// Gradle
testImplementation 'org.mockito:mockito-core:3.12.4'

2. 创建Mock对象

在使用Mockito进行单元测试时，需要先创建Mock对象。可以使用Mockito的静态方法mock创建Mock对象。例如，创建一个名为userService的Mock对象，可以使用以下代码：

UserService userService = Mockito.mock(UserService.class);

3. 模拟方法调用

在使用Mockito进行单元测试时，可以使用方法来模拟方法调用。例如，要模拟userService的getUser方法返回一个名为“张三”的用户对象，可以使用以下代码：

User user = new User();
user.setName("张三");
when(userService.getUser(1)).thenReturn(user);

在上面的代码中，使用when方法模拟userService的getUser方法，当传入参数为1时，返回一个名为“张三”的用户对象。

4. 验证方法调用

在使用Mockito进行单元测试时，可以使用verify方法来验证方法调用。例如，要验证userService的saveUser方法是否被调用过一次，可以使用以下代码：

User user = new User();
userService.saveUser(user);
verify(userService, times(1)).saveUser(user);

在上面的代码中，使用verify方法验证userService的saveUser方法是否被调用过一次，并且传入的参数为user对象。

5. 使用注解@Mock和@InjectMocks

在使用Mockito进行单元测试时，可以使用注解@Mock和@InjectMocks简化Mock对象的创建和注入。例如，要创建一个名为userService的Mock对象，并将其注入到名为userController的对象中，可以使用以下代码：

public class UserControllerTest {
    @Mock
    private UserService userService;

    @InjectMocks
    private UserController userController;

    @BeforeEach
    public void setUp() {
        MockitoAnnotations.openMocks(this);
    }

    @Test
    public void testGetUser() {
        User user = new User();
        user.setName("张三");
        when(userService.getUser(1)).thenReturn(user);

        User result = userController.getUser(1);
        assertEquals("张三", result.getName());
    }
}

在上面的代码中，使用注解@Mock和@InjectMocks来创建Mock对象和注入对象，并使用@BeforeEach注解来初始化Mock对象。

示例1：模拟异常

在Java单元测试中，我们可以使用Mockito来模拟异常的抛出。例如，要测试userService的getUser方法在传入参数为0时是否会抛出异常，可以使用以下代码：

@Test
public void testGetUserWithException() {
    when(userService.getUser(0)).thenThrow(new IllegalArgumentException());
    assertThrows(IllegalArgumentException.class, () -> userService.getUser(0));
}

在上面的代码中，使用when方法模拟userService的getUser方法，当传入参数为0时，抛出IllegalArgumentException异常。然后使用assertThrows方法来验证是否抛出了该异常。

示例2：模拟方法调用次数

在Java单元测试中，我们可以使用Mockito来模拟方法用的次数。例如，要测试userService的saveUser方法是否被调用过两次，可以使用以下代码：

@Test
public void testSaveUser() {
    User user = new User();
    userService.saveUser(user);
    userService.saveUser(user);
    verify(userService, times(2)).saveUser(user);
}

在上面的代码中，使用verify方法验证userService的saveUser方法是否被调用过两次，并且传入的参数为user对象。

总的来说，使用Mockito进行Java单元测试可以使得测试更加灵活和可控。通过上述的攻略，可以轻松地创建Mock对象、模拟方法调用、验证方法调用和使用注解@Mock和@InjectMocks，实现对Java代码的单元测试。同时，Mockito还支持模拟异常和模拟方法调用次数等高级功能，可以更好地满足测试需求。