Mockito-mock ApplicationContext

Question

我有一个Springboot应用程序，它在运行时根据用户传递的输入参数从ApplicationContext中查找bean。对于这种方法，我试图编写Mockito测试用例，但它不起作用，并抛出NullPointerException。

引导应用程序的类：

@SpringBootApplication
public class MyApplication {

    private static ApplicationContext appContext;
    
    public static void main(String[] args) {
        appContext = SpringApplication.run(MyApplication.class, args);
    }
    
    public static ApplicationContext getApplicationContext() {
        return appContext;
    }
    
}

我试图为其编写测试用例的类：

@Service
public class Mailbox {

    @Autowired
    MailProcessor processor;
    
    public void processUserInput(Envelope object) {
    
        processor.setCommand(MyApplication.getApplicationContext().getBean(object.getAction(), Command.class));
        processor.allocateEnvelopes(object);
        
    }
} 

我的测试用例如下：

@RunWith(MockitoJUnitRunner.class)
@SpringBootTest
@ActiveProfiles("test")
public class MailboxTest {

    @Mock
    MailProcessor processor;
    
    @InjectMocks
    Mailbox mailbox;
    
    @Test
    public void testProcessUserInput() {
        
        Envelope message = new Envelope();
        message.setAction("userAction");
        message.setValue("userInput");
        
        doNothing().when(processor).setCommand(any());
        doNothing().when(processor).allocateEnvelopes(any());
        
        mailbox.processUserInput(message);
        
        Mockito.verify(processor).allocateEnvelopes(any());
        
    }
    
    
}

每当我运行测试用例时，它都会在处理器上给出NullPointerException。setCommand（MyApplication.getApplicationContext（）。getBean（object.getAction（），命令。阶级） 在邮箱类中。如何模拟ApplicationContext查找？我错过了任何嘲弄的步骤吗？




Answer 1

在第一次测试之前，尝试通过注入处理器初始化邮箱对象。

mail=new Mailbox（处理器）；

Answer 2

不能说肯定没有调试，但它看起来像MyApplication.getApplication Context（）返回null。

与其将其存储在静态变量中，不如尝试在@Service类中需要的地方注入ApplicationContext：

@Autowired
private ApplicationContext appContext;

Answer 3

Spring方面，您的代码看起来不太好，尤其是不可单元测试。我会解释：

1. 您的邮箱服务在任何级别都不应该知道MyApplication。它是spring boot应用程序的入口点，您的业务逻辑不应该依赖于此。确实，您可以将应用程序上下文直接注入到类中。请参见下面的示例。这里的另一个（更“老派”的）选项是在邮箱服务中使用ApplicationContextAware接口（参见本例）。然而，这仍然是一个糟糕的代码：

@Service
public class Mailbox {
 private final ApplicationContext ctx;
 ...
 public Mailbox(ApplicationContext ctx) {
     this.ctx = ctx;
 }
 ...
}

即使您解决了这个问题，一般来说，依赖ApplicationContext也不是一个好主意。因为这样你就变得依赖于spring，而在Mailbox类中没有理由这样做。不过，这个类将成为单元可测试的。

在决议方面：

在Spring，你可以注入一个映射

public interface Command {
 void execute();
}

@Component("delete") // note this "delete" word - it will be a key in the map in the Mailbox
public class DeleteMailCommand implements Command {
    @Override
    public void execute() {
        System.out.println("Deleting email");
    }
}

@Component("send")
public class SendMailCommand implements Command{
    @Override
    public void execute() {
        System.out.println("Sending Mail");
    }
}

请注意，所有命令都必须由spring驱动（这似乎是您的情况）。现在，邮箱将如下所示：

@Service
public class Mailbox {
    private final Map<String, Command> allCommands;
    private final MailProcessor processor;
    // Note this map: it will be ["delete" -> <bean of type DeleteMailCommand>, "send" -> <bean of type SendMailCommand>]
    public Mailbox(Map<String, Command> allCommands, MailProcessor mailProcessor) {
        this.allCommands = allCommands;
        this.processor = mailProcessor;
    }

    public void processUserInput(Envelope envelope) {
        Command cmd = allCommands.get(envelope.getAction());
        processor.executeCommand(cmd);

    }
}

这个解决方案很容易进行单元测试，因为如果愿意，您可以用mock命令填充映射，而不需要处理应用程序上下文。

更新

我现在看了一下你的测试，它也不太好，对不起：）@RunBy（MockitoJUnitRunner.class）用于运行单元测试（根本没有Spring）。将此注释与运行完整系统测试的@SpringBootTest放在一起是没有意义的：启动整个Spring启动应用程序，加载配置等等。

因此，请确保要运行哪种类型的测试，并使用适当的注释。