当前位置：首页 > 文章列表 > Golang > Go问答 > 正确的连接测试方法在go语言中是什么？

正确的连接测试方法在go语言中是什么？

来源：stackoverflow 2024-03-13 15:06:30 0浏览收藏

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《正确的连接测试方法在go语言中是什么？》，聊聊，希望可以帮助到正在努力赚钱的你。

问题内容

我正在通过测试覆盖项目，为此目的，我需要虚拟 tcp 服务器，它可以接受连接、向其写入/读取数据、关闭它等...我在堆栈溢出上发现了这个问题，涵盖了模拟连接，但它没有涵盖我实际需要测试的内容。

我的想法依赖于这篇文章作为起点，但是当我开始实现通道让服务器将一些数据写入新打开的连接时，我在写入通道时遇到了不可调试的死锁。

我想要实现的是将一些数据写入服务器的通道，例如sendingqueue chan *[]byte，这样稍后对应的[]byte将被发送到新建立的连接。

在这些小研究期间，我尝试在将数据发送到通道之前/之后调试和打印消息，并尝试在程序的不同位置从通道发送/读取数据。

我发现了什么：

如果我使用
直接在 handleconnection 中添加数据，我的想法就会起作用
```
go func() {
   f := []byte("test.")
   t.sendingqueue <- &f
}()
```
如果我以任何形式将数据从 testutils_testingtcpserver_writesrequest 推送到通道，无论是使用 func (t *tcpserver) put(data *[]byte) (err error) 或直接使用：
```
go func(queue chan *[]byte, data *[]byte) {
       queue <- data
}(t.sendingqueue, &payload)
```
通道是否缓冲并不重要。

所以，显然，我调试代码的方式有问题（我没有深入研究 cli dlv，仅使用 ide 调试器），或者我完全怀念使用 go 通道、goroutines 或 net 的方式.conn模块。

为了方便起见，我们提供了带有完整代码的公共要点。注意 - testutils_testingtcpserver_writesrequest 中有 // init 部分，这是运行/调试单个测试所必需的。在目录下运行go test时应将其注释掉。

utils.go：

    // newserver creates a new server using given protocol
    // and addr.
    func newtestingtcpserver(protocol, addr string) (*tcpserver, error) {
        switch strings.tolower(protocol) {
        case "tcp":
            return &tcpserver{
                addr:         addr,
                sendingqueue: make(chan *[]byte, 10),
            }, nil
        case "udp":
        }
        return nil, errors.new("invalid protocol given")
    }

    // tcpserver holds the structure of our tcp
    // implementation.
    type tcpserver struct {
        addr         string
        server       net.listener
        sendingqueue chan *[]byte
    }

    func (t *tcpserver) run() (err error) {}
    func (t *tcpserver) close() (err error) {}
    func (t *tcpserver) put(data *[]byte) (err error) {}
    func (t *tcpserver) handleconnection(conn net.conn){
        // <...>

        // putting data here successfully sends it via freshly established
        // connection:

        // go func() {
        //  f := []byte("test.")
        //  t.sendingqueue <- &f
        // }()
        for {
            fmt.printf("started for loop\n")
            select {
            case data := <-readerchannel:
                fmt.printf("read written data\n")
                writebuffer.write(*data)
                writebuffer.flush()
            case data := <-t.sendingqueue:
                fmt.printf("read pushed data\n")
                writebuffer.write(*data)
                writebuffer.flush()
            case <-ticker:
                fmt.printf("tick\n")
                return
            }
            fmt.printf("finished for loop\n")
        }
    }

utils_test.go

    func TestUtils_TestingTCPServer_WritesRequest(t *testing.T) {

        payload := []byte("hello world\n")

        // <...> In gist here is placed INIT piece, which
        // is required to debug single test

        fmt.Printf("Putting payload into queue\n")
        // This doesn't affect channel
        err = utilTestingSrv.Put(&payload)
        assert.Nil(t, err)

        // This doesn't work either
        //go func(queue chan *[]byte, data *[]byte) {
        //       queue <- data
        //}(utilTestingSrv.sendingQueue, &payload)

        conn, err := net.Dial("tcp", ":41123")
        if !assert.Nil(t, err) {
            t.Error("could not connect to server: ", err)
        }
        defer conn.Close()

        out := make([]byte, 1024)
        if _, err := conn.Read(out); assert.Nil(t, err) {
            // Need to remove trailing byte 0xa from bytes array to make sure bytes array are equal.
            if out[len(payload)] == 0xa {
                out[len(payload)] = 0x0
            }
            assert.Equal(t, payload, bytes.Trim(out, "\x00"))
        } else {
            t.Error("could not read from connection")
        }
    }