1 // Copyright 2011 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.
12 var alphabet = []byte("abcdefghijklmnopqrstuvwxyz")
14 func TestBufferReadwrite(t *testing.T) {
16 b.write(alphabet[:10])
17 r, _ := b.Read(make([]byte, 10))
19 t.Fatalf("Expected written == read == 10, written: 10, read %d", r)
24 r, _ = b.Read(make([]byte, 10))
26 t.Fatalf("Expected written == read == 5, written: 5, read %d", r)
30 b.write(alphabet[:10])
31 r, _ = b.Read(make([]byte, 5))
33 t.Fatalf("Expected written == 10, read == 5, written: 10, read %d", r)
38 b.write(alphabet[5:15])
39 r, _ = b.Read(make([]byte, 10))
40 r2, _ := b.Read(make([]byte, 10))
41 if r != 10 || r2 != 5 || 15 != r+r2 {
42 t.Fatal("Expected written == read == 15")
46 func TestBufferClose(t *testing.T) {
48 b.write(alphabet[:10])
50 _, err := b.Read(make([]byte, 5))
52 t.Fatal("expected read of 5 to not return EOF")
55 b.write(alphabet[:10])
57 r, err := b.Read(make([]byte, 5))
58 r2, err2 := b.Read(make([]byte, 10))
59 if r != 5 || r2 != 5 || err != nil || err2 != nil {
60 t.Fatal("expected reads of 5 and 5")
64 b.write(alphabet[:10])
66 r, err = b.Read(make([]byte, 5))
67 r2, err2 = b.Read(make([]byte, 10))
68 r3, err3 := b.Read(make([]byte, 10))
69 if r != 5 || r2 != 5 || r3 != 0 || err != nil || err2 != nil || err3 != io.EOF {
70 t.Fatal("expected reads of 5 and 5 and 0, with EOF")
74 b.write(make([]byte, 5))
75 b.write(make([]byte, 10))
77 r, err = b.Read(make([]byte, 9))
78 r2, err2 = b.Read(make([]byte, 3))
79 r3, err3 = b.Read(make([]byte, 3))
80 r4, err4 := b.Read(make([]byte, 10))
81 if err != nil || err2 != nil || err3 != nil || err4 != io.EOF {
82 t.Fatalf("Expected EOF on forth read only, err=%v, err2=%v, err3=%v, err4=%v", err, err2, err3, err4)
84 if r != 9 || r2 != 3 || r3 != 3 || r4 != 0 {
85 t.Fatal("Expected written == read == 15", r, r2, r3, r4)