对 Go 中的 GRPC 流发送功能进行测试评估

各位小伙伴们，大家好呀！看看今天我又给各位带来了什么文章？本文标题是《对 Go 中的 GRPC 流发送功能进行测试评估》，很明显是关于Golang的文章哈哈哈，其中内容主要会涉及到等等，如果能帮到你，觉得很不错的话，欢迎各位多多点评和分享！

我有一个go grpc服务器端流功能：

func (server *server) getclients(req *iam.getclientsrequest, client iam.iam_getclientsserver) error {
    ctx := client.(interface{ context() context.context }).context()

    userid, err := getuseridstream(client)
    if err != nil {
        return err
    }

    clients, err := server.db.querybyuserid(ctx, userid)
    if err != nil {
        return grpc.errorf(codes.internal, apiutils.servererror)
    }

    for _, value := range clients {
        converted, err := server.frominternalclient(value)
        if err != nil {
            return err
        }

        if err := client.send(converted); err != nil {
            return err
        }
    }

    return nil
}

我正在这样测试它：

It("GetClients - Send fails - Error", func() {

    handler := createHandler(db)
    lis := bufconn.Listen(bufSize)
    server := grpc.NewServer()

    iam.RegisterIAMServer(server, NewServer(handler))
    go func() {
        if err := server.Serve(lis); err != nil {
            log.Fatalf("Server exited with error: %v", err)
        }
    }()

    defer lis.Close()
    defer server.GracefulStop()

    conn, err := grpc.DialContext(context.Background(), "bufnet",
        grpc.WithContextDialer(createBufDialier(lis)), grpc.WithInsecure())
    Expect(err).ShouldNot(HaveOccurred())
    defer conn.Close()
    client := iam.NewIAMClient(conn)

    cclient, _ := client.GetClients(addAccessToken(context.Background()), new(iam.GetClientsRequest))
    resp, err := cclient.Recv()

    Expect(resp).Should(BeNil())
    Expect(err).Should(HaveOccurred())
    Expect(err.Error()).Should(Equal(message))
})

我的问题是，我不确定如何在 send 上引发故障，以便我可以测试响应。由于我使用的是实际的测试服务器和客户端，因此我不能只是模拟该对象，而且我不想走这条路。有什么办法可以做到这一点吗？

正确答案

最初，我试图通过将 bufsize 设置为人为的低值来强制 send 失败。但是，这并没有产生错误，因此我决定尝试修改服务器上的 maxsendmessagesize：

opts := []grpc.ServerOption{}
if sendFails {
    opts = append(opts, grpc.MaxSendMsgSize(10))
}

lis := bufconn.Listen(bufSize)
server := grpc.NewServer(opts...)

这导致了错误的产生。

本篇关于《对 Go 中的 GRPC 流发送功能进行测试评估》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！