1package agent
2
3import (
4	"bytes"
5	"testing"
6)
7
8func TestGatedWriter(t *testing.T) {
9	buf := new(bytes.Buffer)
10	w := &GatedWriter{Writer: buf}
11	w.Write([]byte("foo\n"))
12	w.Write([]byte("bar\n"))
13
14	if buf.String() != "" {
15		t.Fatalf("bad: %s", buf.String())
16	}
17
18	w.Flush()
19
20	if buf.String() != "foo\nbar\n" {
21		t.Fatalf("bad: %s", buf.String())
22	}
23
24	w.Write([]byte("baz\n"))
25
26	if buf.String() != "foo\nbar\nbaz\n" {
27		t.Fatalf("bad: %s", buf.String())
28	}
29}
30