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

时间:2018-04-17 07:13:19

标签: java mockito grpc

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测试!!

2 个答案:

答案 0 :(得分:2)

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

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

修改:我们现在建议GrpcCleanupRule超过GrpcServerRule。您仍然可以参考hello world示例。

答案 1 :(得分:0)

这个想法是将响应和流观察者存根。

@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));
}