SpringBootTest - 如何在运行时配置中替换一个 bean？

Question

我正在为 Spring Boot 应用程序编写集成测试。只要我使用 100% 运行时配置进行测试，一切都会顺利进行。但是当我试图为 bean 提供一个自定义 bean 时，一切都会中断。

@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class CombinedControllerIntegrationTest2 {

@TestConfiguration
static class ContextConfiguration {

    @Bean
    @Primary
    public SolrDocumentTypeMapRepository solrDocumentTypeMapRepository() {
        LOG.debug("SolrDocumentTypeMapRepository is being initialized.");

// etc.

上面的代码变体导致加载真实的运行时 SolrDocumentTypeMapRepository。我的测试类中的 ContextConfiguration 被忽略了。

如果我尝试在我的内部 ContextConfiguration 上使用 @Configuration 而不是 @TestConfiguration，则执行会落到另一个极端 - 它以

结束

org.springframework.context.ApplicationContextException：无法 由于缺少而启动 EmbeddedWebApplicationContext EmbeddedServletContainerFactory bean。

因为显然没有加载其余配置。

如果我尝试放

@ContextConfiguration(类 = {CombinedControllerIntegrationTest2.ContextConfiguration.class,GatewayApplication.class})

在我的主要测试类中，它以与 #1 相同的方式失败 - 即我的 ContextConfiguration 被忽略。

有什么想法吗？

附：我知道我可以使用@MockBean（这甚至在其他情况下也有效），但是在这里，因为在某些时候我依赖于主代码中的@PostConsruct 方法，@MockBeans 没用。

Answer 1

使用@Mockean 模拟您的存储库。并在测试中添加行为：

@SpringBootTest
@AutoConfigureMockMvc
@RunWith(SpringRunner.class)      
public abstract class IntegrationTest {

  @MockBean
  SolrDocumentTypeMapRepository solrDocumentTypeMapRepository;

  @Test
  public void mySuperTest(){ 
Mockito.when(solrDocumentTypeMapRepository.getById(Mockito.any())).thenReturn(someInstance);
Assert.assertEquals(solrDocumentTypeMapRepository.getById("someId"), someInstance);
}
}

Answer 2

使用一个 Bean 进行单元测试

只需使用@RunWith(SpringRunner.class) 注释，它应该可以工作。您也可以使用@RunWith(SpringJUnit4ClassRunner.class)。两者都应该工作。

请不要使用@SpringBootTest注解。它将连接整个应用程序。

这是您更新的示例，

@RunWith(SpringRunner.class)
public class CombinedControllerIntegrationTest2 {

    @TestConfiguration
    static class ContextConfiguration {

        @Bean
        public SolrDocumentTypeMapRepository solrDocumentTypeMapRepository() {
            LOG.debug("SolrDocumentTypeMapRepository is being initialized.");
            return new SolrDocumentTypeMapRepository(...);
        }
    }

    @Autowired
    private SolrDocumentTypeMapRepository repository;

    @Test
    public void test() {
        assertNotNull(repository);
    }
}

使用替换的 Bean 进行集成测试

• 创建一个新的测试 Spring Boot 应用程序。它应该排除负责创建SolrDocumentTypeMapRepository bean 的配置类（例如SolrConfiguration）。

  @SpringBootApplication
  @ComponentScan(basePackages = {
          "com.abc.pkg1",
          "com.abc.pk2"},
          excludeFilters = {
                  @ComponentScan.Filter(type = FilterType.ASSIGNABLE_TYPE, 
                  value = SolrConfiguration.class)})
  public class TestApplication {
      public static void main(String[] args) throws Exception {
          SpringApplication.run(TestApplication.class, args);
      }
  }
  

• 现在，在您的测试类中使用@ContextConfiguration 注解来添加TestApplication.class 和ContextConfiguration.class。这会将您的应用程序与所有必需的 bean 连接起来，包括被替换的 bean。下图是更新后的测试类，

  @ActiveProfiles("test")
  @RunWith(SpringJUnit4ClassRunner.class)
  @SpringBootTest(webEnvironment = 
      SpringBootTest.WebEnvironment.RANDOM_PORT)
  @ContextConfiguration(classes = {TestApplication.class, 
      CombinedControllerIntegrationTest2.ContextConfiguration.class})
  public class CombinedControllerIntegrationTest2 {
  
      @TestConfiguration
      static class ContextConfiguration {
  
          @Bean
          public SolrDocumentTypeMapRepository solrDocumentTypeMapRepository() {
              LOG.debug("SolrDocumentTypeMapRepository is being initialized.");
              return new SolrDocumentTypeMapRepository(...);
          }
      }
  
      ...
  }
  

Answer 3

不要使用@Runwith(SpringJunit4Classrunner.class)，而是使用@RunWith(SpringRunner.class)。这应该可以帮助您解决错误。