当 applicationcontext 根据 application.properties 中的环境值初始化时 Springboot 单元测试

Question

我现在有一个工作的 Spring Boot 应用程序并添加单元测试。我正在尝试为服务层编写一个单元测试，该层有两个可能的处理程序。在某一时刻，任何一个处理程序都将根据服务的部署而处于活动状态。

我的代码如下所示。

配置-

package com.org.app.configs;

@Configuration
public class DeploymentFactoryManager {
    @Value("${deployment:}")
    private String deploymentEnv;

    @Autowired
    ApplicationContext ctx;

    @Bean
    public IDeploymentHandler getDeploymentInterface() throws InvalidConfigException {
        IDeploymentHandler deploymentHandler = null;
        if(deploymentEnv.equalsIgnoreCase("kubernetes")){
            deploymentHandler = ctx.getBean(KubernetesDeployHandler.class);
        }else if(deploymentEnv.equalsIgnoreCase("docker")){
            deploymentHandler =  ctx.getBean(DockerDeployHandler.class);
        } else {
            throw new InvalidConfigException("Unknown queue type : "+deploymentEnv);
        }
        return deploymentHandler;
    }
}

服务-

package com.org.app.services;

public List<String> status() throws InvalidConfigException{
        
    @Autowired
    DeploymentFactoryManager deploymentFactoryManager;
        
    public List<String> getStatus(){
        IDeploymentHandler iDeploymentHandler = deploymentFactoryManager.getDeploymentInterface();
        return iDeploymentHandler.list();
    }

我在编写单元测试时遇到问题。到目前为止，我已经尝试使用 ReflectionTestUtils 设置字段，但失败并显示消息 必须指定字段的目标对象或目标类。 我尝试使用 @TestPropertySource 但这似乎也不起作用。代码似乎不适合单元测试。

@BeforeAll
    public static void setup(){
        ReflectionTestUtils.setField(deploymentFactoryManager, "deploymentEnv", "kubernetes");
    }

有什么建议吗？

Answer 1

附带说明：如果注入每个依赖项，您的代码将更容易测试，例如：

public IDeploymentHandler getDeploymentInterface( @Value("${deployment:}") String deploymentEnv, ApplicationContext ctx ) throws InvalidConfigException {

一般来说，从上下文中获取 bean 并不是一个好的策略；这与 Spring 的理念相反：告诉你想要什么，但不要索取。