rsc.io/go@v0.0.0-20150416155037-e040fd465409/src/net/conn_test.go (about)

     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  package net
     9  
    10  import (
    11  	"testing"
    12  	"time"
    13  )
    14  
    15  // someTimeout is used just to test that net.Conn implementations
    16  // don't explode when their SetFooDeadline methods are called.
    17  // It isn't actually used for testing timeouts.
    18  const someTimeout = 10 * time.Second
    19  
    20  func TestConnAndListener(t *testing.T) {
    21  	handler := func(ls *localServer, ln Listener) { transponder(t, ln) }
    22  	for _, network := range []string{"tcp", "unix", "unixpacket"} {
    23  		if !testableNetwork(network) {
    24  			t.Logf("skipping %s test", network)
    25  			continue
    26  		}
    27  
    28  		ls, err := newLocalServer(network)
    29  		if err != nil {
    30  			t.Fatalf("Listen failed: %v", err)
    31  		}
    32  		defer ls.teardown()
    33  		if err := ls.buildup(handler); err != nil {
    34  			t.Fatal(err)
    35  		}
    36  		if ls.Listener.Addr().Network() != network {
    37  			t.Fatalf("got %s; want %s", ls.Listener.Addr().Network(), network)
    38  		}
    39  
    40  		c, err := Dial(ls.Listener.Addr().Network(), ls.Listener.Addr().String())
    41  		if err != nil {
    42  			t.Fatalf("Dial failed: %v", err)
    43  		}
    44  		defer c.Close()
    45  		if c.LocalAddr().Network() != network || c.LocalAddr().Network() != network {
    46  			t.Fatalf("got %v->%v; want %v->%v", c.LocalAddr().Network(), c.RemoteAddr().Network(), network, network)
    47  		}
    48  		c.SetDeadline(time.Now().Add(someTimeout))
    49  		c.SetReadDeadline(time.Now().Add(someTimeout))
    50  		c.SetWriteDeadline(time.Now().Add(someTimeout))
    51  
    52  		if _, err := c.Write([]byte("CONN TEST")); err != nil {
    53  			t.Fatalf("Conn.Write failed: %v", err)
    54  		}
    55  		rb := make([]byte, 128)
    56  		if _, err := c.Read(rb); err != nil {
    57  			t.Fatalf("Conn.Read failed: %v", err)
    58  		}
    59  	}
    60  }
    61  
    62  func transponder(t *testing.T, ln Listener) {
    63  	switch ln := ln.(type) {
    64  	case *TCPListener:
    65  		ln.SetDeadline(time.Now().Add(someTimeout))
    66  	case *UnixListener:
    67  		ln.SetDeadline(time.Now().Add(someTimeout))
    68  	}
    69  	c, err := ln.Accept()
    70  	if err != nil {
    71  		t.Errorf("Listener.Accept failed: %v", err)
    72  		return
    73  	}
    74  	defer c.Close()
    75  
    76  	network := ln.Addr().Network()
    77  	if c.LocalAddr().Network() != network || c.LocalAddr().Network() != network {
    78  		t.Errorf("got %v->%v; expected %v->%v", c.LocalAddr().Network(), c.RemoteAddr().Network(), network, network)
    79  		return
    80  	}
    81  	c.SetDeadline(time.Now().Add(someTimeout))
    82  	c.SetReadDeadline(time.Now().Add(someTimeout))
    83  	c.SetWriteDeadline(time.Now().Add(someTimeout))
    84  
    85  	b := make([]byte, 128)
    86  	n, err := c.Read(b)
    87  	if err != nil {
    88  		t.Errorf("Conn.Read failed: %v", err)
    89  		return
    90  	}
    91  	if _, err := c.Write(b[:n]); err != nil {
    92  		t.Errorf("Conn.Write failed: %v", err)
    93  		return
    94  	}
    95  }