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
| package agent
|
| import (
| "bytes"
| "testing"
| )
|
| func TestGatedWriter(t *testing.T) {
| buf := new(bytes.Buffer)
| w := &GatedWriter{Writer: buf}
| w.Write([]byte("foo\n"))
| w.Write([]byte("bar\n"))
|
| if buf.String() != "" {
| t.Fatalf("bad: %s", buf.String())
| }
|
| w.Flush()
|
| if buf.String() != "foo\nbar\n" {
| t.Fatalf("bad: %s", buf.String())
| }
|
| w.Write([]byte("baz\n"))
|
| if buf.String() != "foo\nbar\nbaz\n" {
| t.Fatalf("bad: %s", buf.String())
| }
| }
|
|
