naza/pkg/connection/connection_test.go

package connection

import (
	"github.com/q191201771/nezha/pkg/assert"
	"github.com/q191201771/nezha/pkg/nazalog"
	"net"
	"testing"
)

// TODO chef: 补充单元测试

func TestWriteTimeout(t *testing.T) {
	// 开启一个 tcp 服务器，只accept一个连接，之后对这个连接不做任何读写
	// 使用 Connection 设置写超时后，死循环往服务器发送数据
	ch := make(chan struct{}, 1)
	l, err := net.Listen("tcp", ":10027")
	assert.Equal(t, nil, err)
	defer l.Close()
	go func() {
		conn, _ := l.Accept()
		defer conn.Close()
		<-ch
	}()
	conn, err := net.Dial("tcp", ":10027")
	c := New(conn, func(opt *Option) {
		opt.WriteTimeoutMS = 1000
	})
	assert.Equal(t, nil, err)
	b := make([]byte, 128)
	for {
		n, err := c.Write(b)
		nazalog.Infof("%d %+v", n, err)
		if err != nil {
			break
		}
	}
	ch <- struct{}{}
}