diff --git a/spring-boot-project/spring-boot-test/src/main/java/org/springframework/boot/test/context/SpringBootContextLoader.java b/spring-boot-project/spring-boot-test/src/main/java/org/springframework/boot/test/context/SpringBootContextLoader.java index 8b2be553c2b..b737e6e0fe4 100644 --- a/spring-boot-project/spring-boot-test/src/main/java/org/springframework/boot/test/context/SpringBootContextLoader.java +++ b/spring-boot-project/spring-boot-test/src/main/java/org/springframework/boot/test/context/SpringBootContextLoader.java @@ -152,6 +152,8 @@ public class SpringBootContextLoader extends AbstractContextLoader { } private void setActiveProfiles(ConfigurableEnvironment environment, String[] profiles) { + environment.setActiveProfiles(profiles); + // Also add as properties to override any application.properties String[] pairs = new String[profiles.length]; for (int i = 0; i < profiles.length; i++) { pairs[i] = "spring.profiles.active[" + i + "]=" + profiles[i]; diff --git a/spring-boot-project/spring-boot-test/src/test/java/org/springframework/boot/test/context/SpringBootTestWithActiveProfilesAndEnvironmentPropertyTests.java b/spring-boot-project/spring-boot-test/src/test/java/org/springframework/boot/test/context/SpringBootTestWithActiveProfilesAndEnvironmentPropertyTests.java new file mode 100644 index 00000000000..6d069ef7962 --- /dev/null +++ b/spring-boot-project/spring-boot-test/src/test/java/org/springframework/boot/test/context/SpringBootTestWithActiveProfilesAndEnvironmentPropertyTests.java @@ -0,0 +1,74 @@ +/* + * Copyright 2012-2020 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.springframework.boot.test.context; + +import java.util.LinkedHashMap; +import java.util.Map; + +import org.junit.jupiter.api.Test; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.annotation.Configuration; +import org.springframework.core.env.ConfigurableEnvironment; +import org.springframework.core.env.Environment; +import org.springframework.core.env.MapPropertySource; +import org.springframework.core.env.MutablePropertySources; +import org.springframework.test.context.ActiveProfiles; +import org.springframework.test.context.ContextConfiguration; + +import static org.assertj.core.api.Assertions.assertThat; + +/** + * Integration tests for {@link SpringBootTest @SpringBootTest} with an + * {@link ActiveProfiles @ActiveProfiles} annotation. + * + * @author Johnny Lim + * @author Phillip Webb + */ +@SpringBootTest +@ActiveProfiles({ "test1", "test2" }) +@ContextConfiguration(loader = SpringBootTestWithActiveProfilesAndEnvironmentPropertyTests.Loader.class) +public class SpringBootTestWithActiveProfilesAndEnvironmentPropertyTests { + + @Autowired + private Environment environment; + + @Test + void getActiveProfiles() { + assertThat(this.environment.getActiveProfiles()).containsOnly("test1", "test2"); + } + + @Configuration + static class Config { + + } + + static class Loader extends SpringBootContextLoader { + + @Override + protected ConfigurableEnvironment getEnvironment() { + ConfigurableEnvironment environment = super.getEnvironment(); + MutablePropertySources sources = environment.getPropertySources(); + Map map = new LinkedHashMap<>(); + map.put("spring.profiles.active", "local"); + sources.addLast(new MapPropertySource("profiletest", map)); + return environment; + } + + } + +} diff --git a/spring-boot-project/spring-boot-test/src/test/java/org/springframework/boot/test/context/SpringBootTestWithActiveProfilesAndSytemEnvironmentPropertyTests.java b/spring-boot-project/spring-boot-test/src/test/java/org/springframework/boot/test/context/SpringBootTestWithActiveProfilesAndSytemEnvironmentPropertyTests.java new file mode 100644 index 00000000000..fabc8db33cc --- /dev/null +++ b/spring-boot-project/spring-boot-test/src/test/java/org/springframework/boot/test/context/SpringBootTestWithActiveProfilesAndSytemEnvironmentPropertyTests.java @@ -0,0 +1,79 @@ +/* + * Copyright 2012-2020 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.springframework.boot.test.context; + +import java.util.LinkedHashMap; +import java.util.Map; + +import org.junit.jupiter.api.Test; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.annotation.Configuration; +import org.springframework.core.env.ConfigurableEnvironment; +import org.springframework.core.env.Environment; +import org.springframework.core.env.MapPropertySource; +import org.springframework.core.env.MutablePropertySources; +import org.springframework.core.env.PropertySource; +import org.springframework.core.env.StandardEnvironment; +import org.springframework.test.context.ActiveProfiles; +import org.springframework.test.context.ContextConfiguration; + +import static org.assertj.core.api.Assertions.assertThat; + +/** + * Integration tests for {@link SpringBootTest @SpringBootTest} with an + * {@link ActiveProfiles @ActiveProfiles} annotation. + * + * @author Johnny Lim + * @author Phillip Webb + */ +@SpringBootTest +@ActiveProfiles({ "test1", "test2" }) +@ContextConfiguration(loader = SpringBootTestWithActiveProfilesAndSytemEnvironmentPropertyTests.Loader.class) +public class SpringBootTestWithActiveProfilesAndSytemEnvironmentPropertyTests { + + @Autowired + private Environment environment; + + @Test + void getActiveProfiles() { + assertThat(this.environment.getActiveProfiles()).containsOnly("test1", "test2"); + } + + @Configuration + static class Config { + + } + + static class Loader extends SpringBootContextLoader { + + @Override + @SuppressWarnings("unchecked") + protected ConfigurableEnvironment getEnvironment() { + ConfigurableEnvironment environment = super.getEnvironment(); + MutablePropertySources sources = environment.getPropertySources(); + PropertySource source = sources.get(StandardEnvironment.SYSTEM_ENVIRONMENT_PROPERTY_SOURCE_NAME); + Map map = new LinkedHashMap<>((Map) source.getSource()); + map.put("SPRING_PROFILES_ACTIVE", "local"); + sources.replace(StandardEnvironment.SYSTEM_ENVIRONMENT_PROPERTY_SOURCE_NAME, + new MapPropertySource(StandardEnvironment.SYSTEM_ENVIRONMENT_PROPERTY_SOURCE_NAME, map)); + return environment; + } + + } + +}