diff --git a/spring-cloud-zookeeper-config/pom.xml b/spring-cloud-zookeeper-config/pom.xml
index f24f5924..0633b795 100644
--- a/spring-cloud-zookeeper-config/pom.xml
+++ b/spring-cloud-zookeeper-config/pom.xml
@@ -34,6 +34,10 @@
org.apache.curator
curator-x-discovery
+
+ org.springframework.cloud
+ spring-cloud-commons
+
org.springframework.cloud
spring-cloud-context
diff --git a/spring-cloud-zookeeper-config/src/main/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataMissingEnvironmentPostProcessor.java b/spring-cloud-zookeeper-config/src/main/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataMissingEnvironmentPostProcessor.java
index 0f199107..afe0a8c5 100644
--- a/spring-cloud-zookeeper-config/src/main/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataMissingEnvironmentPostProcessor.java
+++ b/spring-cloud-zookeeper-config/src/main/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataMissingEnvironmentPostProcessor.java
@@ -16,21 +16,18 @@
package org.springframework.cloud.zookeeper.config;
-import org.springframework.boot.SpringApplication;
import org.springframework.boot.context.config.ConfigDataEnvironmentPostProcessor;
import org.springframework.boot.diagnostics.AbstractFailureAnalyzer;
import org.springframework.boot.diagnostics.FailureAnalysis;
-import org.springframework.boot.env.EnvironmentPostProcessor;
+import org.springframework.cloud.commons.ConfigDataMissingEnvironmentPostProcessor;
import org.springframework.cloud.zookeeper.ZookeeperProperties;
-import org.springframework.core.Ordered;
-import org.springframework.core.env.ConfigurableEnvironment;
-import org.springframework.util.StringUtils;
+import org.springframework.core.env.Environment;
import static org.springframework.cloud.util.PropertyUtils.bootstrapEnabled;
import static org.springframework.cloud.util.PropertyUtils.useLegacyProcessing;
import static org.springframework.cloud.zookeeper.config.ZookeeperConfigDataLocationResolver.PREFIX;
-public class ZookeeperConfigDataMissingEnvironmentPostProcessor implements EnvironmentPostProcessor, Ordered {
+public class ZookeeperConfigDataMissingEnvironmentPostProcessor extends ConfigDataMissingEnvironmentPostProcessor {
/**
* Order of post processor, set to run after
@@ -44,10 +41,10 @@ public int getOrder() {
}
@Override
- public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
+ protected boolean shouldProcessEnvironment(Environment environment) {
// don't run if using bootstrap or legacy processing
if (bootstrapEnabled(environment) || useLegacyProcessing(environment)) {
- return;
+ return false;
}
boolean coreEnabled = environment.getProperty(ZookeeperProperties.PREFIX + ".enabled", Boolean.class,
true);
@@ -56,26 +53,14 @@ public void postProcessEnvironment(ConfigurableEnvironment environment, SpringAp
boolean importCheckEnabled = environment.getProperty(ZookeeperConfigProperties.PREFIX + ".import-check.enabled",
Boolean.class, true);
if (!coreEnabled || !configEnabled || !importCheckEnabled) {
- return;
- }
- String property = environment.getProperty("spring.config.import");
- if (!StringUtils.hasText(property)) {
- throw new ImportException("No spring.config.import set", false);
- }
- if (!property.contains(PREFIX)) {
- throw new ImportException("spring.config.import missing " + PREFIX, true);
+ return false;
}
+ return true;
}
- static class ImportException extends RuntimeException {
-
- final boolean missingPrefix;
-
- ImportException(String message, boolean missingPrefix) {
- super(message);
- this.missingPrefix = missingPrefix;
- }
-
+ @Override
+ protected String getPrefix() {
+ return PREFIX;
}
static class ImportExceptionFailureAnalyzer extends AbstractFailureAnalyzer {
diff --git a/spring-cloud-zookeeper-config/src/test/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataMissingEnvironmentPostProcessorTests.java b/spring-cloud-zookeeper-config/src/test/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataMissingEnvironmentPostProcessorTests.java
new file mode 100644
index 00000000..14852874
--- /dev/null
+++ b/spring-cloud-zookeeper-config/src/test/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataMissingEnvironmentPostProcessorTests.java
@@ -0,0 +1,106 @@
+/*
+ * Copyright 2015-2021 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.cloud.zookeeper.config;
+
+import org.junit.jupiter.api.Test;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.mock.env.MockEnvironment;
+
+import static org.assertj.core.api.Assertions.assertThatCode;
+import static org.assertj.core.api.Assertions.assertThatThrownBy;
+import static org.mockito.Mockito.mock;
+
+/**
+ * @author Ryan Baxter
+ */
+class ZookeeperConfigDataMissingEnvironmentPostProcessorTests {
+
+ @Test
+ void noSpringConfigImport() {
+ MockEnvironment environment = new MockEnvironment();
+ SpringApplication app = mock(SpringApplication.class);
+ ZookeeperConfigDataMissingEnvironmentPostProcessor processor = new ZookeeperConfigDataMissingEnvironmentPostProcessor();
+ assertThatThrownBy(() -> processor.postProcessEnvironment(environment, app))
+ .isInstanceOf(ZookeeperConfigDataMissingEnvironmentPostProcessor.ImportException.class);
+ }
+
+ @Test
+ void boostrap() {
+ MockEnvironment environment = new MockEnvironment();
+ environment.setProperty("spring.cloud.bootstrap.enabled", "true");
+ SpringApplication app = mock(SpringApplication.class);
+ ZookeeperConfigDataMissingEnvironmentPostProcessor processor = new ZookeeperConfigDataMissingEnvironmentPostProcessor();
+ assertThatCode(() -> processor.postProcessEnvironment(environment, app)).doesNotThrowAnyException();
+ }
+
+ @Test
+ void legacy() {
+ MockEnvironment environment = new MockEnvironment();
+ environment.setProperty("spring.config.use-legacy-processing", "true");
+ SpringApplication app = mock(SpringApplication.class);
+ ZookeeperConfigDataMissingEnvironmentPostProcessor processor = new ZookeeperConfigDataMissingEnvironmentPostProcessor();
+ assertThatCode(() -> processor.postProcessEnvironment(environment, app)).doesNotThrowAnyException();
+ }
+
+ @Test
+ void configNotEnabled() {
+ MockEnvironment environment = new MockEnvironment();
+ environment.setProperty("spring.cloud.zookeeper.enabled", "false");
+ SpringApplication app = mock(SpringApplication.class);
+ ZookeeperConfigDataMissingEnvironmentPostProcessor processor = new ZookeeperConfigDataMissingEnvironmentPostProcessor();
+ assertThatCode(() -> processor.postProcessEnvironment(environment, app)).doesNotThrowAnyException();
+ }
+
+ @Test
+ void importCheckNotEnabled() {
+ MockEnvironment environment = new MockEnvironment();
+ environment.setProperty("spring.cloud.zookeeper.config.import-check.enabled", "false");
+ SpringApplication app = mock(SpringApplication.class);
+ ZookeeperConfigDataMissingEnvironmentPostProcessor processor = new ZookeeperConfigDataMissingEnvironmentPostProcessor();
+ assertThatCode(() -> processor.postProcessEnvironment(environment, app)).doesNotThrowAnyException();
+ }
+
+ @Test
+ void importSinglePropertySource() {
+ MockEnvironment environment = new MockEnvironment();
+ environment.setProperty("spring.config.import", "zookeeper:http://localhost:8888");
+ SpringApplication app = mock(SpringApplication.class);
+ ZookeeperConfigDataMissingEnvironmentPostProcessor processor = new ZookeeperConfigDataMissingEnvironmentPostProcessor();
+ assertThatCode(() -> processor.postProcessEnvironment(environment, app)).doesNotThrowAnyException();
+ }
+
+ @Test
+ void importMultiplePropertySource() {
+ MockEnvironment environment = new MockEnvironment();
+ environment.setProperty("spring.config.import", "zookeeper:http://localhost:8888,file:./app.properties");
+ SpringApplication app = mock(SpringApplication.class);
+ ZookeeperConfigDataMissingEnvironmentPostProcessor processor = new ZookeeperConfigDataMissingEnvironmentPostProcessor();
+ assertThatCode(() -> processor.postProcessEnvironment(environment, app)).doesNotThrowAnyException();
+ }
+
+ @Test
+ void importMultiplePropertySourceAsList() {
+ MockEnvironment environment = new MockEnvironment();
+ environment.setProperty("spring.config.import[0]", "zookeeper:http://localhost:8888");
+ environment.setProperty("spring.config.import[1]", "file:./app.properties");
+ SpringApplication app = mock(SpringApplication.class);
+ ZookeeperConfigDataMissingEnvironmentPostProcessor processor = new ZookeeperConfigDataMissingEnvironmentPostProcessor();
+ assertThatCode(() -> processor.postProcessEnvironment(environment, app)).doesNotThrowAnyException();
+ }
+
+}
diff --git a/spring-cloud-zookeeper-config/src/test/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataNoImportIntegrationTests.java b/spring-cloud-zookeeper-config/src/test/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataNoImportIntegrationTests.java
index d8e77955..85a65e13 100644
--- a/spring-cloud-zookeeper-config/src/test/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataNoImportIntegrationTests.java
+++ b/spring-cloud-zookeeper-config/src/test/java/org/springframework/cloud/zookeeper/config/ZookeeperConfigDataNoImportIntegrationTests.java
@@ -25,7 +25,7 @@
import org.springframework.boot.builder.SpringApplicationBuilder;
import org.springframework.boot.test.system.CapturedOutput;
import org.springframework.boot.test.system.OutputCaptureExtension;
-import org.springframework.cloud.zookeeper.config.ZookeeperConfigDataMissingEnvironmentPostProcessor.ImportException;
+import org.springframework.cloud.commons.ConfigDataMissingEnvironmentPostProcessor;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Configuration;
@@ -43,7 +43,7 @@ public class ZookeeperConfigDataNoImportIntegrationTests {
@Test
public void exceptionThrownIfNoImport(CapturedOutput output) {
Assertions.assertThatThrownBy(() -> new SpringApplicationBuilder(Config.class).web(WebApplicationType.NONE)
- .run("--spring.application.name=" + APP_NAME)).isInstanceOf(ImportException.class);
+ .run("--spring.application.name=" + APP_NAME)).isInstanceOf(ConfigDataMissingEnvironmentPostProcessor.ImportException.class);
assertThat(output).contains("No spring.config.import property has been defined")
.contains("Add a spring.config.import=zookeeper: property to your configuration");
@@ -53,7 +53,7 @@ public void exceptionThrownIfNoImport(CapturedOutput output) {
public void exceptionThrownIfImportMissingZookeeper(CapturedOutput output) {
Assertions.assertThatThrownBy(() -> new SpringApplicationBuilder(Config.class).web(WebApplicationType.NONE).run(
"--spring.config.import=optional:file:somefile.properties", "--spring.application.name=" + APP_NAME))
- .isInstanceOf(ImportException.class);
+ .isInstanceOf(ConfigDataMissingEnvironmentPostProcessor.ImportException.class);
assertThat(output).contains("spring.config.import property is missing a " + PREFIX)
.contains("Add a spring.config.import=zookeeper: property to your configuration");