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

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

花韻仙語

发布： 2025-08-31 15:25:22

原创

775人浏览过

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

在单元测试中，我们经常需要验证某个方法是否被调用，以及调用时传入的参数是否正确。当被验证的方法是在 Executor.execute 内部被调用的，由于其异步执行的特性，Mockito 可能会无法捕捉到该方法的调用，导致验证失败。

例如，我们有以下代码：

public class MyClass {

    private final MessageHandler messageHandler;
    private final Executor executor;

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

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

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

interface MessageHandler {
    void handleMessage(Message message);
}

登录后复制

我们想要验证 messageHandler.handleMessage(message) 是否被调用。如果直接使用 Mockito.verify(messageHandler).handleMessage(message)，可能会遇到验证失败的情况，因为 handleMessage 方法是在 executor 的线程中执行的。

为了解决这个问题，我们可以使用 SynchronousExecutor。SynchronousExecutor 会立即执行提交的任务，而不是将其放入队列等待执行。

import java.util.concurrent.Executor;

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

登录后复制

接下来，我们需要将 SynchronousExecutor 注入到 MyClass 中。在测试用例中，我们可以这样做：

法语写作助手

法语助手旗下的AI智能写作平台，支持语法、拼写自动纠错，一键改写、润色你的法语作文。

查看详情

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

import java.util.concurrent.Executor;

import static org.mockito.Mockito.verify;

public class MyClassTest {

    @Test
    public void testProcessMessage() {
        MessageHandler messageHandler = Mockito.mock(MessageHandler.class);
        Executor executor = new SynchronousExecutor();
        MyClass myClass = new MyClass(messageHandler, executor);
        Message message = new Message("test message");

        myClass.processMessage(message);

        verify(messageHandler).handleMessage(message);
    }

    // 假设的 Message 类
    static class Message {
        private String content;

        public Message(String content) {
            this.content = content;
        }

        public String getContent() {
            return content;
        }
    }
}

登录后复制

在这个测试用例中，我们创建了一个 MessageHandler 的 Mock 对象，并使用 SynchronousExecutor 实例化了 MyClass。调用 myClass.processMessage(message) 后，prepareContext 方法会立即在当前线程中执行，从而 messageHandler.handleMessage(message) 也会立即被调用。此时，Mockito.verify(messageHandler).handleMessage(message) 就能正确地验证该方法是否被调用。

注意事项：

使用 SynchronousExecutor 会将异步执行变为同步执行，这可能会影响测试的真实性。因此，只应在需要验证 Executor.execute 内部方法调用时使用。
确保在测试用例中正确地注入 SynchronousExecutor。
如果被测代码依赖于 Executor 的其他特性（例如线程池大小、任务队列等），则可能需要使用更复杂的 Mock 对象来模拟 Executor 的行为。

总结：

通过使用 SynchronousExecutor 并将其注入到被测代码中，我们可以解决使用 Mockito 验证 Executor.execute 内部方法调用时遇到的线程问题。这种方法可以确保代码同步执行，从而使 Mockito 能够正确地验证方法调用。记住，这种方法只适用于需要验证特定方法调用，并且了解同步执行不会对测试产生负面影响的情况。

以上就是使用 Mockito 验证 Executor.execute 内部方法调用的方法的详细内容，更多请关注php中文网其它相关文章！