libgo/go/net/timeout_test.go

   1 // Copyright 2009 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 package net
   6
   7 import (
   8         "fmt"
   9         "runtime"
  10         "testing"
  11         "time"
  12 )
  13
  14 func testTimeout(t *testing.T, network, addr string, readFrom bool) {
  15         fd, err := Dial(network, addr)
  16         if err != nil {
  17                 t.Errorf("dial %s %s failed: %v", network, addr, err)
  18                 return
  19         }
  20         defer fd.Close()
  21         what := "Read"
  22         if readFrom {
  23                 what = "ReadFrom"
  24         }
  25
  26         errc := make(chan error, 1)
  27         go func() {
  28                 t0 := time.Now()
  29                 fd.SetReadDeadline(time.Now().Add(100 * time.Millisecond))
  30                 var b [100]byte
  31                 var n int
  32                 var err1 error
  33                 if readFrom {
  34                         n, _, err1 = fd.(PacketConn).ReadFrom(b[0:])
  35                 } else {
  36                         n, err1 = fd.Read(b[0:])
  37                 }
  38                 t1 := time.Now()
  39                 if n != 0 || err1 == nil || !err1.(Error).Timeout() {
  40                         errc <- fmt.Errorf("fd.%s on %s %s did not return 0, timeout: %v, %v", what, network, addr, n, err1)
  41                         return
  42                 }
  43                 if dt := t1.Sub(t0); dt < 50*time.Millisecond || dt > 250*time.Millisecond {
  44                         errc <- fmt.Errorf("fd.%s on %s %s took %s, expected 0.1s", what, network, addr, dt)
  45                         return
  46                 }
  47                 errc <- nil
  48         }()
  49         select {
  50         case err := <-errc:
  51                 if err != nil {
  52                         t.Error(err)
  53                 }
  54         case <-time.After(1 * time.Second):
  55                 t.Errorf("%s on %s %s took over 1 second, expected 0.1s", what, network, addr)
  56         }
  57 }
  58
  59 func TestTimeoutUDP(t *testing.T) {
  60         if runtime.GOOS == "plan9" {
  61                 return
  62         }
  63         testTimeout(t, "udp", "127.0.0.1:53", false)
  64         testTimeout(t, "udp", "127.0.0.1:53", true)
  65 }
  66
  67 func TestTimeoutTCP(t *testing.T) {
  68         if runtime.GOOS == "plan9" {
  69                 return
  70         }
  71         // set up a listener that won't talk back
  72         listening := make(chan string)
  73         done := make(chan int)
  74         go runServe(t, "tcp", "127.0.0.1:0", listening, done)
  75         addr := <-listening
  76
  77         testTimeout(t, "tcp", addr, false)
  78         <-done
  79 }