Spring中设置环境变量及测试时System.getProperty()问题求助

嘿，我之前也碰到过一模一样的问题——给每个测试类手动加VM参数真的太麻烦了！下面给你几个靠谱的方案，能在Spring应用上下文加载前就设置好系统属性/环境变量：

1. 利用JUnit的前置静态方法（快速直接） #

不管是JUnit 5还是JUnit 4，都可以通过静态前置方法在Spring上下文初始化前完成系统属性的设置，还能在测试后还原属性避免污染其他用例：

JUnit 5 示例 #

import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;

@SpringBootTest
public class MyTest {

    private static String originalProperty;

    @BeforeAll
    static void setupSystemProperties() {
        // 先保存原有属性值，避免影响其他测试
        originalProperty = System.getProperty("my.custom.property");
        // 设置测试用的系统属性
        System.setProperty("my.custom.property", "test-environment-value");
    }

    @AfterAll
    static void restoreSystemProperties() {
        // 测试结束后还原属性
        if (originalProperty != null) {
            System.setProperty("my.custom.property", originalProperty);
        } else {
            System.clearProperty("my.custom.property");
        }
    }

    @Test
    void testSystemPropertyUsage() {
        // 这里就能获取到提前设置的属性值
        String propertyValue = System.getProperty("my.custom.property");
        // 业务逻辑测试...
    }
}

JUnit 4 示例 #

用@BeforeClass替代JUnit 5的@BeforeAll即可：

import org.junit.BeforeClass;
import org.junit.AfterClass;
import org.junit.Test;
import org.springframework.test.context.ContextConfiguration;

@ContextConfiguration(classes = MyAppConfig.class)
public class JUnit4Test {

    private static String originalProp;

    @BeforeClass
    public static void setupProperties() {
        originalProp = System.getProperty("test.db.url");
        System.setProperty("test.db.url", "jdbc:h2:mem:test-db");
    }

    @AfterClass
    public static void teardownProperties() {
        if (originalProp != null) {
            System.setProperty("test.db.url", originalProp);
        } else {
            System.clearProperty("test.db.url");
        }
    }

    @Test
    void testDbConnection() {
        // 验证属性生效...
    }
}

2. 自定义Spring上下文初始化器（可复用） #

如果多个测试类都需要相同的属性配置，自定义ApplicationContextInitializer是更优雅的复用方案，它会在Spring上下文启动前自动执行：

第一步：实现初始化器类 #

import org.springframework.context.ApplicationContextInitializer;
import org.springframework.context.ConfigurableApplicationContext;

public class TestSystemPropertyInitializer implements ApplicationContextInitializer<ConfigurableApplicationContext> {

    @Override
    public void initialize(ConfigurableApplicationContext applicationContext) {
        // 批量设置测试用系统属性
        System.setProperty("app.env", "test");
        System.setProperty("api.timeout", "5000");
    }
}

第二步：在测试类中指定初始化器 #

import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.ContextConfiguration;

@SpringBootTest
@ContextConfiguration(initializers = TestSystemPropertyInitializer.class)
public class ReusableTest {

    @Test
    void testWithGlobalProperties() {
        // 这里的System.getProperty()能拿到初始化器中设置的值
    }
}

如果想让所有测试类自动应用这个初始化器，只需在src/test/resources下创建META-INF/spring.factories文件，添加以下内容：

org.springframework.context.ApplicationContextInitializer=com.your.package.TestSystemPropertyInitializer

3. 优化建议：改用Spring Environment替代直接调用System.getProperty()（更符合Spring生态） #

其实更推荐的做法是，把业务代码中直接调用System.getProperty()的逻辑，改成注入Spring的Environment来获取属性：

import org.springframework.core.env.Environment;
import org.springframework.stereotype.Component;

@Component
public class BusinessService {

    private final Environment environment;

    public BusinessService(Environment environment) {
        this.environment = environment;
    }

    public String getConfigValue() {
        // 用Spring环境获取属性，兼容多种配置来源
        return environment.getProperty("business.config.key");
    }
}

这种情况下，你可以直接用@TestPropertySource快速设置测试属性，完全不需要操作系统属性：

@SpringBootTest
@TestPropertySource(properties = {"business.config.key=test-config-value"})
public class BusinessServiceTest {

    @Autowired
    private BusinessService businessService;

    @Test
    void testConfigValue() {
        assertEquals("test-config-value", businessService.getConfigValue());
    }
}

这种方式既不用手动加VM参数，也不用担心系统属性的全局污染，完全贴合Spring的配置生态！

内容的提问来源于stack exchange，提问作者murthy