Spring Boot单元测试：Mock数据库组件解决服务测试启动失败问题

方案1：纯单元测试（不启动Spring上下文） #

这是最轻量化的方案，完全不需要Spring参与，直接用Mockito模拟所有依赖的Repository，专注测试Service的业务逻辑，没有任何数据库相关的初始化开销。

步骤：

• 使用@ExtendWith(MockitoExtension.class)启用Mockito的测试扩展
• 用@Mock注解模拟所有JpaRepository实例
• 用@InjectMocks将模拟的Repository注入到Service A中

示例代码：

@ExtendWith(MockitoExtension.class)
class ServiceATest {
    // 模拟所有依赖的Repository
    @Mock
    private UserRepository userRepository;
    @Mock
    private OrderRepository orderRepository;
    
    // 把模拟的Repository注入到Service A
    @InjectMocks
    private ServiceA serviceA;

    @Test
    void testDoBusinessLogic() {
        // 预设Repository的返回值
        User mockUser = new User(1L, "张三");
        when(userRepository.findById(1L)).thenReturn(Optional.of(mockUser));
        
        // 执行Service方法
        String result = serviceA.getUserNameById(1L);
        
        // 断言结果正确
        assertEquals("张三", result);
        // 验证Repository方法被正确调用
        verify(userRepository).findById(1L);
    }
}

方案2：限制Spring上下文加载范围（需Spring环境时用） #

如果Service A依赖其他Spring Bean（比如配置类、工具类Bean），必须启动部分Spring上下文，那可以通过排除JPA相关自动配置，避免数据库连接初始化，同时用@MockBean替换真实的Repository。

步骤：

• 用@SpringBootTest指定仅加载Service A和必要的依赖Bean
• 用@EnableAutoConfiguration排除DataSource、JPA相关的自动配置类
• 用@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE)禁用测试数据库自动配置

示例代码：

@SpringBootTest(classes = {ServiceA.class, BusinessToolBean.class})
@EnableAutoConfiguration(exclude = {
    DataSourceAutoConfiguration.class,
    HibernateJpaAutoConfiguration.class,
    DataSourceTransactionManagerAutoConfiguration.class
})
@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE)
class ServiceASpringContextTest {
    // 用MockBean替换真实的Repository
    @MockBean
    private UserRepository userRepository;
    @MockBean
    private OrderRepository orderRepository;
    
    @Autowired
    private ServiceA serviceA;

    @Test
    void testServiceWithSpringBean() {
        // 预设Repository行为
        when(orderRepository.countByUserId(1L)).thenReturn(5L);
        
        // 执行Service方法
        long orderCount = serviceA.getUserOrderCount(1L);
        
        // 断言结果
        assertEquals(5L, orderCount);
    }
}

为什么之前的方案失效？ #

• 直接用@SpringBootTest未排除JPA自动配置：Spring会默认尝试初始化DataSource和EntityManagerFactory，导致需要数据库连接
• Mock DataSource失败：其他Bean依赖DataSource做初始化，Mock后的DataSource无法提供有效连接，引发空指针或初始化失败
• H2方案冗余：既然已经Mock了所有Repository，数据库完全没用，反而增加启动开销

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