From 501329009465b86e769bbefc5f1228c807658189 Mon Sep 17 00:00:00 2001
From: Cameron Sparr <cameronsparr@gmail.com>
Date: Wed, 10 Feb 2016 15:50:07 -0700
Subject: [PATCH] Add Serializer plugins, and 'file' output plugin

---
 plugins/outputs/file/file_test.go | 195 ++++++++++++++++++++++++++++++
 1 file changed, 195 insertions(+)

diff --git a/plugins/outputs/file/file_test.go b/plugins/outputs/file/file_test.go
index b691ba57a..a2f15fc08 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))
+}