如何在 Golang 的单元测试中测试 net.Conn？

知识点掌握了，还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战，手把手教大家学习《如何在 Golang 的单元测试中测试 net.Conn？》，在实现功能的过程中也带大家重新温习相关知识点，温故而知新，回头看看说不定又有不一样的感悟！

问题内容

我目前正在考虑为 Go 中的 net.Conn 接口以及在该功能之上构建的其他功能创建一些单元测试，我想知道在 Google Go 中进行单元测试的最佳方法是什么？我的代码如下所示：

conn, _:=net.Dial("tcp", "127.0.0.1:8080")
...
fmt.Fprintf(conn, "test")
...
buffer:=make([]byte, 100)
conn.Read(buffer)

测试此代码和使用这些函数来启动单独的 goroutine 以像服务器一样运行的代码的最有效方法是使用net.http.httptest 包还是其他方式？

正确答案

尽管这取决于您的特定案例的实现细节，但一般方法是启动一个服务器（在一个单独的 goroutine 中，正如您已经暗示的那样），并监听传入的连接。

例如，让我们启动一个服务器并验证我们从连接中读取的内容确实是我们从客户端发送的内容：

func TestConn(t *testing.T) {
    message := "Hi there!\n"

    go func() {
        conn, err := net.Dial("tcp", ":3000")
        if err != nil {
            t.Fatal(err)
        }
        defer conn.Close()

        if _, err := fmt.Fprintf(conn, message); err != nil {
            t.Fatal(err)
        }
    }()

    l, err := net.Listen("tcp", ":3000")
    if err != nil {
        t.Fatal(err)
    }
    defer l.Close()
    for {
        conn, err := l.Accept()
        if err != nil {
            return
        }
        defer conn.Close()

        buf, err := ioutil.ReadAll(conn)
        if err != nil {
            t.Fatal(err)
        }

        fmt.Println(string(buf[:]))
        if msg := string(buf[:]); msg != message {
            t.Fatalf("Unexpected message:\nGot:\t\t%s\nExpected:\t%s\n", msg, message)
        }
        return // Done
    }

}

请注意，这里我不是在 goroutine 中启动服务器，否则测试用例很可能在侦听器运行测试之前完成。

以上就是《如何在 Golang 的单元测试中测试 net.Conn？》的详细内容，更多关于golang的资料请关注golang学习网公众号！