欺骗grpc UnaryHandler用于Go中的gRPC单元测试

Question

我正在努力提高Go gRPC服务器的覆盖率，但我在为服务器的拦截器功能编写测试时遇到了麻烦，因为我无法有意义地满足UnaryHandler类型。

我有一个带有以下签名的函数拦截器：

Interceptor func(
  ctx context.Context,
  req interface{},
  info *grpc.UnaryServerInfo,
  handler grpc.UnaryHandler, // <- my issue comes from here
) (interface{}, error)

我假设任何gRPC方法都会满足UnaryHandler的签名：

type UnaryHandler func(ctx context.Context, req interface{}) (interface{}, error)

所以我尝试传入一个具有此签名的方法：

GetToken(ctx context.Context, req *AuthData) (*Token, error)

我想象这会起作用，因为这是拦截器实际正在做的事情（转发RPC），但出于某种原因Go抱怨道：

不能使用AuthService。GetToken（类型func（上下文。上下文，*AuthData）（*Token，错误））作为grpc类型。UnaryHandler在Interceptor的参数中

我继续写了一个虚拟函数，它正确地满足：

func genericHandler(ctx context.Context, req interface{}) (interface{}, error) {
    return req, nil
}

这很好，因为在测试拦截器时，我并不特别需要运行任何特定的方法。然而，我很好奇，为什么实际的方法不满足约束条件，因为（根据我的理解）每当我在野外调用RPC时，它都会被传递给hood下的拦截器函数。

最有可能的解释是，grpc UnaryHandler不做我认为它做的事情，但它做什么呢？

Answer 1

不，是功能

GetToken(ctx context.Context, req *AuthData) (*Token, error)

不是和

type UnaryHandler func(ctx context.Context, req interface{}) (interface{}, error)

在GetToken中，第二个参数req的类型为*AuthData，而在UnaryHandlerreq的类型为接口{}。返回的*Token与接口{}不是同一类型。这就是您无法将GetToken直接传递给拦截器的原因。

在grpc服务中，您可以编写如下方法

GetToken(ctx context.Context, req *AuthData) (*Token, error)

作为处理程序来完成您的服务器工作。但是，它不是人们可能认为的UnaryHandler。

大部分转换由grpc/protobuf代码生成器完成。根据您的原型定义，它生成一个接口，如下所示：

type XXXServer interface {
    GetToken(ctx context.Context, req *AuthData) (*Token, error)
}

您可以看到，您的处理程序满足的是这个接口（而不是非惯用者）。

在引擎盖下，如果你看一下生成的xxx.pb.go文件，你会发现一些实际上正在做处理程序工作的_XXX_GetToken_Handler。在这个函数中，定义了一个（实际的）UnaryHandler，如下所示：

func _XXX_GetToken_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
    // skip other preparations...
    // 
    handler := func(ctx context.Context, req interface{}) (interface{}, error) {
        return srv.(XXXServer).GetToken(ctx, req.(*AuthData))
    }
    return interceptor(ctx, in, info, handler)
}

在这个UnaryHandler中，它将把您的服务器强制转换到XXXServer界面，然后调用您的处理程序（您的代码）。这显示了如何调用拦截器。