单元测试：Mockito框架下Firebase Mock的结构设计选择

嘿，这个问题是Mockito单元测试里非常典型的实践抉择，我结合你测试DataRepository的场景来给你掰扯清楚～

首先明确核心原则：单元测试的独立性是第一要务——每个测试用例应该只关注自己的场景，不会因为其他测试的Mock设置而受影响。基于这个原则，我们来对比两种方案：

方案1：在@Before中统一设置所有Mock行为 #

这种做法适合当绝大多数测试用例需要的Mock行为完全一致时，比如你的FirebaseFirestore实例在80%的测试里都返回相同的基础数据。好处是减少重复代码，但缺点也很明显：

• 如果后续某个测试需要修改Mock行为，你要么在测试用例里覆盖（可能导致逻辑混乱），要么修改@Before里的代码，这会影响所有依赖这个设置的测试，增加维护风险。
• 测试用例之间会产生隐式依赖，排查问题时很难定位是哪个设置出了问题。

方案2：每个测试用例单独设置Mock行为 #

这是更推荐的做法，完全符合单元测试的独立性要求。每个测试只针对特定场景，给Mock设置对应行为，逻辑清晰，出问题也好定位。

结合你的DataRepository.getPosts()场景，我给你写个具体示例：

第一步：声明Mock和待测试类 #

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.junit.jupiter.api.extension.ExtendWith;
import io.reactivex.Maybe;
import io.reactivex.Single;
import java.util.Arrays;
import java.util.Collections;

@ExtendWith(MockitoExtension.class) // JUnit 5的Mockito扩展，自动初始化Mock
public class DataRepositoryTest {

    @Mock
    private FirebaseFirestore firestore; // 注入的依赖Mock

    private DataRepository dataRepository; // 待测试类

第二步：@BeforeEach仅做初始化，不设置行为 #

@BeforeEach
    void setUp() {
        dataRepository = new DataRepository(firestore); // 初始化待测试类，传入Mock
        // 这里不设置任何Mock的具体行为，留到每个测试用例里按需定义
    }

第三步：每个测试用例单独设置Mock行为 #

比如测试「成功返回帖子列表」的场景：

@Test
    void getPosts_WhenPostExists_ReturnsValidPostList() {
        // 1. 准备测试数据
        List<Post> mockPosts = Arrays.asList(
            new Post("post1", "第一篇测试帖"),
            new Post("post2", "第二篇测试帖")
        );

        // 2. 设置FirebaseFirestore的Mock行为：模拟查询成功返回包含帖子的快照
        DocumentSnapshot mockSnapshot = Mockito.mock(DocumentSnapshot.class);
        when(mockSnapshot.toObject(Post.class)).thenReturn(mockPosts.get(0)); // 这里根据你的实际实现调整
        when(firestore.collection("posts").document("valid-post-id").get())
            .thenReturn(Single.just(mockSnapshot));

        // 3. 执行测试方法
        Maybe<List<Post>> result = dataRepository.getPosts("valid-post-id");

        // 4. 断言结果符合预期
        result.test().assertValue(mockPosts);
    }

再比如测试「无帖子返回空列表」的场景：

@Test
    void getPosts_WhenNoPostsExist_ReturnsEmptyList() {
        // 设置Mock行为：模拟查询返回空快照
        DocumentSnapshot emptySnapshot = Mockito.mock(DocumentSnapshot.class);
        when(emptySnapshot.exists()).thenReturn(false);
        when(firestore.collection("posts").document("empty-post-id").get())
            .thenReturn(Single.just(emptySnapshot));

        Maybe<List<Post>> result = dataRepository.getPosts("empty-post-id");
        result.test().assertValue(Collections.emptyList());
    }

还有「Firebase调用失败返回错误」的场景：

@Test
    void getPosts_WhenFirebaseCallFails_EmitsError() {
        // 设置Mock行为：模拟查询抛出异常
        FirebaseFirestoreException mockError = new FirebaseFirestoreException(
            "连接失败", FirebaseFirestoreException.Code.UNAVAILABLE
        );
        when(firestore.collection("posts").document("error-post-id").get())
            .thenReturn(Single.error(mockError));

        Maybe<List<Post>> result = dataRepository.getPosts("error-post-id");
        result.test().assertError(FirebaseFirestoreException.class);
    }
}

总结最佳实践 #

1. 优先选择每个测试用例单独设置Mock行为：保证测试独立性，降低维护成本，逻辑更清晰。
2. 共享行为谨慎使用：如果确实有多个测试需要完全一致的Mock行为，可以提取到@BeforeEach里，但一定要确保这些行为不会影响测试的独立性，必要时可以在测试用例开头用reset(firestore)重置Mock状态。
3. 利用Mockito扩展简化代码：比如JUnit 5的MockitoExtension可以自动初始化Mock，不用手动调用MockitoAnnotations.openMocks(this)。

内容的提问来源于stack exchange，提问作者Skepsis Dev