在Java中使用工厂方法创建通用Mockito存根的实践问询

我在项目开发中踩过不少重复Mock的坑：团队里有个被所有Web服务依赖的通用辅助工具类，每次测试这些服务时，都要写一堆重复的mock()和when()代码来模拟它的行为——不仅繁琐，还容易因为不同测试里Mock逻辑不一致，导致奇怪的测试失败问题。后来我尝试用测试专用工厂类替代重复Mock，效率提升简直立竿见影！

为什么要这么做？ #

想象一下，你的项目里有个CommonHelper类，负责提供获取当前用户ID、系统时间这类核心通用能力，所有Web服务的业务逻辑都依赖它。原来的测试代码可能是这样的：

@Test
public void testUserProfileService() {
    // 每次测试都要Mock这个类
    CommonHelper mockHelper = Mockito.mock(CommonHelper.class);
    // 还要逐个方法指定返回值
    when(mockHelper.getCurrentUserId()).thenReturn("test-user-001");
    when(mockHelper.getCurrentTime()).thenReturn(LocalDateTime.of(2024, 6, 1, 14, 30));
    
    UserProfileService service = new UserProfileService(mockHelper);
    // 执行测试断言...
}

如果有10个、20个服务都依赖这个类，你就要重复写10次、20次几乎一样的Mock代码——既浪费时间，又很难保证所有测试的Mock行为统一。

解决方案：测试专用工厂类 #

我们可以创建一个完全符合原工具类接口的测试工厂，它提供预设的、符合测试需求的实现，测试时直接从工厂获取实例，彻底告别重复的Mock操作。

步骤1：定义测试工厂类 #

假设原辅助工具类是接口CommonHelper，我们的工厂类可以这么写：

public class TestCommonHelperFactory {
    // 返回默认测试用实例
    public static CommonHelper createDefaultHelper() {
        return new CommonHelper() {
            @Override
            public String getCurrentUserId() {
                return "default-test-user"; // 固定测试用户ID
            }

            @Override
            public LocalDateTime getCurrentTime() {
                // 返回固定测试时间，避免时间变动影响测试结果
                return LocalDateTime.of(2024, 6, 1, 12, 0);
            }
        };
    }

    // 定制化方法：返回指定用户ID的实例
    public static CommonHelper createHelperWithUserId(String userId) {
        CommonHelper defaultHelper = createDefaultHelper();
        return new CommonHelper() {
            @Override
            public String getCurrentUserId() {
                return userId;
            }

            @Override
            public LocalDateTime getCurrentTime() {
                return defaultHelper.getCurrentTime();
            }
        };
    }

    // 定制化方法：返回指定时间的实例
    public static CommonHelper createHelperWithCustomTime(LocalDateTime customTime) {
        CommonHelper defaultHelper = createDefaultHelper();
        return new CommonHelper() {
            @Override
            public String getCurrentUserId() {
                return defaultHelper.getCurrentUserId();
            }

            @Override
            public LocalDateTime getCurrentTime() {
                return customTime;
            }
        };
    }
}

步骤2：重构测试代码 #

现在测试代码变得异常简洁，再也不用写Mock和when了：

@Test
public void testUserProfileService() {
    // 直接从工厂获取默认测试实例
    CommonHelper testHelper = TestCommonHelperFactory.createDefaultHelper();
    UserProfileService service = new UserProfileService(testHelper);
    
    // 执行测试逻辑和断言...
}

@Test
public void testAdminService() {
    // 获取定制化实例，测试管理员场景
    CommonHelper adminHelper = TestCommonHelperFactory.createHelperWithUserId("admin-001");
    AdminService service = new AdminService(adminHelper);
    
    // 执行测试逻辑和断言...
}

这么做的核心好处 #

• 减少重复代码：不用在每个测试类里重复写Mock初始化和行为指定的代码
• 统一测试依赖：所有测试用的辅助工具类实现统一，避免不同测试Mock行为不一致导致的测试不稳定
• 定制化灵活：工厂类可以提供多种预设实例，轻松满足不同测试场景的需求
• 可读性更高：测试代码直接表达“我需要一个默认测试辅助类”，而非一堆Mock细节，逻辑更清晰

内容的提问来源于stack exchange，提问作者Andrei Zhaleznichenka