本文旨在解决在使用Mockito进行单元测试时，遇到的变量值无法被Mock覆盖的问题。通过分析问题原因，并提供清晰的代码示例和解决方案，帮助开发者更好地理解和使用Mockito进行有效的单元测试，确保代码的正确性和可靠性。
在进行单元测试时，Mockito是一个强大的Mocking框架，允许我们创建Mock对象，模拟依赖项的行为，从而隔离被测试的代码。然而，有时我们会遇到Mockito无法覆盖变量值的情况，导致测试结果不符合预期。本文将深入探讨这个问题，并提供解决方案。
理解Mockito的工作原理
Mockito通过动态代理或字节码操作来创建Mock对象。当我们使用when(...).thenReturn(...)或doReturn(...).when(...)方法时，实际上是在定义Mock对象的行为，即当调用特定方法时，返回预设的值。
重要的是理解，Mockito只能Mock方法调用，而不能直接修改对象的属性值。因此，如果代码中直接访问对象的属性，而不是通过getter方法，Mockito将无法覆盖该属性的值。
常见问题及解决方案
1. 直接访问属性而非Getter方法
如果被测试的代码直接访问对象的属性，例如userEntityTo.followedByEntity，而不是通过userEntityTo.getFollowedByEntity()方法，Mockito将无法覆盖该属性的值。
解决方案：
修改被测试的代码，使用Getter方法访问属性。
// 修改前
if (userEntityTo.followedByEntity == null || userEntityTo.followedByEntity.isEmpty()) {
throw new FollowerNotFoundException("Follower Not Found");
}
// 修改后
if (userEntityTo.getFollowedByEntity() == null || userEntityTo.getFollowedByEntity().isEmpty()) {
throw new FollowerNotFoundException("Follower Not Found");
}2. 逻辑上的问题导致Mock无效
在提供的代码示例中，followingRequestEntities在followUser方法中始终包含一个元素，因此userEntityTo.getFollowedByEntity().isEmpty()永远为false，导致异常无法抛出。
解决方案：
检查测试逻辑，确保Mock的行为能够触发预期的结果。例如，可以Mock userRepository.findById()方法，使其返回一个空的Optional对象，从而触发UserNotFoundException。
when(userRepository.findById(userEntityFrom.getId())).thenReturn(Optional.empty());
3. Mock的对象类型不匹配
确保Mock的对象类型与实际使用的对象类型一致。如果类型不匹配，Mockito可能无法正确地Mock对象的行为。
解决方案：
仔细检查Mock对象的类型，确保其与被测试代码中使用的对象类型完全一致。
4. Mockito的作用域
Mockito的Mock对象只在当前测试方法的作用域内有效。如果在不同的测试方法中使用相同的Mock对象，需要重新定义其行为。
解决方案：
在每个测试方法中，都明确地定义Mock对象的行为。可以使用@BeforeEach注解来初始化Mock对象，并在每个测试方法中根据需要修改其行为。
代码示例
以下是一个完整的代码示例，展示了如何使用Mockito进行单元测试，并解决变量值无法被覆盖的问题。
package com.example.demo;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.when;
class MyService {
private final MyRepository myRepository;
MyService(MyRepository myRepository) {
this.myRepository = myRepository;
}
public String doWork() {
return myRepository.get().toUpperCase();
}
}
class MyRepository {
public String get() {
return "a real thing from the DB";
}
}
@ExtendWith(MockitoExtension.class)
public class MyTests {
@Mock
MyRepository myRepository;
@InjectMocks
MyService myService;
@Test
public void aTest() {
when(myRepository.get()).thenReturn("something fake");
assertEquals("SOMETHING FAKE", myService.doWork());
}
}代码解释：
- @ExtendWith(MockitoExtension.class): 启用Mockito扩展，简化Mock对象的创建和注入。
- @Mock: 标记一个字段为Mock对象。
- @InjectMocks: 标记一个字段，Mockito会将Mock对象注入到该字段中。
- when(myRepository.get()).thenReturn("something fake"): 定义Mock对象的行为，当调用myRepository.get()方法时，返回"something fake"。
- assertEquals("SOMETHING FAKE", myService.doWork()): 断言测试结果是否符合预期。
注意事项和总结
- Mockito只能Mock方法调用，不能直接修改对象的属性值。
- 确保使用Getter方法访问对象的属性。
- 检查测试逻辑，确保Mock的行为能够触发预期的结果。
- 确保Mock的对象类型与实际使用的对象类型一致。
- 在每个测试方法中，都明确地定义Mock对象的行为。
通过理解Mockito的工作原理，并遵循上述建议，可以有效地解决Mockito使用中变量值无法被覆盖的问题，编写出高质量的单元测试代码。
