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