blob: 5560a79ebcd73b042d19c37807991c9d05b1f875 [file] [log] [blame]
// Copyright 2015 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package livelog
import (
"io"
"sync"
"testing"
"time"
)
func TestBuffer(t *testing.T) {
var wg sync.WaitGroup
testConc := func(prefix string, r io.Reader, want string, wantErr error) {
wg.Add(1)
go func() {
defer wg.Done()
testRead(t, prefix, r, want, wantErr)
}()
}
sleep := func() { time.Sleep(time.Millisecond) }
const w1, w2, w3, w4 = "first", "second", "third", "fourth"
var buf Buffer
buf.Write([]byte(w1))
r1 := buf.Reader()
testRead(t, "r1, w1", r1, w1, nil)
testConc("r1, w2", r1, w2, nil)
sleep()
buf.Write([]byte(w2))
r2 := buf.Reader()
testRead(t, "r2, w1+w2", r2, w1+w2, nil)
wg.Wait()
testConc("r1, w3", r1, w3, nil)
testConc("r2, w3", r2, w3, nil)
sleep()
buf.Write([]byte(w3))
wg.Wait()
r3 := buf.Reader()
testRead(t, "r3, w1+w2+w3", r3, w1+w2+w3, nil)
testConc("r1, w4", r1, w4, nil)
testConc("r3, eof", r3, "", io.EOF)
sleep()
r3.Close()
buf.Write([]byte(w4))
testRead(t, "r2, w4", r2, w4, nil)
wg.Wait()
testConc("r1 eof", r1, "", io.EOF)
sleep()
buf.Close()
testRead(t, "r2 eof", r2, "", io.EOF)
wg.Wait()
r4 := buf.Reader()
testRead(t, "r4 all", r4, w1+w2+w3+w4, nil)
testRead(t, "r4 eof", r4, "", io.EOF)
}
func testRead(t *testing.T, prefix string, r io.Reader, want string, wantErr error) {
b := make([]byte, 1024)
n, err := r.Read(b)
if err != wantErr {
t.Errorf("%s: got err %v, want %v", prefix, err, wantErr)
return
}
ok := true
if n != len(want) {
t.Errorf("%s: got n = %v, want %v", prefix, n, len(want))
ok = false
}
if s := string(b[:n]); s != want {
t.Errorf("%s: read %q, want %q", prefix, s, want)
ok = false
}
if ok {
t.Logf("%s: ok", prefix)
}
}