在Go语言中测试GRPC流式发送函数

huangapple
go评论118阅读模式
英文:

Testing around GRPC stream Send function in Go

问题

我有一个Go GRPC服务器端流式传输函数：

  1. func (server *Server) GetClients(req *iam.GetClientsRequest, client iam.IAM_GetClientsServer) error {
  2.     ctx := client.(interface{ Context() context.Context }).Context()
  4.     userID, err := getUserIDStream(client)
  5.     if err != nil {
  6.         return err
  7.     }
  9.     clients, err := server.db.QueryByUserID(ctx, userID)
  10.     if err != nil {
  11.         return grpc.Errorf(codes.Internal, apiutils.ServerError)
  12.     }
  14.     for _, value := range clients {
  15.         converted, err := server.fromInternalClient(value)
  16.         if err != nil {
  17.             return err
  18.         }
  20.         if err := client.Send(converted); err != nil {
  21.             return err
  22.         }
  23.     }
  25.     return nil
  26. }

我这样测试它：

  1. It("GetClients - Send fails - Error", func() {
  3.     handler := createHandler(db)
  4.     lis := bufconn.Listen(bufSize)
  5.     server := grpc.NewServer()
  7.     iam.RegisterIAMServer(server, NewServer(handler))
  8.     go func() {
  9.         if err := server.Serve(lis); err != nil {
  10.             log.Fatalf("Server exited with error: %v", err)
  11.         }
  12.     }()
  14.     defer lis.Close()
  15.     defer server.GracefulStop()
  17.     conn, err := grpc.DialContext(context.Background(), "bufnet",
  18.         grpc.WithContextDialer(createBufDialier(lis)), grpc.WithInsecure())
  19.     Expect(err).ShouldNot(HaveOccurred())
  20.     defer conn.Close()
  21.     client := iam.NewIAMClient(conn)
  23.     cclient, _ := client.GetClients(addAccessToken(context.Background()), new(iam.GetClientsRequest))
  24.     resp, err := cclient.Recv()
  26.     Expect(resp).Should(BeNil())
  27.     Expect(err).Should(HaveOccurred())
  28.     Expect(err.Error()).Should(Equal(message))
  29. })

我的问题是，我不确定如何在Send上引发失败，以便我可以测试响应。由于我正在使用实际的测试服务器和客户端，我不能只是模拟对象，而且我也不想走这条路。有没有办法可以做到这一点？

英文:

I have a Go GRPC server-side streaming function:

  1. func (server *Server) GetClients(req *iam.GetClientsRequest, client iam.IAM_GetClientsServer) error {
  2. 	ctx := client.(interface{ Context() context.Context }).Context()
  4. 	userID, err := getUserIDStream(client)
  5. 	if err != nil {
  6. 		return err
  7. 	}
  9. 	clients, err := server.db.QueryByUserID(ctx, userID)
  10. 	if err != nil {
  11. 		return grpc.Errorf(codes.Internal, apiutils.ServerError)
  12. 	}
  14. 	for _, value := range clients {
  15. 		converted, err := server.fromInternalClient(value)
  16. 		if err != nil {
  17. 			return err
  18. 		}
  20. 		if err := client.Send(converted); err != nil {
  21. 			return err
  22. 		}
  23. 	}
  25. 	return nil
  26. }

and I'm testing it like this:

  1. It("GetClients - Send fails - Error", func() {
  3. 	handler := createHandler(db)
  4. 	lis := bufconn.Listen(bufSize)
  5. 	server := grpc.NewServer()
  7. 	iam.RegisterIAMServer(server, NewServer(handler))
  8. 	go func() {
  9. 		if err := server.Serve(lis); err != nil {
  10. 			log.Fatalf("Server exited with error: %v", err)
  11. 		}
  12. 	}()
  14. 	defer lis.Close()
  15. 	defer server.GracefulStop()
  17.     conn, err := grpc.DialContext(context.Background(), "bufnet",
  18. 		grpc.WithContextDialer(createBufDialier(lis)), grpc.WithInsecure())
  19. 	Expect(err).ShouldNot(HaveOccurred())
  20. 	defer conn.Close()
  21. 	client := iam.NewIAMClient(conn)
  23. 	cclient, _ := client.GetClients(addAccessToken(context.Background()), new(iam.GetClientsRequest))
  24. 	resp, err := cclient.Recv()
  26. 	Expect(resp).Should(BeNil())
  27. 	Expect(err).Should(HaveOccurred())
  28. 	Expect(err.Error()).Should(Equal(message))
  29. })

My issue is that I'm not sure how to induce a failure on Send so I can test the response. Since I'm using an actual test server and client, I can't just mock out the object and I'd prefer not to go that route anyway. Is there a way I can do this?

答案1

得分: 0

最初，我试图通过将bufSize设置为一个人为较低的值来强制Send失败。然而，这并没有产生错误，所以我决定尝试修改服务器上的maxSendMessageSize：

  1. opts := []grpc.ServerOption{}
  2. if sendFails {
  3. 	opts = append(opts, grpc.MaxSendMsgSize(10))
  4. }
  6. lis := bufconn.Listen(bufSize)
  7. server := grpc.NewServer(opts...)

这样做确实导致了错误的发生。

英文:

Originally, I was trying to force Send to fail by setting bufSize to an artificially low value. However, this wasn't producing an error so I decided to try modifying the maxSendMessageSize on the server:

  1. opts := []grpc.ServerOption{}
  2. if sendFails {
  3. 	opts = append(opts, grpc.MaxSendMsgSize(10))
  4. }
  6. lis := bufconn.Listen(bufSize)
  7. server := grpc.NewServer(opts...)

And this worked in producing the error.

huangapple
  • 本文由 发表于 2022年10月21日 17:31:11
  • 转载请务必保留本文链接：https://go.coder-hub.com/74151616.html
  • go
  • grpc-go
  • testing
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定