如何使用mockito测试和模拟用Java编写的GRPC服务

Question

protobuf的定义如下：

syntax = "proto3";

package helloworld;

// The greeting service definition.
service Greeter {
  // Sends a greeting
  rpc SayHello (HelloRequest) returns (HelloReply) {}
}

// The request message containing the user's name.
message HelloRequest {
  string name = 1;
}

// The response message containing the greetings
message HelloReply {
  string message = 1;
}

我需要使用mockito和junit测试!!

Answer 1

我们的想法是将响应和流观察者存根。

@Test
public void shouldTestGreeterService() throws Exception {

    Greeter service = new Greeter();

    HelloRequest req = HelloRequest.newBuilder()
            .setName("hello")
            .build();

    StreamObserver<HelloRequest> observer = mock(StreamObserver.class);

    service.sayHello(req, observer);

    verify(observer, times(1)).onCompleted();

    ArgumentCaptor<HelloReply> captor = ArgumentCaptor.forClass(HelloReply.class);

    verify(observer, times(1)).onNext(captor.capture());

    HelloReply response = captor.getValue();

    assertThat(response.getStatus(), is(true));
}

Answer 2

鼓励测试服务的方法是使用进程内传输和普通存根。然后，您可以像平常一样与服务进行通信，而无需进行大量的嘲笑。过度使用的模拟会产生脆弱的测试，这些测试不会对测试的代码产生信心。

GrpcServerRule在幕后使用进程内传输。我们现在建议从hello world开始查看示例的测试。

编辑：我们现在推荐GrpcCleanupRule超过GrpcServerRule。您仍然可以参考hello world示例。