goutils/testio/testio_test.go

221 lines
3.6 KiB
Go
Raw Permalink Normal View History

2016-04-28 18:37:26 +00:00
package testio
import (
"bytes"
"os"
"testing"
)
func TestBrokenWriter(t *testing.T) {
buf := NewBrokenWriter(2)
data := []byte{1, 2}
n, err := buf.Write(data)
if err != nil {
t.Fatalf("%v", err)
} else if n != 2 {
t.Fatalf("expected write size of 2, have %d", n)
}
_, err = buf.Write(data)
if err == nil {
t.Fatal("expected a write failure")
}
buf.Reset()
_, err = buf.Write(data)
if err == nil {
t.Fatalf("expected a write failure after reset")
}
buf.Extend(2)
_, err = buf.Write(data)
if err != nil {
t.Fatalf("%v", err)
}
}
func TestBufCloser(t *testing.T) {
var data = []byte{1, 2}
var read = make([]byte, 2)
buf := NewBufCloser(data)
_, err := buf.Read(read)
if err != nil {
t.Fatalf("%v", err)
}
_, err = buf.Write(data)
if err != nil {
t.Fatalf("%v", err)
}
buf.Close()
buf.Reset()
s := "hi"
buf = NewBufCloserString(s)
read = buf.Bytes()
if string(read) != s {
t.Fatalf("expected %s, have %s", s, read)
}
}
func TestLoggingBuffer(t *testing.T) {
src := &bytes.Buffer{}
data := []byte("AB")
lb := NewLoggingBuffer(src)
_, err := lb.Write(data)
if err != nil {
t.Fatalf("%v", err)
}
src.Reset()
lb.SetName("TEST")
out := &bytes.Buffer{}
lb.LogTo(out)
_, err = lb.Write(data)
if err != nil {
t.Fatalf("%v", err)
}
expected := "[TEST] [WRITE] 4142\n"
if string(out.Bytes()) != expected {
t.Fatalf("expected '%s', have '%s'", expected, string(out.Bytes()))
}
out.Reset()
src = bytes.NewBuffer(data)
read := make([]byte, 2)
_, err = lb.Read(read)
if err != nil {
t.Fatalf("%v", err)
}
expected = "[TEST] [READ] 4142\n"
if string(out.Bytes()) != expected {
t.Fatalf("expected '%s', have '%s'", expected, string(out.Bytes()))
}
out.Reset()
lb.SetName("")
lb.LogTo(os.Stderr)
lb.Write([]byte("AB"))
lb.LogTo(out)
_, err = lb.Read(read)
if err != nil {
t.Fatalf("%v", err)
}
expected = "[READ] 4142\n"
if string(out.Bytes()) != expected {
t.Fatalf("expected '%s', have '%s'", expected, string(out.Bytes()))
}
src.Reset()
_, err = lb.Read(read)
if err == nil {
t.Fatal("expected a read failure")
}
}
func TestBrokenReadWriter(t *testing.T) {
brw := NewBrokenReadWriter(0, 0)
lb := NewLoggingBuffer(brw)
var p = make([]byte, 2)
var data = []byte("HI")
_, err := lb.Write(data)
if err == nil {
t.Fatal("expected a write failure")
}
_, err = lb.Read(p)
if err == nil {
t.Fatal("expected a read failure")
}
brw.Extend(1, 0)
_, err = lb.Write(data)
if err == nil {
t.Fatal("expected a write failure")
}
brw.Extend(2, 0)
_, err = lb.Write(data)
if err != nil {
t.Fatalf("%v", err)
}
brw.Extend(4, 1)
_, err = lb.Write(data)
if err != nil {
t.Fatalf("%v", err)
}
_, err = lb.Read(p)
if err == nil {
t.Fatal("expected a read failure")
}
brw.Reset()
brw.Extend(10, 2)
_, err = lb.Write(data)
if err != nil {
t.Fatalf("%v", err)
}
p = make([]byte, 1)
_, err = lb.Read(p)
if err != nil {
t.Fatalf("%v", err)
}
}
func TestBufferConn(t *testing.T) {
bc := NewBufferConn()
client := []byte("AB")
peer := []byte("XY")
_, err := bc.WritePeer(peer)
if err != nil {
t.Fatalf("%v", err)
}
var p = make([]byte, 2)
_, err = bc.Write(client)
if err != nil {
t.Fatalf("%v", err)
}
_, err = bc.Read(p)
if err != nil {
t.Fatalf("%v", err)
}
if !bytes.Equal(p, peer) {
t.Fatalf("client should have read %x, but read %x",
peer, p)
}
_, err = bc.ReadClient(p)
if err != nil {
t.Fatalf("%v", err)
}
if !bytes.Equal(client, p) {
t.Fatalf("client should have sent %x, but sent %x",
client, p)
}
err = bc.Close()
if err != nil {
t.Fatalf("Close should always return nil, but it returned %v", err)
}
}