conn_test.go

Documentation: net

		 1  // Copyright 2012 The Go Authors. All rights reserved.
		 2  // Use of this source code is governed by a BSD-style
		 3  // license that can be found in the LICENSE file.
		 4  
		 5  // This file implements API tests across platforms and will never have a build
		 6  // tag.
		 7  
		 8  //go:build !js
		 9  // +build !js
		10  
		11  package net
		12  
		13  import (
		14  	"testing"
		15  	"time"
		16  )
		17  
		18  // someTimeout is used just to test that net.Conn implementations
		19  // don't explode when their SetFooDeadline methods are called.
		20  // It isn't actually used for testing timeouts.
		21  const someTimeout = 10 * time.Second
		22  
		23  func TestConnAndListener(t *testing.T) {
		24  	for i, network := range []string{"tcp", "unix", "unixpacket"} {
		25  		if !testableNetwork(network) {
		26  			t.Logf("skipping %s test", network)
		27  			continue
		28  		}
		29  
		30  		ls, err := newLocalServer(network)
		31  		if err != nil {
		32  			t.Fatal(err)
		33  		}
		34  		defer ls.teardown()
		35  		ch := make(chan error, 1)
		36  		handler := func(ls *localServer, ln Listener) { ls.transponder(ln, ch) }
		37  		if err := ls.buildup(handler); err != nil {
		38  			t.Fatal(err)
		39  		}
		40  		if ls.Listener.Addr().Network() != network {
		41  			t.Fatalf("got %s; want %s", ls.Listener.Addr().Network(), network)
		42  		}
		43  
		44  		c, err := Dial(ls.Listener.Addr().Network(), ls.Listener.Addr().String())
		45  		if err != nil {
		46  			t.Fatal(err)
		47  		}
		48  		defer c.Close()
		49  		if c.LocalAddr().Network() != network || c.RemoteAddr().Network() != network {
		50  			t.Fatalf("got %s->%s; want %s->%s", c.LocalAddr().Network(), c.RemoteAddr().Network(), network, network)
		51  		}
		52  		c.SetDeadline(time.Now().Add(someTimeout))
		53  		c.SetReadDeadline(time.Now().Add(someTimeout))
		54  		c.SetWriteDeadline(time.Now().Add(someTimeout))
		55  
		56  		if _, err := c.Write([]byte("CONN AND LISTENER TEST")); err != nil {
		57  			t.Fatal(err)
		58  		}
		59  		rb := make([]byte, 128)
		60  		if _, err := c.Read(rb); err != nil {
		61  			t.Fatal(err)
		62  		}
		63  
		64  		for err := range ch {
		65  			t.Errorf("#%d: %v", i, err)
		66  		}
		67  	}
		68  }
		69
View as plain text