使用 Go 进行 gRPC 单元测试中的 UnaryHandler 欺骗

一分耕耘，一分收获！既然打开了这篇文章《使用 Go 进行 gRPC 单元测试中的 UnaryHandler 欺骗》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

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

我有一个具有以下签名的函数 interceptor：

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 (type func(context.context, *authdata) (*token, error)) 作为类型 grpc.unaryhandler

我继续编写了一个正确满足以下条件的虚拟函数：

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

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

最可能的解释是 grpc unaryhandler 没有做我想象的那样，但是它做了什么？

解决方案

不，函数

gettoken(ctx context.context, req *authdata) (*token, error)

与

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

在 gettoken 中，第二个参数 req 的类型为 *authdata，而在 unaryhandler req 中，第二个参数 interface{} 的类型。返回的 *token 与 interface{} 不是同一类型。这就是为什么你不能直接将 gettoken 传递给拦截器的原因。

在您的 grpc 服务中，您可以编写类似的方法

gettoken(ctx context.context, req *authdata) (*token, error)

作为处理程序来完成您的服务器工作。然而，它并不像人们想象的那样是 unaryhandler。

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

type xxxserver interface {
    gettoken(ctx context.context, req *authdata) (*token, error)
}

您可以看到您的处理程序满足的是这个接口（而不是 unaryhander）。

在幕后，如果您查看生成的 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 接口，然后调用您的处理程序（您的代码）。这显示了 interceptor 是如何调用的。

到这里，我们也就讲完了《使用 Go 进行 gRPC 单元测试中的 UnaryHandler 欺骗》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！