使用 Mockito 验证 Executor.execute 内部方法调用-java教程-PHP中文网

使用 Mockito 验证 Executor.execute 内部方法调用

霞舞

发布： 2025-08-31 15:44:01

原创

517人浏览过

使用 mockito 验证 executor.execute 内部方法调用

本文旨在解决在使用 Mockito 验证Executor.execute方法内部调用的方法时遇到的线程问题。通过使用SynchronousExecutor，可以确保代码同步执行，从而允许 Mockito 正确地验证预期的方法调用。本文将详细介绍如何使用SynchronousExecutor进行单元测试。

在单元测试中，我们经常需要验证某个方法是否被调用。当方法调用发生在异步线程中，例如通过Executor.execute()执行时，Mockito 的验证可能会失败，因为它无法捕捉到异步线程中的调用。解决此问题的关键在于使异步执行变为同步执行，以便 Mockito 能够正确地验证方法调用。

使用 SynchronousExecutor 实现同步执行

SynchronousExecutor 是一个简单的Executor实现，它直接在调用线程中执行任务，而不是将其提交到另一个线程。通过将 SynchronousExecutor 注入到你的代码中，你可以确保 prepareContext() 方法在与测试相同的线程中执行，从而允许 Mockito 验证 handleMessage() 方法的调用。

示例代码：

首先，创建一个 SynchronousExecutor 类：

动态WEB网站中的PHP和MySQL：直观的QuickPro指南第2版

动态WEB网站中的PHP和MySQL详细反映实际程序的需求，仔细地探讨外部数据的验证(例如信用卡卡号的格式)、用户登录以及如何使用模板建立网页的标准外观。动态WEB网站中的PHP和MySQL的内容不仅仅是这些。书中还提到如何串联JavaScript与PHP让用户操作时更快、更方便。还有正确处理用户输入错误的方法，让网站看起来更专业。另外还引入大量来自PEAR外挂函数库的强大功能，对常用的、强大的包

508

查看详情

import java.util.concurrent.Executor;

public class SynchronousExecutor implements Executor {
    @Override
    public void execute(Runnable command) {
        command.run();
    }
}

登录后复制

然后，修改你的代码，以便可以注入 Executor 实例。例如，假设你的 prepareContext() 方法位于一个名为 MyClass 的类中：

public class MyClass {

    private final Executor executor;
    private final MessageHandler messageHandler;

    public MyClass(Executor executor, MessageHandler messageHandler) {
        this.executor = executor;
        this.messageHandler = messageHandler;
    }

    public void processMessage(Message message) {
        executor.execute(() -> prepareContext(message));
    }

    private void prepareContext(Message message) {
        messageHandler.handleMessage(message);
    }
}

interface MessageHandler {
    void handleMessage(Message message);
}

interface Message {}

登录后复制

在你的单元测试中，使用 SynchronousExecutor 替换真实的 Executor 实例：

import org.junit.jupiter.api.Test;
import org.mockito.Mockito;

import static org.mockito.Mockito.verify;

public class MyClassTest {

    @Test
    public void testHandleMessageIsCalled() {
        // Arrange
        MessageHandler messageHandler = Mockito.mock(MessageHandler.class);
        SynchronousExecutor executor = new SynchronousExecutor();
        MyClass myClass = new MyClass(executor, messageHandler);
        Message message = Mockito.mock(Message.class);

        // Act
        myClass.processMessage(message);

        // Assert
        verify(messageHandler).handleMessage(message);
    }
}

登录后复制

解释：

我们创建了一个 SynchronousExecutor 实例。
我们将 SynchronousExecutor 注入到 MyClass 的构造函数中。
在测试中，我们调用 myClass.processMessage(message)，由于使用了 SynchronousExecutor，prepareContext() 方法将在与测试相同的线程中执行。
最后，我们使用 Mockito.verify() 验证 messageHandler.handleMessage(message) 是否被调用。

注意事项

依赖注入： 确保你的代码允许注入 Executor 实例。这通常通过构造函数注入或 setter 注入来实现。
测试环境： 仅在测试环境中使用 SynchronousExecutor。在生产环境中，应使用合适的异步执行器。
并发问题： 如果你的代码涉及到更复杂的并发场景，SynchronousExecutor 可能无法完全模拟真实环境。在这种情况下，你可能需要使用更高级的测试技术，例如使用 CountDownLatch 等并发工具来同步线程。