grpc自动化测试例子

[chatgpt] gRPC 是一个高性能、开源、通用的 RPC 框架，支持多种编程语言。在进行自动化测试时，可以使用 gRPC 提供的测试工具和库来方便地进行测试。

以下是一个使用 golang 编写的 gRPC 自动化测试示例，包括服务端和客户端代码：

服务端代码：

package main

import (
	"context"
	"log"
	"net"

	"google.golang.org/grpc"
	"google.golang.org/grpc/reflection"

	pb "github.com/example/grpc-demo/helloworld"
)

type server struct{}

func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) {
	log.Printf("Received: %v", in.GetName())
	return &pb.HelloReply{Message: "Hello " + in.GetName()}, nil
}

func main() {
	lis, err := net.Listen("tcp", ":50051")
	if err != nil {
		log.Fatalf("failed to listen: %v", err)
	}
	s := grpc.NewServer()
	pb.RegisterGreeterServer(s, &server{})
	reflection.Register(s)
	if err := s.Serve(lis); err != nil {
		log.Fatalf("failed to serve: %v", err)
	}
}

客户端代码：

package main

import (
	"context"
	"log"
	"time"

	"google.golang.org/grpc"

	pb "github.com/example/grpc-demo/helloworld"
)

const (
	address = "localhost:50051"
	defaultName = "world"
)

func main() {
	conn, err := grpc.Dial(address, grpc.WithInsecure(), grpc.WithBlock())
	if err != nil {
		log.Fatalf("did not connect: %v", err)
	}
	defer conn.Close()
	c := pb.NewGreeterClient(conn)

	name := defaultName
	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
	defer cancel()
	r, err := c.SayHello(ctx, &pb.HelloRequest{Name: name})
	if err != nil {
		log.Fatalf("could not greet: %v", err)
	}
	log.Printf("Greeting: %s", r.GetMessage())
}

该示例包含一个简单的 gRPC 服务端和客户端，客户端向服务端发送 "Hello" 请求，并接收响应。可以使用 golang 的 testing 框架编写测试用例，例如：

package main

import (
	"context"
	"testing"
	"time"

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

	pb "github.com/example/grpc-demo/helloworld"
)

func TestSayHello(t *testing.T) {
	conn, err := grpc.Dial(address, grpc.WithInsecure(), grpc.WithBlock())
	if err != nil {
		t.Fatalf("did not connect: %v", err)
	}
	defer conn.Close()
	c := pb.NewGreeterClient(conn)

	name := defaultName
	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
	defer cancel()
	r, err := c.SayHello(ctx, &pb.HelloRequest{Name: name})
	if err != nil {
		t.Fatalf("could not greet: %v", err)
	}
	assert.Equal(t, "Hello "+name, r.GetMessage(), "response message is not correct")
}

该测试用例使用了第三方的断言库 testify/assert，可以方便地进行断言。测试用例使用了 gRPC 客户端来向服务端发送请求，并对响应结果进行断言，验证服务端的功能是否正常。

备注：这是ChatGPT3.5 对grpc怎么做自动化测试的回答，感觉挺智能的，记录一下