зеркало из https://github.com/golang/build.git
88 строки
1.8 KiB
Go
88 строки
1.8 KiB
Go
// 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)
|
|
}
|
|
}
|