make NewBufRead etc. idempotent
R=rsc
DELTA=63 (59 added, 0 deleted, 4 changed)
OCL=27143
CL=27143
diff --git a/src/lib/bufio_test.go b/src/lib/bufio_test.go
index ef97bc4..d8c77c1 100644
--- a/src/lib/bufio_test.go
+++ b/src/lib/bufio_test.go
@@ -330,3 +330,50 @@
}
}
+func TestNewBufReadSizeIdempotent(t *testing.T) {
+ const BufSize = 1000;
+ b, err := NewBufReadSize(newByteReader(io.StringBytes("hello world")), BufSize);
+ if err != nil {
+ t.Error("NewBufReadSize create fail", err);
+ }
+ // Does it recognize itself?
+ b1, err2 := NewBufReadSize(b, BufSize);
+ if err2 != nil {
+ t.Error("NewBufReadSize #2 create fail", err2);
+ }
+ if b1 != b {
+ t.Error("NewBufReadSize did not detect underlying BufRead");
+ }
+ // Does it wrap if existing buffer is too small?
+ b2, err3 := NewBufReadSize(b, 2*BufSize);
+ if err3 != nil {
+ t.Error("NewBufReadSize #3 create fail", err3);
+ }
+ if b2 == b {
+ t.Error("NewBufReadSize did not enlarge buffer");
+ }
+}
+
+func TestNewBufWriteSizeIdempotent(t *testing.T) {
+ const BufSize = 1000;
+ b, err := NewBufWriteSize(newByteWriter(), BufSize);
+ if err != nil {
+ t.Error("NewBufWriteSize create fail", err);
+ }
+ // Does it recognize itself?
+ b1, err2 := NewBufWriteSize(b, BufSize);
+ if err2 != nil {
+ t.Error("NewBufWriteSize #2 create fail", err2);
+ }
+ if b1 != b {
+ t.Error("NewBufWriteSize did not detect underlying BufWrite");
+ }
+ // Does it wrap if existing buffer is too small?
+ b2, err3 := NewBufWriteSize(b, 2*BufSize);
+ if err3 != nil {
+ t.Error("NewBufWriteSize #3 create fail", err3);
+ }
+ if b2 == b {
+ t.Error("NewBufWriteSize did not enlarge buffer");
+ }
+}
