Function TestNAConn

naconn/conn_test.go:73–158

(t *testing.T)

}

func TestNAConn(t *testing.T) {
	Convey("Test dial timeout", t, func(c C) {
		// Register node with invalid IP address
		resolver := &simpleResolver{}
		nodeinfo := thisNode()
		So(nodeinfo, ShouldNotBeNil)
		resolver.registerNode(&proto.Node{
			Addr:      "240.0.0.1:8080",
			ID:        nodeinfo.ID,
			PublicKey: nodeinfo.PublicKey,
			Nonce:     nodeinfo.Nonce,
		})
		// Register resolver
		RegisterResolver(resolver)
		// Test dial timeout
		_, err := Dial(nodeinfo.ID)
		nerr, ok := errors.Cause(err).(net.Error)
		t.Logf("err: %v\n", err)
		So(ok, ShouldBeTrue)
		So(nerr.Timeout(), ShouldBeTrue)
	})
	Convey("Test simple NAConn", t, func(c C) {
		l, err := net.Listen("tcp", "localhost:0")
		So(err, ShouldBeNil)
		defer func() { _ = l.Close() }()
		// Register node
		resolver := &simpleResolver{}
		nodeinfo := thisNode()
		So(nodeinfo, ShouldNotBeNil)
		resolver.registerNode(&proto.Node{
			Addr:      l.Addr().String(),
			ID:        nodeinfo.ID,
			PublicKey: nodeinfo.PublicKey,
			Nonce:     nodeinfo.Nonce,
		})
		// Register resolver
		RegisterResolver(resolver)
		// Serve
		rounds := 100
		message := [1024]byte{}
		rand.Read(message[:])
		wg := &sync.WaitGroup{}
		wg.Add(1)
		go func(c C) {
			defer wg.Done()
			for i := 0; i < rounds; i++ {
				conn, err := l.Accept()
				c.So(err, ShouldBeNil)
				wg.Add(1)
				go func(c C, conn net.Conn) {
					defer wg.Done()
					naconn, err := Accept(conn)
					c.So(err, ShouldBeNil)
					defer func() { _ = naconn.Close() }()
					t.Logf("accept conn %s <- %s", naconn.LocalAddr(), naconn.RemoteAddr())
					buffer, err := ioutil.ReadAll(naconn)
					c.So(err, ShouldBeNil)
					c.So(buffer, ShouldResemble, message[:])