如何在go中对gRPC服务进行单元测试

Question

我在go中对gRPC服务进行单元测试时遇到困难。

我看了一下测试gRPC服务，但它对我不起作用，不确定我做错了什么。

Add方法的gRPC服务实现：

type server struct{}

func main() {
    listener, err := net.Listen("tcp", ":4040")
    if err != nil {
      panic(err)
    }

    srv := grpc.NewServer()
    service.RegisterMathOpServiceServer(srv, &server{})
    reflection.Register(srv)

    if e := srv.Serve(listener); e != nil {
          panic(e)
    }

    fmt.Println("Service is running on localhost:4040")
}

func (s *server) Add(ctx context.Context, request *service.AddRequest) 
(*service.AddResponse, error) {
    a, b := request.GetX(), request.GetY()
    result := a + b

    return &service.AddResponse{Answer: result}, nil
}

Add方法的单元测试：

package test

import (
    "context"
    pb "mathOp_service/src/service"
    "testing"
)
func TestAddition(t *testing.T) {
    s := server{}   <-- error occurs here
    req := pb.AddRequest{X: int64(1), Y: int64(2)}
    res, err := s.Add(context.Background(), req)

    if res.Answer != 3 {
        t.Error("Expected 1 + 2 to equal 3")
    }
}

项目目录结构

s:=server{}抛出一个错误，称为undefined:server。

我对Go非常陌生，找不到解决方法。

Answer 1

看看您的代码组织，您似乎来自Java背景，将src和test放在各自的文件夹中。当我开始写围棋的时候，我也遇到了同样的问题。我学到的是，最好将src和test放在同一个包和文件夹中，因此，要使代码习惯化，它还可以解析同一个包中的任何变量/方法引用。所以在测试中，您只需要导入pb包。

我也会鼓励你们看看https://github.com/golang-standards/project-layout用于Golang中的各种代码组织技术。