package sbuf import ( "bytes" "crypto/rand" "testing" "golang.org/x/crypto/nacl/box" ) var ( buf = &Buffer{} testMessage1 = []byte("round and round and round we go, where we stop, no one knows") testMessage2 = []byte("the deconstruction of falling stars") ) func TestWrite(t *testing.T) { n, err := buf.Write(testMessage1) if err != nil { t.Fatalf("%v", err) } if n != len(testMessage1) { t.Fatalf("expected to write %d bytes, but only wrote %d bytes", len(testMessage1), n) } if buf.Len() != len(testMessage1) { t.Fatalf("expected a length of %d, but have a length of %d", len(testMessage1), buf.Len()) } if buf.Cap() != (len(testMessage1) * 2) { t.Fatalf("expected a capacity of %d, but have a capacity of %d", len(testMessage1)*2, buf.Cap()) } n, err = buf.Write(testMessage2) if err != nil { t.Fatalf("%v", err) } if n != len(testMessage2) { t.Fatalf("expected to write %d bytes, but only wrote %d bytes", len(testMessage2), n) } n, err = buf.Write(testMessage2) if err != nil { t.Fatalf("%v", err) } if n != len(testMessage2) { t.Fatalf("expected to write %d bytes, but only wrote %d bytes", len(testMessage2), n) } } func TestRead(t *testing.T) { var p = make([]byte, len(testMessage1)) n, err := buf.Read(p) if err != nil { t.Fatalf("%v", err) } if n != len(testMessage1) { t.Fatalf("expected to read %d bytes, but only read %d bytes", len(testMessage1), n) } if !bytes.Equal(p, testMessage1) { t.Fatalf("expected p='%s', but p='%s'", testMessage1, p) } p = make([]byte, len(testMessage2)) n, err = buf.Read(p) if err != nil { t.Fatalf("%v", err) } if n != len(testMessage2) { t.Fatalf("expected to read %d bytes, but only read %d bytes", len(testMessage2), n) } if !bytes.Equal(p, testMessage2) { t.Fatalf("expected p='%s', but p='%s'", testMessage2, p) } buf.Close() n, err = buf.Read(p) if err == nil { t.Fatal("expected EOF") } else if n != 0 { t.Fatalf("expect n=0, but n=%d", n) } p = nil n, err = buf.Read(p) if err != nil { t.Fatalf("%v", err) } else if n != 0 { t.Fatalf("expect n=0, but n=%d", n) } } func TestShortRead(t *testing.T) { tmpMessage := []byte("hello, world") buf.Write(tmpMessage) var p = make([]byte, len(testMessage1)) n, err := buf.Read(p) if err != nil { t.Fatalf("%v", err) } if n != len(tmpMessage) { t.Fatalf("expected to read %d bytes, but only read %d bytes", len(testMessage1), n) } } func TestNewBuffer(t *testing.T) { buf := NewBuffer(32) if len(buf.buf) != 0 { t.Fatalf("expected new buffer length to be 0, have %d", len(buf.buf)) } if cap(buf.buf) != 32 { t.Fatalf("expected new buffer capacity to be 0, have %d", cap(buf.buf)) } } func TestNewBufferFrom(t *testing.T) { p := make([]byte, len(testMessage1)) copy(p, testMessage1) buf := NewBufferFrom(p) if !bytes.Equal(buf.buf, testMessage1) { t.Fatal("new buffer wasn't constructed properly") } } func TestBytes(t *testing.T) { p := make([]byte, len(testMessage1)) copy(p, testMessage1) buf := NewBufferFrom(p) out := buf.Bytes() if buf.buf != nil { t.Fatal("buffer was not closed") } if !bytes.Equal(out, testMessage1) { t.Fatal("buffer did not return the right data") } out = buf.Bytes() if out != nil { t.Fatal("a closed buffer should return nil for Bytes") } } func TestRWByte(t *testing.T) { buf := NewBuffer(0) c := byte(42) err := buf.WriteByte(c) if err != nil { t.Fatalf("%v", err) } c, err = buf.ReadByte() if err != nil { t.Fatalf("%v", err) } if c != 42 { t.Fatal("Expected 42, have %d", c) } _, err = buf.ReadByte() if err == nil { t.Fatal("Expected EOF") } } func BenchmarkRead(b *testing.B) { b.N = 2000 pub, priv, err := box.GenerateKey(rand.Reader) if err != nil { b.Fatalf("%v", err) } b.ReportAllocs() for i := 0; i < b.N; i++ { _, err := buf.Write(priv[:]) if err != nil { b.Fatalf("%v", err) } _, err = buf.Write(pub[:]) if err != nil { b.Fatalf("%v", err) } b.SetBytes(64) } } func BenchmarkFixed(b *testing.B) { pub, priv, err := box.GenerateKey(rand.Reader) if err != nil { b.Fatalf("%v", err) } buf = NewBuffer(64 * b.N) b.ReportAllocs() for i := 0; i < b.N; i++ { _, err := buf.Write(priv[:]) if err != nil { b.Fatalf("%v", err) } _, err = buf.Write(pub[:]) if err != nil { b.Fatalf("%v", err) } b.SetBytes(64) } } func BenchmarkWrite(b *testing.B) { var pub = new([32]byte) var priv = new([32]byte) b.ReportAllocs() for i := 0; i < b.N && buf.Len() >= 64; i++ { _, err := buf.Read(priv[:]) if err != nil { b.Fatalf("%v", err) } _, err = buf.Read(pub[:]) if err != nil { b.Fatalf("%v", err) } b.SetBytes(64) } }