在 Spring Boot 中,可以通过在测试类上添加 @TestPropertySource 注解来设置测试环境变量。该注解可以指定一个或多个 properties 文件,或者一个或多个 key-value 形式的环境变量。
例如,如果您有一个名为 application-test.properties 的测试配置文件,可以在测试类上添加如下注解:
@TestPropertySource(locations = "classpath:application-test.properties")
如果您想直接在注解中指定环境变量,可以使用 properties 属性,如下所示:
@TestPropertySource(properties = {"key1=value1", "key2=value2"})
值得注意的是,当您使用 @TestPropertySource 注解时,它会覆盖掉 application.properties 文件中的相同属性。
如果你想在测试类中使用环境变量,可以使用 @Autowired
注解,在测试类中引入 org.springframework.core.env.Environment
类。在测试类中使用该类的 getProperty()
方法获取环境变量。