Add capability to configure the Spring Boot, SpringApplicationBuilder.contextClass(..).

This commit is contained in:
John Blum
2018-10-05 17:53:00 -07:00
parent 7c99575ed4
commit e372410f29

View File

@@ -51,11 +51,16 @@ public abstract class SpringBootApplicationIntegrationTestsSupport
@Override
protected ConfigurableApplicationContext newApplicationContext(Class<?>... annotatedClasses) {
return setApplicationContext(processBeforeRun(processBeforeBuild(new SpringApplicationBuilder(annotatedClasses)
.initializers(this::processBeforeRefresh)
.web(getWebApplicationType()))
.build())
.run(getArguments()));
return setApplicationContext(processBeforeRun(processBeforeBuild(contextClass(
new SpringApplicationBuilder(annotatedClasses)
.initializers(this::processBeforeRefresh)
.web(getWebApplicationType())))
.build())
.run(getArguments()));
}
protected SpringApplicationBuilder contextClass(SpringApplicationBuilder builder) {
return builder;
}
protected SpringApplicationBuilder processBeforeBuild(SpringApplicationBuilder springApplicationBuilder) {