Java测试中如何模拟gRPC响应的延迟？

当然可以实现，根据你使用的Mock方式，延迟逻辑的插入位置有所不同，以下是几种常见场景的实现方案：

1. Mock gRPC Stub（以Mockito为例） #

不管是同步阻塞Stub还是异步非阻塞Stub，都可以在Mock方法的响应逻辑中直接插入Thread.sleep()模拟延迟：

同步阻塞Stub示例 #

HelloServiceBlockingStub mockStub = Mockito.mock(HelloServiceBlockingStub.class);
when(mockStub.sayHello(any(HelloRequest.class)))
    .thenAnswer(invocation -> {
        // 模拟1秒处理延迟
        Thread.sleep(1000);
        return HelloResponse.newBuilder().setMessage("Hello").build();
    });

异步非阻塞Stub示例 #

HelloServiceStub asyncMockStub = Mockito.mock(HelloServiceStub.class);
doAnswer(invocation -> {
    // 先延迟再触发回调
    Thread.sleep(1000);
    StreamObserver<HelloResponse> responseObserver = invocation.getArgument(1);
    responseObserver.onNext(HelloResponse.newBuilder().setMessage("Hello").build());
    responseObserver.onCompleted();
    return null;
}).when(asyncMockStub).sayHello(any(HelloRequest.class), any(StreamObserver.class));

2. 使用gRPC测试服务器模拟延迟 #

如果需要更贴近真实服务的场景，可以启动一个本地测试gRPC服务器，在服务实现的方法中加入延迟：

public class TestHelloService extends HelloServiceGrpc.HelloServiceImplBase {
    @Override
    public void sayHello(HelloRequest request, StreamObserver<HelloResponse> responseObserver) {
        try {
            // 模拟2秒处理延迟
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
        responseObserver.onNext(HelloResponse.newBuilder().setMessage("Hello").build());
        responseObserver.onCompleted();
    }
}

// 启动测试服务器
Server testServer = ServerBuilder.forPort(8080)
    .addService(new TestHelloService())
    .build()
    .start();

3. 注意事项 #

• Thread.sleep()会阻塞当前线程，在异步测试场景中，要确保测试主线程不会提前结束，可以配合CountDownLatch等同步工具。
• 建议封装通用的延迟工具方法，避免重复代码：

private void simulateDelay(long millis) {
    try {
        Thread.sleep(millis);
    } catch (InterruptedException e) {
        Thread.currentThread().interrupt();
        throw new RuntimeException("延迟被中断", e);
    }
}

• 测试超时逻辑时，务必配置客户端的超时参数，比如设置请求截止时间：

ManagedChannel channel = ManagedChannelBuilder.forAddress("localhost", 8080)
    .usePlaintext()
    .defaultCallOptions(CallOptions.DEFAULT.withDeadlineAfter(500, TimeUnit.MILLISECONDS))
    .build();

内容的提问来源于stack exchange，提问作者Evgenii Golgovskikh