Browse Source

Add support for = separator in EnvironmentTestUtils

pull/234/head
Dave Syer 12 years ago
parent
commit
dba6a5bfaf
  1. 21
      spring-boot/src/test/java/org/springframework/boot/test/EnvironmentTestUtils.java

21
spring-boot/src/test/java/org/springframework/boot/test/EnvironmentTestUtils.java

@ -19,8 +19,10 @@ package org.springframework.boot.test; @@ -19,8 +19,10 @@ package org.springframework.boot.test;
import java.util.HashMap;
import java.util.Map;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.core.env.Environment;
import org.springframework.core.env.MapPropertySource;
/**
@ -30,15 +32,32 @@ import org.springframework.core.env.MapPropertySource; @@ -30,15 +32,32 @@ import org.springframework.core.env.MapPropertySource;
*/
public abstract class EnvironmentTestUtils {
/**
* Add additional (high priority) values to an {@link Environment} owned by an
* {@link ApplicationContext}. Name-value pairs can be specified with colon (":") or
* equals ("=") separators.
*
* @param context the context with an environment to modify
* @param pairs the name:value pairs
*/
public static void addEnvironment(ConfigurableApplicationContext context,
String... pairs) {
addEnvironment(context.getEnvironment(), pairs);
}
public static void addEnvironment(ConfigurableEnvironment environment, String... pairs) {
/**
* Add additional (high priority) values to an {@link Environment}. Name-value pairs
* can be specified with colon (":") or equals ("=") separators.
*
* @param environment the environment to modify
* @param pairs the name:value pairs
*/
public static void addEnvironment(ConfigurableEnvironment environment,
String... pairs) {
Map<String, Object> map = new HashMap<String, Object>();
for (String pair : pairs) {
int index = pair.indexOf(":");
index = index < 0 ? index = pair.indexOf("=") : index;
String key = pair.substring(0, index > 0 ? index : pair.length());
String value = index > 0 ? pair.substring(index + 1) : "";
map.put(key.trim(), value.trim());

Loading…
Cancel
Save