在 golang 上测试双向 RPC

在 Go 语言中测试双向 RPC 时，编写测试用例至关重要，但缺乏测试经验可能会带来挑战。本文提供了针对双向 RPC 客户端连接、流创建和数据接收的测试用例示例，并探讨了扩展测试以覆盖更多场景的策略。通过创建具有清晰、可读、可理解性的单独测试，可以有效地隔离问题并确保应用程序的可靠性。

我正在遵循一些创建双向 grpc 客户端和服务器的教程。客户端将传递一些值，当服务器上的最后一个最大值发生更改时，它将以当前最大值响应客户端。最后，我想写下一些测试用例，但我没有测试场景的经验，这就是为什么我不确定我是否做了正确的事情。

func TestClientConnection(t *testing.T) {

    creds, _ := credentials.NewClientTLSFromFile("../server-cert.pem", "")
    conn, err := grpc.Dial(address, grpc.WithTransportCredentials(creds))
    if err != nil {
        t.Error("Had problem with connection, NOT PASSED")
    }
    defer conn.Close()

    c := proto.NewHerdiusServerClient(conn)

    stream, err := c.CheckMax(context.Background())
    if err != nil {
        t.Error("Had problem with stream, NOT PASSED")
        return
    }

    err = stream.Send(&proto.MaxRequest{Val: int32(10)})
    err = stream.Send(&proto.MaxRequest{Val: int32(12)})
    err = stream.Send(&proto.MaxRequest{Val: int32(13)})
    err = stream.Send(&proto.MaxRequest{Val: int32(9)})

    if err != nil {
        t.Error("Had problem with stream, NOT PASSED")
        return
    }

    return
}

现在，当我使用 go test 测试此场景时，它通过了，但我还想测试是否从服务器端收到了某些内容。

我的第二个问题是，如果我想将此测试扩展到不同的场景，例如检查服务器是否已连接或流是否已连接或从服务器端收到响应，我该怎么做？我应该创建另一个类来检索连接和流并在测试函数上使用吗？

解决方案

创建一个带有超时context.WithTimeout的上下文，并在发送数据后在流上调用Recv。检查您是否在超时时间内收到任何内容。

具体细节取决于此处的协议 - 如果您必须同时发送服务器数据，您可能需要 Recv 的 goroutine。

关于你的第二个问题，Go 的哲学是为每个场景提供清晰、明确、可读的测试。如果某些代码重复也没关系。更重要的是，每个单独的测试都具有可读性和可理解性。在测试非常重复的情况下，应该使用 table driven tests，但在您描述的情况下，这对我来说听起来像是单独的测试。

进行“构建”功能的测试很有用。一个测试测试连接，另一个测试连接和发送，另一个测试连接和发送和接收。这样，当测试失败时，通过单独重新运行它们，您甚至可以在查看测试代码之前快速隔离问题。

理论要掌握，实操不能落！以上关于《在 golang 上测试双向 RPC》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！