vendor/golang.org/x/net/nettest/conntest_test.go

   1 // Copyright 2016 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 // +build go1.8
   6
   7 package nettest
   8
   9 import (
  10         "net"
  11         "os"
  12         "runtime"
  13         "testing"
  14
  15         "golang.org/x/net/internal/nettest"
  16 )
  17
  18 func TestTestConn(t *testing.T) {
  19         tests := []struct{ name, network string }{
  20                 {"TCP", "tcp"},
  21                 {"UnixPipe", "unix"},
  22                 {"UnixPacketPipe", "unixpacket"},
  23         }
  24
  25         for _, tt := range tests {
  26                 t.Run(tt.name, func(t *testing.T) {
  27                         if !nettest.TestableNetwork(tt.network) {
  28                                 t.Skipf("not supported on %s", runtime.GOOS)
  29                         }
  30
  31                         mp := func() (c1, c2 net.Conn, stop func(), err error) {
  32                                 ln, err := nettest.NewLocalListener(tt.network)
  33                                 if err != nil {
  34                                         return nil, nil, nil, err
  35                                 }
  36
  37                                 // Start a connection between two endpoints.
  38                                 var err1, err2 error
  39                                 done := make(chan bool)
  40                                 go func() {
  41                                         c2, err2 = ln.Accept()
  42                                         close(done)
  43                                 }()
  44                                 c1, err1 = net.Dial(ln.Addr().Network(), ln.Addr().String())
  45                                 <-done
  46
  47                                 stop = func() {
  48                                         if err1 == nil {
  49                                                 c1.Close()
  50                                         }
  51                                         if err2 == nil {
  52                                                 c2.Close()
  53                                         }
  54                                         ln.Close()
  55                                         switch tt.network {
  56                                         case "unix", "unixpacket":
  57                                                 os.Remove(ln.Addr().String())
  58                                         }
  59                                 }
  60
  61                                 switch {
  62                                 case err1 != nil:
  63                                         stop()
  64                                         return nil, nil, nil, err1
  65                                 case err2 != nil:
  66                                         stop()
  67                                         return nil, nil, nil, err2
  68                                 default:
  69                                         return c1, c2, stop, nil
  70                                 }
  71                         }
  72
  73                         TestConn(t, mp)
  74                 })
  75         }
  76 }