diff --git a/plugins/outputs/file/file_test.go b/plugins/outputs/file/file_test.go index b691ba57a438c..a2f15fc08e581 100644 --- a/plugins/outputs/file/file_test.go +++ b/plugins/outputs/file/file_test.go @@ -1 +1,196 @@ package file + +import ( + "bytes" + "io" + "io/ioutil" + "os" + "testing" + + "github.com/stretchr/testify/assert" + + "github.com/influxdata/telegraf/internal" + "github.com/influxdata/telegraf/plugins/serializers" + "github.com/influxdata/telegraf/testutil" +) + +const ( + expNewFile = "test1,tag1=value1 value=1 1257894000000000000\n" + expExistFile = "cpu,cpu=cpu0 value=100 1455312810012459582\n" + + "test1,tag1=value1 value=1 1257894000000000000\n" +) + +func TestFileExistingFile(t *testing.T) { + fh := createFile() + s, _ := serializers.NewInfluxSerializer() + f := File{ + Files: []string{fh.Name()}, + serializer: s, + } + + err := f.Connect() + assert.NoError(t, err) + + err = f.Write(testutil.MockMetrics()) + assert.NoError(t, err) + + validateFile(fh.Name(), expExistFile, t) + + err = f.Close() + assert.NoError(t, err) +} + +func TestFileNewFile(t *testing.T) { + s, _ := serializers.NewInfluxSerializer() + fh := tmpFile() + f := File{ + Files: []string{fh}, + serializer: s, + } + + err := f.Connect() + assert.NoError(t, err) + + err = f.Write(testutil.MockMetrics()) + assert.NoError(t, err) + + validateFile(fh, expNewFile, t) + + err = f.Close() + assert.NoError(t, err) +} + +func TestFileExistingFiles(t *testing.T) { + fh1 := createFile() + fh2 := createFile() + fh3 := createFile() + + s, _ := serializers.NewInfluxSerializer() + f := File{ + Files: []string{fh1.Name(), fh2.Name(), fh3.Name()}, + serializer: s, + } + + err := f.Connect() + assert.NoError(t, err) + + err = f.Write(testutil.MockMetrics()) + assert.NoError(t, err) + + validateFile(fh1.Name(), expExistFile, t) + validateFile(fh2.Name(), expExistFile, t) + validateFile(fh3.Name(), expExistFile, t) + + err = f.Close() + assert.NoError(t, err) +} + +func TestFileNewFiles(t *testing.T) { + s, _ := serializers.NewInfluxSerializer() + fh1 := tmpFile() + fh2 := tmpFile() + fh3 := tmpFile() + f := File{ + Files: []string{fh1, fh2, fh3}, + serializer: s, + } + + err := f.Connect() + assert.NoError(t, err) + + err = f.Write(testutil.MockMetrics()) + assert.NoError(t, err) + + validateFile(fh1, expNewFile, t) + validateFile(fh2, expNewFile, t) + validateFile(fh3, expNewFile, t) + + err = f.Close() + assert.NoError(t, err) +} + +func TestFileBoth(t *testing.T) { + fh1 := createFile() + fh2 := tmpFile() + + s, _ := serializers.NewInfluxSerializer() + f := File{ + Files: []string{fh1.Name(), fh2}, + serializer: s, + } + + err := f.Connect() + assert.NoError(t, err) + + err = f.Write(testutil.MockMetrics()) + assert.NoError(t, err) + + validateFile(fh1.Name(), expExistFile, t) + validateFile(fh2, expNewFile, t) + + err = f.Close() + assert.NoError(t, err) +} + +func TestFileStdout(t *testing.T) { + // keep backup of the real stdout + old := os.Stdout + r, w, _ := os.Pipe() + os.Stdout = w + + s, _ := serializers.NewInfluxSerializer() + f := File{ + Files: []string{"stdout"}, + serializer: s, + } + + err := f.Connect() + assert.NoError(t, err) + + err = f.Write(testutil.MockMetrics()) + assert.NoError(t, err) + + err = f.Close() + assert.NoError(t, err) + + outC := make(chan string) + // copy the output in a separate goroutine so printing can't block indefinitely + go func() { + var buf bytes.Buffer + io.Copy(&buf, r) + outC <- buf.String() + }() + + // back to normal state + w.Close() + // restoring the real stdout + os.Stdout = old + out := <-outC + + assert.Equal(t, expNewFile, out) +} + +func createFile() *os.File { + f, err := ioutil.TempFile("", "") + if err != nil { + panic(err) + } + f.WriteString("cpu,cpu=cpu0 value=100 1455312810012459582\n") + return f +} + +func tmpFile() string { + d, err := ioutil.TempDir("", "") + if err != nil { + panic(err) + } + return d + internal.RandomString(10) +} + +func validateFile(fname, expS string, t *testing.T) { + buf, err := ioutil.ReadFile(fname) + if err != nil { + panic(err) + } + assert.Equal(t, expS, string(buf)) +}