How to override Spring Bean in integration test with custom bean definition?

Question

I would like to reuse Spring production context configuration, but replace a few beans with another ones. If I would like to override them with a mock, I would use @MockBean, which does exactly what I need (overrides bean), but does not allow me to configure a new bean myselves.

I know there is another way to use @ContextConfiguration but it seems too verbose to me.

Thanks.

Answer 1

You can use @SpyBean - then bean can be stubbed for specific cases (like in the case of @MockBean), but otherwise real bean will be used.

Also, if you actually need to define custom bean definition for tests, then combination of @Primary / @Profile / @ContextConfiguration can be used for this purpose.

For example:

@RunWith(SpringRunner.class)
@SpringBootTest
@ActiveProfiles("test")
@ContextConfiguration(classes = {TestConfig.class, ApplicationConfig.class})
public class ApplicatonTest {
    @Profile("test")
    @Configuration
    static class TestConfig {

        @Bean
        @Primary
        public SomeBean testBeanDefinition() {
            SomeBean testBean = new SomeBean();
            // configure SomeBean for test
            return testBean;
        }
    }
    // tests
}