root
/
spring-boot
mirror of https://github.com/spring-projects/spring-boot.git
1
0
Fork 0
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Now custom SpringApplication class can be used when extending SpringApplicationContextLoader

This commit is contained in:
Jakub Kubrynski 2014-02-11 15:44:20 +01:00 committed by Dave Syer
parent 76c56c6aa9
commit 6f2e133a0a
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
spring-boot/src/main/java/org/springframework/boot/test/SpringApplicationContextLoader.java
View File

@ -58,7 +58,7 @@ public class SpringApplicationContextLoader extends AbstractContextLoader {
public ApplicationContext loadContext(MergedContextConfiguration mergedConfig) public ApplicationContext loadContext(MergedContextConfiguration mergedConfig)
throws Exception { throws Exception {
SpringApplication application = new SpringApplication(); SpringApplication application = getSpringApplication();
application.setSources(getSources(mergedConfig)); application.setSources(getSources(mergedConfig));
if (!ObjectUtils.isEmpty(mergedConfig.getActiveProfiles())) { if (!ObjectUtils.isEmpty(mergedConfig.getActiveProfiles())) {
application.setAdditionalProfiles(mergedConfig.getActiveProfiles()); application.setAdditionalProfiles(mergedConfig.getActiveProfiles());
@ -77,6 +77,15 @@ public class SpringApplicationContextLoader extends AbstractContextLoader {
return application.run(); return application.run();
} }
/**
* Builds new {@link org.springframework.boot.SpringApplication} instance. You can override
* this method to add custom behaviour
* @return {@link org.springframework.boot.SpringApplication} instance
*/
protected SpringApplication getSpringApplication() {
return new SpringApplication();
}
private Set<Object> getSources(MergedContextConfiguration mergedConfig) { private Set<Object> getSources(MergedContextConfiguration mergedConfig) {
Set<Object> sources = new LinkedHashSet<Object>(); Set<Object> sources = new LinkedHashSet<Object>();
sources.addAll(Arrays.asList(mergedConfig.getClasses())); sources.addAll(Arrays.asList(mergedConfig.getClasses()));