mockito单元测试环境变量

### Mockito单元测试中配置和使用环境变量的方法 在Mockito单元测试中，环境变量的配置和使用可以通过多种方式实现。以下是几种常见的方法： #### 1. 使用`System.setProperty`设置环境变量 在单元测试中，可以直接通过`System.setProperty`方法临时设置环境变量[^1]。这种方式适用于需要在测试运行期间动态修改环境变量的场景。 ```java import org.junit.jupiter.api.Test; import static org.mockito.Mockito.*; public class EnvironmentVariableTest { @Test public void testEnvironmentVariable() { // 设置环境变量 System.setProperty("MY_ENV_VAR", "testValue"); // 验证环境变量是否正确设置 String envVar = System.getProperty("MY_ENV_VAR"); mockStatic(System.class); when(System.getProperty("MY_ENV_VAR")).thenReturn(envVar); // 测试逻辑 assert System.getProperty("MY_ENV_VAR").equals("testValue"); // 清理环境变量 System.clearProperty("MY_ENV_VAR"); } } ``` #### 2. 使用`PowerMockito`模拟系统属性 如果需要更复杂的控制，可以使用`PowerMockito`来模拟系统属性的行为[^2]。这种方式允许对`System.getProperty`方法进行完全控制。 ```java import org.junit.jupiter.api.Test; import org.powermock.api.mockito.PowerMockito; import static org.mockito.Mockito.*; public class PowerMockitoEnvironmentTest { @Test public void testPowerMockitoEnvironmentVariable() throws Exception { PowerMockito.mockStatic(System.class); when(System.getenv("MY_ENV_VAR")).thenReturn("mockedValue"); String envVar = System.getenv("MY_ENV_VAR"); assert envVar.equals("mockedValue"); } } ``` #### 3. 使用外部配置文件加载环境变量 对于需要从外部配置文件中加载环境变量的场景，可以结合`Properties`类读取配置文件，并将其注入到测试环境中[^3]。 ```java import java.io.FileInputStream; import java.util.Properties; public class EnvironmentLoader { public static void loadEnvironmentVariables(String configFilePath) throws Exception { Properties properties = new Properties(); try (FileInputStream fis = new FileInputStream(configFilePath)) { properties.load(fis); for (String key : properties.stringPropertyNames()) { System.setProperty(key, properties.getProperty(key)); } } } } ``` 在测试中调用该方法以加载环境变量： ```java import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; public class EnvironmentVariableLoaderTest { @BeforeEach public void setUp() throws Exception { EnvironmentLoader.loadEnvironmentVariables("config.properties"); } @Test public void testLoadedEnvironmentVariable() { String envVar = System.getProperty("MY_ENV_VAR"); assert envVar.equals("expectedValue"); } } ``` #### 注意事项 - 在单元测试结束后，务必清理设置的环境变量，以免影响其他测试[^4]。 - 如果使用了`PowerMockito`，需要确保项目中包含相关依赖，并正确配置测试框架。 ### 总结 上述方法提供了在Mockito单元测试中配置和使用环境变量的不同方式。具体选择取决于测试需求和项目的复杂性。