如何在 Spring 测试中设置环境变量或系统属性？

Question

我想编写一些测试来检查已部署 WAR 的 XML Spring 配置。不幸的是，有些 bean 需要设置一些环境变量或系统属性。使用@ContextConfiguration方便的测试样式时，如何在spring bean初始化之前设置环境变量？

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = "classpath:whereever/context.xml")
public class TestWarSpringContext { ... }

如果我使用注释配置应用程序上下文，我看不到在 spring 上下文初始化之前我可以做某事的钩子。

Answer 1

您可以在静态初始化程序中初始化 System 属性：

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = "classpath:whereever/context.xml")
public class TestWarSpringContext {

    static {
        System.setProperty("myproperty", "foo");
    }

}

静态初始化代码将在spring应用上下文初始化之前执行。

Answer 2

从 Spring 4.1 开始，正确的做法是使用 @TestPropertySource 注释。

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = "classpath:whereever/context.xml")
@TestPropertySource(properties = {"myproperty = foo"})
public class TestWarSpringContext {
    ...    
}

参见Spring docs 和Javadocs 中的@TestPropertySource。

Answer 3

也可以使用测试 ApplicationContextInitializer 来初始化系统属性：

public class TestApplicationContextInitializer implements ApplicationContextInitializer<ConfigurableApplicationContext>
{
    @Override
    public void initialize(ConfigurableApplicationContext applicationContext)
    {
        System.setProperty("myproperty", "value");
    }
}

然后在测试类上配置它除了Spring上下文配置文件位置：

@ContextConfiguration(initializers = TestApplicationContextInitializer.class, locations = "classpath:whereever/context.xml", ...)
@RunWith(SpringJUnit4ClassRunner.class)
public class SomeTest
{
...
}

如果应该为所有单元测试设置某个系统属性，这样可以避免代码重复。

Answer 4

目前这里的所有答案都只讨论与设置更复杂的环境变量不同的系统属性，尤其是。用于测试。值得庆幸的是，可以使用下面的类，并且类文档有很好的例子

EnvironmentVariables.html

文档中的一个简单示例，已修改为使用 @SpringBootTest

@SpringBootTest
public class EnvironmentVariablesTest {
   @ClassRule
   public final EnvironmentVariables environmentVariables = new EnvironmentVariables().set("name", "value");

   @Test
   public void test() {
     assertEquals("value", System.getenv("name"));
   }
 }

Answer 5

如果您希望您的变量对所有测试都有效，您可以在您的测试资源目录中有一个application.properties 文件（默认情况下：src/test/resources），它看起来像这样：

MYPROPERTY=foo

这将被加载和使用，除非您通过 @TestPropertySource 或类似方法定义 - 加载属性的确切顺序可以在 Spring 文档章节 24. Externalized Configuration 中找到。

Answer 6

您可以将系统属性设置为 VM 参数。

如果您的项目是 maven 项目，那么您可以在运行测试类时执行以下命令：

mvn test -Dapp.url="https://stackoverflow.com"

测试类：

public class AppTest  {
@Test
public void testUrl() {
    System.out.println(System.getProperty("app.url"));
    }
}

如果你想在 Eclipse 中运行单独的测试类或方法：

1) 转到运行 -> 运行配置

2) 在左侧的 Junit 部分下选择您的测试类。

3) 执行以下操作：

Answer 7

对于单元测试，系统变量在我执行“mvn clean install”时尚未实例化，因为没有服务器运行应用程序。所以为了设置系统属性，我需要在 pom.xml 中进行。像这样：

<plugin>
    <groupId>org.apache.maven.plugins</groupId>
    <artifactId>maven-surefire-plugin</artifactId>
    <version>2.21.0</version>
    <configuration>
        <systemPropertyVariables>
            <propertyName>propertyValue</propertyName>
            <MY_ENV_VAR>newValue</MY_ENV_VAR>
            <ENV_TARGET>olqa</ENV_TARGET>
            <buildDirectory>${project.build.directory}</buildDirectory>
        </systemPropertyVariables>
    </configuration>
</plugin>

Answer 8

对于springboot，我认为这是最简单的方法，可以在java中使用@SpringBootTest注解：

@SpringBootTest(
    properties = { "spring.application.name=example", "ENV_VARIABLE=secret" }
)
public class ApplicationTest {

    // Write your tests here

}

或者在 kotlin 中你可以这样做：

@SpringBootTest(
    properties = ["spring.application.name=example", "ENV_VARIABLE=secret"]
)
internal class ApplicationKTest {

    // Write your tests here

}

就是这样，您的测试应该使用您在注释中定义的属性覆盖属性。 假设您的application.yml 看起来像这样：

spring:
  application:
    name: "app"

db:
  username: "user"
  password: ${ENV_VARIABLE:default}

那么在测试期间它将是：

• spring 属性spring.application.name 将返回值"example"
• 环境变量ENV_VARIABLE 将返回"secret"，因此如果您在代码中使用值db.password，它将返回"secret"。

Answer 9

如果您有很多测试类（启动 tomcat/server 的 IT 测试），并且测试失败，您需要使用设置系统属性 System.setProperty("ccm.configs.dir", configPath); 由于您需要确保在 spring 开始之前设置它，因此您需要将它放在类中的静态上下文中。 并确保任何可能依赖于它的测试都获得此设置的系统属性，请在您的测试文件夹中定义一个简单的配置类来设置该变量。 PS在我的情况下，所需的环境变量是“ccm.configs.dir” 这是我在测试文件夹中添加的内容，

@Configuration
public class ConfigLoader {
  static {
    
    System.setProperty("ccm.configs.dir", "path/to/the/resource");
  }

}

我所有的集成测试类都能够在它们运行时获取已经设置的变量。