在应用程序启动之前配置@MockBean组件

Question

我有一个Spring Boot 1.4.2应用程序。启动期间使用的一些代码如下所示：

@Component 
class SystemTypeDetector{
    public enum SystemType{ TYPE_A, TYPE_B, TYPE_C }
    public SystemType getSystemType(){ return ... }
}

@Component 
public class SomeOtherComponent{
    @Autowired 
    private SystemTypeDetector systemTypeDetector;
    @PostConstruct 
    public void startup(){
        switch(systemTypeDetector.getSystemType()){   // <-- NPE here in test
        case TYPE_A: ...
        case TYPE_B: ...
        case TYPE_C: ...
        }
    }
}

有一个组件决定系统类型。此组件在从其他组件启动期间使用。在生产中，一切正常。

现在，我想使用Spring1.4的MockBean添加一些集成测试。

测试如下所示：

@RunWith(SpringRunner.class)
@SpringBootTest(classes = MyWebApplication.class, webEnvironment = RANDOM_PORT)
public class IntegrationTestNrOne {
    @MockBean 
    private SystemTypeDetector systemTypeDetectorMock;

    @Before 
    public void initMock(){
       Mockito.when(systemTypeDetectorMock.getSystemType()).thenReturn(TYPE_C);
    }

    @Test 
    public void testNrOne(){
      // ...
    }
}

基本上，模拟效果很好。使用我的systemTypeDetectorMock，如果我调用getSystemType-

问题是应用程序没有启动。目前Spring的工作顺序似乎是：

1. 创建所有模拟（无配置，所有方法返回null）
2. 启动应用程序
3. 调用@Before方法（在此处配置模拟）
4. 启动测试

我的问题是，应用程序以未初始化的模拟开始。因此，对getSystemType（）的调用返回null。

我的问题是：如何在应用程序启动之前配置模拟？

编辑：如果有人有同样的问题，一个解决方法是使用@MockBean（answer=CALLS\u REAL\u METHODS）。这将调用真正的组件，在我的情况下，系统将启动。启动后，我可以更改模拟行为。

Answer 1

您可以使用以下技巧：

@Configuration
public class Config {

    @Bean
    public BeanA beanA() {
        return new BeanA();
    }

    @Bean
    public BeanB beanB() {
        return new BeanB(beanA());
    }
}

@RunWith(SpringRunner.class)
@ContextConfiguration(classes = {TestConfig.class, Config.class})
public class ConfigTest {

    @Configuration
    static class TestConfig {

        @MockBean
        BeanA beanA;

        @PostConstruct
        void setUp() {
            when(beanA.someMethod()).thenReturn(...);
        }
    }
}

至少它适用于spring-boot-2.1.9。发布

Answer 2

我能这样修好它

@RunWith(SpringRunner.class)
@SpringBootTest(classes = MyWebApplication.class, webEnvironment = RANDOM_PORT)
public class IntegrationTestNrOne {
    // this inner class must be static!
    @TestConfiguration
    public static class EarlyConfiguration {
       @MockBean 
       private SystemTypeDetector systemTypeDetectorMock;

       @PostConstruct 
       public void initMock(){
          Mockito.when(systemTypeDetectorMock.getSystemType()).thenReturn(TYPE_C);
       }
    }

    // here we can inject the bean created by EarlyConfiguration
    @Autowired 
    private SystemTypeDetector systemTypeDetectorMock;

    @Autowired
    private SomeOtherComponent someOtherComponent;

    @Test 
    public void testNrOne(){
       someOtherComponent.doStuff();
    }
}

Answer 3

在这种情况下，您需要以引入MockBean之前的方式配置Mock，即手动指定一个将在上下文中替换原始bean的主bean。

@SpringBootTest
class DemoApplicationTests {

    @TestConfiguration
    public static class TestConfig {

        @Bean
        @Primary
        public SystemTypeDetector mockSystemTypeDetector() {
            SystemTypeDetector std = mock(SystemTypeDetector.class);
            when(std.getSystemType()).thenReturn(TYPE_C);
            return std;
        }

    }

    @Autowired
    private SystemTypeDetector systemTypeDetector;

    @Test
    void contextLoads() {
        assertThat(systemTypeDetector.getSystemType()).isEqualTo(TYPE_C);
    }
}

由于TestConfiguration类是一个静态的内部类，因此只有通过此测试才能自动选择它。您将在之前放入的完整模拟行为必须移动到初始化bean的方法。