如何在SpringBoot测试中仅加载指定的Repository？

1. 为什么直接指定Repository会报错 #

Spring Data JPA的Repository是动态代理生成的实现类，不是普通的Java类，直接用@ContextConfiguration指定Repository接口，Spring无法自动创建对应的Bean，必须通过Spring Data的专属配置触发代理生成。

2. 配置最小化Spring上下文 #

创建专门的测试配置类，只加载必要组件：

@Configuration
@EnableJpaRepositories(basePackageClasses = TitleRepository.class) // 仅扫描目标Repository所在包
@EntityScan(basePackageClasses = Title.class) // 扫描对应实体类
@Import(TitlesSeeder.class) // 导入Seeder类
public class TestSeedConfig {
    // 配置嵌入式数据库（测试用）
    @Bean
    public DataSource dataSource() {
        return new EmbeddedDatabaseBuilder()
                .setType(EmbeddedDatabaseType.H2)
                .build();
    }

    // 配置JPA实体管理器工厂
    @Bean
    public LocalContainerEntityManagerFactoryBean entityManagerFactory(DataSource dataSource) {
        LocalContainerEntityManagerFactoryBean em = new LocalContainerEntityManagerFactoryBean();
        em.setDataSource(dataSource);
        em.setPackagesToScan(Title.class.getPackageName());
        em.setJpaVendorAdapter(new HibernateJpaVendorAdapter());
        return em;
    }

    // 配置事务管理器
    @Bean
    public PlatformTransactionManager transactionManager(EntityManagerFactory emf) {
        return new JpaTransactionManager(emf);
    }
}

3. 实现测试前一次性执行Seeder #

方式一：在测试类中用@BeforeAll手动触发 #

@ContextConfiguration(classes = TestSeedConfig.class)
@Transactional
public class TitleServiceTest {

    @Autowired
    private TitlesSeeder titlesSeeder;

    @BeforeAll
    static void setup(@Autowired TitlesSeeder seeder) {
        seeder.seed(); // 所有测试执行前仅运行一次
    }

    // 测试方法示例
    @Test
    void testFindAllTitles() {
        List<Title> titles = titleRepository.findAll();
        assertFalse(titles.isEmpty());
    }
}

方式二：让Seeder自动随上下文启动执行 #

修改Seeder类实现CommandLineRunner，上下文启动时自动执行填充逻辑：

@Component
public class TitlesSeeder implements CommandLineRunner {

    private final TitleRepository titleRepository;

    public TitlesSeeder(TitleRepository titleRepository) {
        this.titleRepository = titleRepository;
    }

    @Override
    public void run(String... args) throws Exception {
        // 加判断避免重复填充数据
        if (titleRepository.count() == 0) {
            titleRepository.saveAll(List.of(
                    new Title("Sample Title 1"),
                    new Title("Sample Title 2")
            ));
        }
    }
}

这种方式无需在测试类中手动调用，上下文启动后Seeder自动执行一次。

4. 简化方案：用@DataJpaTest快速配置 #

如果仅测试JPA相关逻辑，用@DataJpaTest可以自动配置嵌入式数据库、JPA组件，且只加载最小上下文：

@DataJpaTest
@Import(TitlesSeeder.class)
public class TitleRepositoryTest {

    @Autowired
    private TitleRepository titleRepository;

    @BeforeAll
    static void setup(@Autowired TitlesSeeder seeder) {
        seeder.seed();
    }

    // 测试方法...
}

关键注意事项 #

• Seeder类需通过@Component或@Import纳入Spring上下文管理，否则无法被注入。
• 用@Transactional注解测试类，测试后会自动回滚数据，避免测试间数据污染。
• 绝对不要直接在@ContextConfiguration中指定Repository接口，必须通过@EnableJpaRepositories扫描生成代理Bean。

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