1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
package linereader
import (
"bytes"
"io"
"time"
"reflect"
"testing"
)
func TestReader(t *testing.T) {
var buf bytes.Buffer
buf.WriteString("foo\nbar\n")
var result []string
r := New(&buf)
for line := range r.Ch {
result = append(result, line)
}
expected := []string{"foo", "bar"}
if !reflect.DeepEqual(result, expected) {
t.Fatalf("bad: %#v", result)
}
}
func TestReader_pause(t *testing.T) {
pr, pw := io.Pipe()
go func() {
defer pw.Close()
pw.Write([]byte("foo\n"))
pw.Write([]byte("bar"))
time.Sleep(200 * time.Millisecond)
pw.Write([]byte("baz\n"))
}()
var result []string
r := New(pr)
for line := range r.Ch {
result = append(result, line)
}
expected := []string{"foo", "bar", "baz"}
if !reflect.DeepEqual(result, expected) {
t.Fatalf("bad: %#v", result)
}
}
|
