在单元测试中，我们经常需要验证某个方法是否被调用，以及调用时传入的参数是否正确。当被验证的方法是在 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 中。 在测试用例中，我们可以这样做：
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 能够正确地验证方法调用。 记住，这种方法只适用于需要验证特定方法调用，并且了解同步执行不会对测试产生负面影响的情况。
