12 changed files with 201 additions and 3 deletions
@ -0,0 +1,55 @@
@@ -0,0 +1,55 @@
|
||||
/* |
||||
* Copyright 2012-2017 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 |
||||
* |
||||
* http://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.context.properties.source; |
||||
|
||||
import java.util.Arrays; |
||||
import java.util.Collections; |
||||
import java.util.HashSet; |
||||
import java.util.Set; |
||||
import java.util.function.Function; |
||||
|
||||
import org.springframework.core.env.PropertySource; |
||||
import org.springframework.core.env.StandardEnvironment; |
||||
|
||||
/** |
||||
* Function used to determine if a {@link ConfigurationPropertySource} should be |
||||
* included when determining unbound elements. If the underlying {@link PropertySource} |
||||
* is a systemEnvironment or systemProperties property source , it will not be considered |
||||
* for unbound element failures. |
||||
* |
||||
* @author Madhura Bhave |
||||
* @since 2.0.0 |
||||
*/ |
||||
public class UnboundElementsSourceFilter implements Function<ConfigurationPropertySource, Boolean> { |
||||
|
||||
private static final Set<String> BENIGN_PROPERTY_SOURCE_NAMES = Collections.unmodifiableSet(new HashSet<>(Arrays.asList( |
||||
StandardEnvironment.SYSTEM_ENVIRONMENT_PROPERTY_SOURCE_NAME, |
||||
StandardEnvironment.SYSTEM_PROPERTIES_PROPERTY_SOURCE_NAME))); |
||||
|
||||
@Override |
||||
public Boolean apply(ConfigurationPropertySource configurationPropertySource) { |
||||
Object underlyingSource = configurationPropertySource.getUnderlyingSource(); |
||||
if (underlyingSource instanceof PropertySource) { |
||||
String name = ((PropertySource) underlyingSource).getName(); |
||||
return !BENIGN_PROPERTY_SOURCE_NAMES.contains(name); |
||||
|
||||
} |
||||
return true; |
||||
} |
||||
|
||||
} |
||||
|
||||
@ -0,0 +1,67 @@
@@ -0,0 +1,67 @@
|
||||
/* |
||||
* Copyright 2012-2017 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 |
||||
* |
||||
* http://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.context.properties.source; |
||||
|
||||
import org.junit.Before; |
||||
import org.junit.Test; |
||||
|
||||
import org.springframework.core.env.StandardEnvironment; |
||||
import org.springframework.mock.env.MockPropertySource; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
import static org.mockito.BDDMockito.given; |
||||
import static org.mockito.Mockito.mock; |
||||
|
||||
/** |
||||
* Tests for {@link UnboundElementsSourceFilter}. |
||||
* |
||||
* @author Madhura Bhave |
||||
*/ |
||||
public class UnboundElementsSourceFilterTests { |
||||
|
||||
private UnboundElementsSourceFilter filter; |
||||
|
||||
private ConfigurationPropertySource source; |
||||
|
||||
@Before |
||||
public void setUp() throws Exception { |
||||
this.filter = new UnboundElementsSourceFilter(); |
||||
this.source = mock(ConfigurationPropertySource.class); |
||||
} |
||||
|
||||
@Test |
||||
public void filterWhenSourceIsSystemEnvironmentPropertySourceShouldReturnFalse() throws Exception { |
||||
MockPropertySource propertySource = new MockPropertySource(StandardEnvironment.SYSTEM_PROPERTIES_PROPERTY_SOURCE_NAME); |
||||
given(this.source.getUnderlyingSource()).willReturn(propertySource); |
||||
assertThat(this.filter.apply(this.source)).isFalse(); |
||||
} |
||||
|
||||
@Test |
||||
public void filterWhenSourceIsSystemPropertiesPropertySourceShouldReturnTrue() throws Exception { |
||||
MockPropertySource propertySource = new MockPropertySource(StandardEnvironment.SYSTEM_ENVIRONMENT_PROPERTY_SOURCE_NAME); |
||||
given(this.source.getUnderlyingSource()).willReturn(propertySource); |
||||
assertThat(this.filter.apply(this.source)).isFalse(); |
||||
} |
||||
|
||||
@Test |
||||
public void filterWhenSourceIsNotSystemShouldReturnTrue() throws Exception { |
||||
MockPropertySource propertySource = new MockPropertySource("test"); |
||||
given(this.source.getUnderlyingSource()).willReturn(propertySource); |
||||
assertThat(this.filter.apply(this.source)).isTrue(); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue