Commit ea1f7b83 authored by Alex Brainman's avatar Alex Brainman

net: change server_test.go so we could see failure messages

R=golang-dev, dave
CC=golang-dev
https://golang.org/cl/7323051
parent 928fe516
...@@ -113,8 +113,7 @@ func TestStreamConnServer(t *testing.T) { ...@@ -113,8 +113,7 @@ func TestStreamConnServer(t *testing.T) {
case "tcp", "tcp4", "tcp6": case "tcp", "tcp4", "tcp6":
_, port, err := SplitHostPort(taddr) _, port, err := SplitHostPort(taddr)
if err != nil { if err != nil {
t.Errorf("SplitHostPort(%q) failed: %v", taddr, err) t.Fatalf("SplitHostPort(%q) failed: %v", taddr, err)
return
} }
taddr = tt.caddr + ":" + port taddr = tt.caddr + ":" + port
} }
...@@ -169,11 +168,11 @@ func TestSeqpacketConnServer(t *testing.T) { ...@@ -169,11 +168,11 @@ func TestSeqpacketConnServer(t *testing.T) {
} }
func runStreamConnServer(t *testing.T, net, laddr string, listening chan<- string, done chan<- int) { func runStreamConnServer(t *testing.T, net, laddr string, listening chan<- string, done chan<- int) {
defer close(done)
l, err := Listen(net, laddr) l, err := Listen(net, laddr)
if err != nil { if err != nil {
t.Errorf("Listen(%q, %q) failed: %v", net, laddr, err) t.Errorf("Listen(%q, %q) failed: %v", net, laddr, err)
listening <- "<nil>" listening <- "<nil>"
done <- 1
return return
} }
defer l.Close() defer l.Close()
...@@ -188,13 +187,14 @@ func runStreamConnServer(t *testing.T, net, laddr string, listening chan<- strin ...@@ -188,13 +187,14 @@ func runStreamConnServer(t *testing.T, net, laddr string, listening chan<- strin
} }
rw.Write(buf[0:n]) rw.Write(buf[0:n])
} }
done <- 1 close(done)
} }
run: run:
for { for {
c, err := l.Accept() c, err := l.Accept()
if err != nil { if err != nil {
t.Logf("Accept failed: %v", err)
continue run continue run
} }
echodone := make(chan int) echodone := make(chan int)
...@@ -203,14 +203,12 @@ run: ...@@ -203,14 +203,12 @@ run:
c.Close() c.Close()
break run break run
} }
done <- 1
} }
func runStreamConnClient(t *testing.T, net, taddr string, isEmpty bool) { func runStreamConnClient(t *testing.T, net, taddr string, isEmpty bool) {
c, err := Dial(net, taddr) c, err := Dial(net, taddr)
if err != nil { if err != nil {
t.Errorf("Dial(%q, %q) failed: %v", net, taddr, err) t.Fatalf("Dial(%q, %q) failed: %v", net, taddr, err)
return
} }
defer c.Close() defer c.Close()
c.SetReadDeadline(time.Now().Add(1 * time.Second)) c.SetReadDeadline(time.Now().Add(1 * time.Second))
...@@ -220,14 +218,12 @@ func runStreamConnClient(t *testing.T, net, taddr string, isEmpty bool) { ...@@ -220,14 +218,12 @@ func runStreamConnClient(t *testing.T, net, taddr string, isEmpty bool) {
wb = []byte("StreamConnClient by Dial\n") wb = []byte("StreamConnClient by Dial\n")
} }
if n, err := c.Write(wb); err != nil || n != len(wb) { if n, err := c.Write(wb); err != nil || n != len(wb) {
t.Errorf("Write failed: %v, %v; want %v, <nil>", n, err, len(wb)) t.Fatalf("Write failed: %v, %v; want %v, <nil>", n, err, len(wb))
return
} }
rb := make([]byte, 1024) rb := make([]byte, 1024)
if n, err := c.Read(rb[0:]); err != nil || n != len(wb) { if n, err := c.Read(rb[0:]); err != nil || n != len(wb) {
t.Errorf("Read failed: %v, %v; want %v, <nil>", n, err, len(wb)) t.Fatalf("Read failed: %v, %v; want %v, <nil>", n, err, len(wb))
return
} }
// Send explicit ending for unixpacket. // Send explicit ending for unixpacket.
...@@ -333,8 +329,7 @@ func TestDatagramPacketConnServer(t *testing.T) { ...@@ -333,8 +329,7 @@ func TestDatagramPacketConnServer(t *testing.T) {
case "udp", "udp4", "udp6": case "udp", "udp4", "udp6":
_, port, err := SplitHostPort(taddr) _, port, err := SplitHostPort(taddr)
if err != nil { if err != nil {
t.Errorf("SplitHostPort(%q) failed: %v", taddr, err) t.Fatalf("SplitHostPort(%q) failed: %v", taddr, err)
return
} }
taddr = tt.caddr + ":" + port taddr = tt.caddr + ":" + port
tt.caddr += ":0" tt.caddr += ":0"
...@@ -397,14 +392,12 @@ func runDatagramConnClient(t *testing.T, net, laddr, taddr string, isEmpty bool) ...@@ -397,14 +392,12 @@ func runDatagramConnClient(t *testing.T, net, laddr, taddr string, isEmpty bool)
case "udp", "udp4", "udp6": case "udp", "udp4", "udp6":
c, err = Dial(net, taddr) c, err = Dial(net, taddr)
if err != nil { if err != nil {
t.Errorf("Dial(%q, %q) failed: %v", net, taddr, err) t.Fatalf("Dial(%q, %q) failed: %v", net, taddr, err)
return
} }
case "unixgram": case "unixgram":
c, err = DialUnix(net, &UnixAddr{laddr, net}, &UnixAddr{taddr, net}) c, err = DialUnix(net, &UnixAddr{laddr, net}, &UnixAddr{taddr, net})
if err != nil { if err != nil {
t.Errorf("DialUnix(%q, {%q, %q}) failed: %v", net, laddr, taddr, err) t.Fatalf("DialUnix(%q, {%q, %q}) failed: %v", net, laddr, taddr, err)
return
} }
} }
defer c.Close() defer c.Close()
...@@ -415,14 +408,12 @@ func runDatagramConnClient(t *testing.T, net, laddr, taddr string, isEmpty bool) ...@@ -415,14 +408,12 @@ func runDatagramConnClient(t *testing.T, net, laddr, taddr string, isEmpty bool)
wb = []byte("DatagramConnClient by Dial\n") wb = []byte("DatagramConnClient by Dial\n")
} }
if n, err := c.Write(wb[0:]); err != nil || n != len(wb) { if n, err := c.Write(wb[0:]); err != nil || n != len(wb) {
t.Errorf("Write failed: %v, %v; want %v, <nil>", n, err, len(wb)) t.Fatalf("Write failed: %v, %v; want %v, <nil>", n, err, len(wb))
return
} }
rb := make([]byte, 1024) rb := make([]byte, 1024)
if n, err := c.Read(rb[0:]); err != nil || n != len(wb) { if n, err := c.Read(rb[0:]); err != nil || n != len(wb) {
t.Errorf("Read failed: %v, %v; want %v, <nil>", n, err, len(wb)) t.Fatalf("Read failed: %v, %v; want %v, <nil>", n, err, len(wb))
return
} }
} }
...@@ -433,20 +424,17 @@ func runDatagramPacketConnClient(t *testing.T, net, laddr, taddr string, isEmpty ...@@ -433,20 +424,17 @@ func runDatagramPacketConnClient(t *testing.T, net, laddr, taddr string, isEmpty
case "udp", "udp4", "udp6": case "udp", "udp4", "udp6":
ra, err = ResolveUDPAddr(net, taddr) ra, err = ResolveUDPAddr(net, taddr)
if err != nil { if err != nil {
t.Errorf("ResolveUDPAddr(%q, %q) failed: %v", net, taddr, err) t.Fatalf("ResolveUDPAddr(%q, %q) failed: %v", net, taddr, err)
return
} }
case "unixgram": case "unixgram":
ra, err = ResolveUnixAddr(net, taddr) ra, err = ResolveUnixAddr(net, taddr)
if err != nil { if err != nil {
t.Errorf("ResolveUxixAddr(%q, %q) failed: %v", net, taddr, err) t.Fatalf("ResolveUxixAddr(%q, %q) failed: %v", net, taddr, err)
return
} }
} }
c, err := ListenPacket(net, laddr) c, err := ListenPacket(net, laddr)
if err != nil { if err != nil {
t.Errorf("ListenPacket(%q, %q) faild: %v", net, laddr, err) t.Fatalf("ListenPacket(%q, %q) faild: %v", net, laddr, err)
return
} }
defer c.Close() defer c.Close()
c.SetReadDeadline(time.Now().Add(1 * time.Second)) c.SetReadDeadline(time.Now().Add(1 * time.Second))
...@@ -456,13 +444,11 @@ func runDatagramPacketConnClient(t *testing.T, net, laddr, taddr string, isEmpty ...@@ -456,13 +444,11 @@ func runDatagramPacketConnClient(t *testing.T, net, laddr, taddr string, isEmpty
wb = []byte("DatagramPacketConnClient by ListenPacket\n") wb = []byte("DatagramPacketConnClient by ListenPacket\n")
} }
if n, err := c.WriteTo(wb[0:], ra); err != nil || n != len(wb) { if n, err := c.WriteTo(wb[0:], ra); err != nil || n != len(wb) {
t.Errorf("WriteTo(%v) failed: %v, %v; want %v, <nil>", ra, n, err, len(wb)) t.Fatalf("WriteTo(%v) failed: %v, %v; want %v, <nil>", ra, n, err, len(wb))
return
} }
rb := make([]byte, 1024) rb := make([]byte, 1024)
if n, _, err := c.ReadFrom(rb[0:]); err != nil || n != len(wb) { if n, _, err := c.ReadFrom(rb[0:]); err != nil || n != len(wb) {
t.Errorf("ReadFrom failed: %v, %v; want %v, <nil>", n, err, len(wb)) t.Fatalf("ReadFrom failed: %v, %v; want %v, <nil>", n, err, len(wb))
return
} }
} }
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment