java - 如何使用 JUnit 测试依赖于环境变量的代码？

Question

我有一段使用环境变量的 Java 代码，代码的行为取决于该变量的值。我想用不同的环境变量值来测试这段代码。我怎样才能在 JUnit 中做到这一点？

我见过some ways to set environment variables in Java一般来说，但我对它的单元测试方面更感兴趣，特别是考虑到测试不应该互相干扰。

Answer 1

图书馆System Lambda有一个方法 withEnvironmentVariable 用于设置环境变量。

import static com.github.stefanbirkner.systemlambda.SystemLambda.*;

public void EnvironmentVariablesTest {
  @Test
  public void setEnvironmentVariable() {
    String value = withEnvironmentVariable("name", "value")
      .execute(() -> System.getenv("name"));
    assertEquals("value", value);
  }
}

对于 Java 5 到 7，库 System Rules有一个名为 EnvironmentVariables 的 JUnit 规则。

import org.junit.contrib.java.lang.system.EnvironmentVariables;

public class EnvironmentVariablesTest {
  @Rule
  public final EnvironmentVariables environmentVariables
    = new EnvironmentVariables();

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

全面披露:我是这两个库的作者。