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