如何在JUnit中自动装配Service的存储过程依赖，无需手动构建？

核心问题分析 #

测试中存储过程对象为NULL的原因是：StoredProc1/2/3类未被Spring容器识别为Bean，或者缺少可注入的DataSource导致无法实例化，因此Spring无法自动装配到MyService中。

具体实现步骤 #

1. 将存储过程类标记为Spring Bean #

给所有继承自StoredProcedure的类添加@Component（或@Service）注解，确保Spring扫描时能将它们注册为容器Bean：

@Component
public class StoredProc1 extends StoredProcedure implements StoredProc {
    public StoredProc1(@Autowired DataSource dataSource) {
        super(dataSource, "实际存储过程名称");
        // 初始化存储过程参数等逻辑
    }

    // 实现StoredProc接口的方法
    @Override
    public void someAction() {
        // ...
    }
}

同理给StoredProc2、StoredProc3添加相同注解。

2. 配置测试用DataSource #

使用内存数据库（如H2）提供测试可用的DataSource，避免依赖真实数据库：

• 添加依赖（以Maven为例）：

  <dependency>
      <groupId>com.h2database</groupId>
      <artifactId>h2</artifactId>
      <scope>test</scope>
  </dependency>
  

• 配置测试环境：在src/test/resources/application-test.properties中添加H2配置：

  spring.datasource.url=jdbc:h2:mem:testdb;DB_CLOSE_DELAY=-1
  spring.datasource.driver-class-name=org.h2.Driver
  spring.datasource.username=sa
  spring.datasource.password=
  

3. 调整测试类确保Bean被扫描 #

如果存储过程类和MyService不在同一个包下，需要在测试类上指定扫描包路径；如果在同一包下，直接使用@SpringBootTest即可：

@SpringBootTest
// 若SP类在其他包，添加@ComponentScan指定包，比如：
// @ComponentScan(basePackages = {"com.your.service", "com.your.storedprocs"})
public class MyServiceTest {
    @Autowired
    MyService myService;

    @Autowired
    StoredProc1 storedProc1;
    @Autowired
    StoredProc2 storedProc2;
    @Autowired
    StoredProc3 storedProc3;

    @Test
    public void testGetStoredProc() {
        // 构造测试参数，触发对应SP的返回逻辑
        Params testParams = new Params();
        testParams.setType("TYPE1"); // 假设此参数会返回storedProc1

        StoredProc result = myService.getStoredProc(testParams);

        // 验证返回的是预期的SP对象
        assertSame(storedProc1, result);
    }
}

备选方案：手动构建测试上下文（轻量版） #

如果不想启动完整的Spring Boot上下文，可以创建测试配置类手动注册Bean：

@Configuration
public class TestSPConfig {
    // 注册内存DataSource
    @Bean
    public DataSource testDataSource() {
        return new EmbeddedDatabaseBuilder()
                .setType(EmbeddedDatabaseType.H2)
                .build();
    }

    // 注册各个存储过程Bean
    @Bean
    public StoredProc1 storedProc1(DataSource dataSource) {
        return new StoredProc1(dataSource);
    }

    @Bean
    public StoredProc2 storedProc2(DataSource dataSource) {
        return new StoredProc2(dataSource);
    }

    @Bean
    public StoredProc3 storedProc3(DataSource dataSource) {
        return new StoredProc3(dataSource);
    }

    // 注册MyService并注入依赖
    @Bean
    public MyService myService(StoredProc1 sp1, StoredProc2 sp2, StoredProc3 sp3) {
        MyService service = new MyService();
        // 字段注入时用ReflectionTestUtils设置字段值
        ReflectionTestUtils.setField(service, "storedProc1", sp1);
        ReflectionTestUtils.setField(service, "storedProc2", sp2);
        ReflectionTestUtils.setField(service, "storedProc3", sp3);
        return service;
    }
}

然后测试类加载该配置：

@ExtendWith(SpringExtension.class)
@ContextConfiguration(classes = TestSPConfig.class)
public class MyServiceTest {
    @Autowired
    MyService myService;

    @Autowired
    StoredProc1 storedProc1;

    @Test
    public void testGetStoredProc() {
        // 测试逻辑同上
    }
}

内容的提问来源于stack exchange，提问作者gene b.