telegraf/plugins/outputs/file/file_test.go

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()
	defer os.Remove(fh.Name())
	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()
	defer os.Remove(fh)
	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()
	defer os.Remove(fh1.Name())
	fh2 := createFile()
	defer os.Remove(fh2.Name())
	fh3 := createFile()
	defer os.Remove(fh3.Name())

	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()
	defer os.Remove(fh1)
	fh2 := tmpFile()
	defer os.Remove(fh2)
	fh3 := tmpFile()
	defer os.Remove(fh3)
	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()
	defer os.Remove(fh1.Name())
	fh2 := tmpFile()
	defer os.Remove(fh2)

	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))
}