Spring/spring-boot
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Provide a way to create custom ApplicationContextFactory

Browse Source 
Update `SpringBootContextLoader` so that `getApplicationContextFactory`
is now a protected that may be overridden to provide a custom
`ApplicationContextFactory` instance.

Closes gh-38205
This commit is contained in:
Phillip Webb
2023-11-07 11:53:10 -08:00
parent f22c1ba7d6
commit 690cfa220a
2 changed files with 48 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
spring-boot-project/spring-boot-test/src/main/java/org/springframework/boot/test/context/SpringBootContextLoader.java
View File

@@ -196,8 +196,7 @@ public class SpringBootContextLoader extends AbstractContextLoader implements Ao
else {
application.setWebApplicationType(WebApplicationType.NONE);
}
application.setApplicationContextFactory(
(webApplicationType) -> getApplicationContextFactory(mergedConfig, webApplicationType));
application.setApplicationContextFactory(getApplicationContextFactory(mergedConfig));
if (mergedConfig.getParent() != null) {
application.setBannerMode(Banner.Mode.OFF);
}
@@ -212,17 +211,26 @@ public class SpringBootContextLoader extends AbstractContextLoader implements Ao
}
}
private ConfigurableApplicationContext getApplicationContextFactory(MergedContextConfiguration mergedConfig,
WebApplicationType webApplicationType) {
if (webApplicationType != WebApplicationType.NONE && !isEmbeddedWebEnvironment(mergedConfig)) {
if (webApplicationType == WebApplicationType.REACTIVE) {
return new GenericReactiveWebApplicationContext();
/**
* Return the {@link ApplicationContextFactory} that should be used for the test. By
* default this method will return a factory that will create an appropriate
* {@link ApplicationContext} for the {@link WebApplicationType}.
* @param mergedConfig the merged context configuration
* @return the application context factory to use
* @since 3.2.0
*/
protected ApplicationContextFactory getApplicationContextFactory(MergedContextConfiguration mergedConfig) {
return (webApplicationType) -> {
if (webApplicationType != WebApplicationType.NONE && !isEmbeddedWebEnvironment(mergedConfig)) {
if (webApplicationType == WebApplicationType.REACTIVE) {
return new GenericReactiveWebApplicationContext();
}
if (webApplicationType == WebApplicationType.SERVLET) {
return new GenericWebApplicationContext();
}
}
if (webApplicationType == WebApplicationType.SERVLET) {
return new GenericWebApplicationContext();
}
}
return ApplicationContextFactory.DEFAULT.create(webApplicationType);
return ApplicationContextFactory.DEFAULT.create(webApplicationType);
};
}
private void prepareEnvironment(MergedContextConfiguration mergedConfig, SpringApplication application,