1 // Copyright 2010 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 "bytes" 9 "io" 10 "testing" 11 ) 12 13 func checkPipeWrite(t *testing.T, w io.Writer, data []byte, c chan int) { 14 n, err := w.Write(data) 15 if err != nil { 16 t.Error(err) 17 } 18 if n != len(data) { 19 t.Errorf("short write: %d != %d", n, len(data)) 20 } 21 c <- 0 22 } 23 24 func checkPipeRead(t *testing.T, r io.Reader, data []byte, wantErr error) { 25 buf := make([]byte, len(data)+10) 26 n, err := r.Read(buf) 27 if err != wantErr { 28 t.Error(err) 29 return 30 } 31 if n != len(data) || !bytes.Equal(buf[0:n], data) { 32 t.Errorf("bad read: got %q", buf[0:n]) 33 return 34 } 35 } 36 37 // TestPipe tests a simple read/write/close sequence. 38 // Assumes that the underlying io.Pipe implementation 39 // is solid and we're just testing the net wrapping. 40 func TestPipe(t *testing.T) { 41 c := make(chan int) 42 cli, srv := Pipe() 43 go checkPipeWrite(t, cli, []byte("hello, world"), c) 44 checkPipeRead(t, srv, []byte("hello, world"), nil) 45 <-c 46 go checkPipeWrite(t, srv, []byte("line 2"), c) 47 checkPipeRead(t, cli, []byte("line 2"), nil) 48 <-c 49 go checkPipeWrite(t, cli, []byte("a third line"), c) 50 checkPipeRead(t, srv, []byte("a third line"), nil) 51 <-c 52 go srv.Close() 53 checkPipeRead(t, cli, nil, io.EOF) 54 cli.Close() 55 } 56
