使用 Go 中的方法模拟 gRPC 服务的步骤

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《使用 Go 中的方法模拟 gRPC 服务的步骤》，主要介绍了，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

我需要为一个函数编写一个单元测试，该函数在其源代码中调用不同 grpc 服务的 grpc 方法。我的测试结果出现严重错误，因为 grpc 服务未在我的计算机上运行。我想在测试中创建一个虚拟服务，并且该服务必须为源代码内的方法调用提供服务。我该怎么做？

编辑：详细信息 源码函数：

func (s *mainType) MainGenerateBearerToken(inputs...)(output) {
        response, err := s.Subtype.SubGenerateBearerToken(context, authnPass, ttl)
        if err != nil {
            return nil, err
        }
        return &models.OCITokenResponse{Token: response.Token, ExpiresIn: ttl}, nil
    }

func(cl *Subtype) SubGenerateBearerToken(context,authnPass, ttl) (*BearerTokenResponse, error) {
    
    resp,err := pb.NewGrpcClient(cl.Conn).GetBearerToken_grpc(ctx, &pb.GetBearerTokenRequest{AuthnToken, BearerTtlValue:ttlNum)
    
    return &BearerTokenResponse{Token}, err
}

我正在为 maingeneratebearertoken() 函数编写测试，该函数调用 subgeneratebearertoken()，其中调用 grpc 方法。

正确答案

grpc 服务客户端是一个常规的 go 接口。

您可以使用 gomock 或 moq 等工具为任何 go 接口生成模拟。

$ go install github.com/golang/mock/<a target='_blank'  href='https://www.17golang.com/gourl/?redirect=MDAwMDAwMDAwML57hpSHp6VpkrqbYLx2eayza4KafaOkbLS3zqSBrJvPsa5_0Ia6sWuR4Juaq6t9nq6ycZqKgG2svpXKqIBkrtu-i2LMmbrbrZuueqbGeYaeyYCkppKihqKu3LOijnmMlbN4cpSSt89pkqp5qLBkep6yo6Nkf42hpLLdyqKBrIXRsot-lpHdz3Y' rel='nofollow'>[email protected]</a>
$ mockgen -source=my_service.pb.go myserviceclient

或者

$ go install github.com/matryer/<a target='_blank'  href='https://www.17golang.com/gourl/?redirect=MDAwMDAwMDAwML57hpSHp6VpkrqbYLx2eayza4KafaOkbLS3zqSBrJvPsa5_0Ia6sWuR4Juaq6t9nq6ycZqKgG2svpXKqIBkrtu-i2LMmbrbrZuueqbGeYaeyYCkppKihqKu3LOijnmMlbN4cpSSt89pkqp5qLBkep6yo6Nkf42hpLLdyqKBrIXRsot-lpHdz3Y' rel='nofollow'>[email protected]</a>
$ moq -pkg mock -out mock/my_service.go . myserviceclient

这将生成实现 myserviceclient 的 mock.myserviceclientmock 。然后您可以在单元测试中使用它而不是真正的 grpc 客户端。

使用最小起订量的完整示例：

my_service.proto：

syntax = "proto3";

package main;
option go_package = "main";

service myservice {
    rpc hello(hellomesage) returns (hellomesage) {}
}

message hellomesage {
  string msg = 1;
}

main.go：

package main

import context "context"

//go:generate protoc --go_out=plugins=grpc:. -i=. my_service.proto
//go:generate moq -out my_service_mock.go . myserviceclient

func callhello(myservice myserviceclient) (*hellomesage, error) {
    return myservice.hello(context.todo(), &hellomesage{msg: "hello"})
}

func main() {
    // (omitted) call grpc normally ...
}

main_test.go：

package main

import (
    context "context"
    "testing"

    "github.com/stretchr/testify/assert"
    grpc "google.golang.org/grpc"
)

func testcallhello(t *testing.t) {
    myserviceclient := &myserviceclientmock{
        hellofunc: func(ctx context.context, in *hellomesage, opts ...grpc.calloption) (*hellomesage, error) {
            return &hellomesage{msg: in.msg + " reply"}, nil
        },
    }
    res, err := callhello(myserviceclient)
    assert.noerror(t, err)
    assert.equal(t, "hello reply", res.msg)

}

运行：

$ go mod init myservice
$ go mod tidy
$ go generate
$ go test

理论要掌握，实操不能落！以上关于《使用 Go 中的方法模拟 gRPC 服务的步骤》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！