登录
首页 >  Golang >  Go问答

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

来源:stackoverflow

时间:2024-03-13 15:06:30 440浏览 收藏

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

问题内容

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

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

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

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

我发现了什么

  1. 如果我使用

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

    go func(queue chan *[]byte, data *[]byte) {
           queue <- data
    }(t.sendingqueue, &payload)
    
  3. 通道是否缓冲并不重要。

所以,显然,我调试代码的方式有问题(我没有深入研究 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")
        }
    }


解决方案


在同事的帮助下并阅读了 article 关于 init 如何工作的文章后,我发现了一个问题。

它位于 init 函数中,由于使用 := 分配,该函数正在重新创建额外的服务器。我还更新了代码以确保服务器在 net.Dialconn.Read 之前运行。

以上就是《正确的连接测试方法在go语言中是什么?》的详细内容,更多关于的资料请关注golang学习网公众号!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>