telegraf/plugins/outputs/instrumental/instrumental_test.go

package instrumental

import (
	"bufio"
	"net"
	"net/textproto"
	"sync"
	"testing"
	"time"

	"github.com/influxdata/telegraf"
	"github.com/influxdata/telegraf/metric"
	"github.com/stretchr/testify/assert"
)

func TestWrite(t *testing.T) {
	var wg sync.WaitGroup
	wg.Add(1)
	go TCPServer(t, &wg)
	// Give the fake TCP server some time to start:
	time.Sleep(time.Millisecond * 100)

	i := Instrumental{
		Host:     "127.0.0.1",
		ApiToken: "abc123token",
		Prefix:   "my.prefix",
	}

	// Default to gauge
	m1, _ := metric.New(
		"mymeasurement",
		map[string]string{"host": "192.168.0.1"},
		map[string]interface{}{"myfield": float64(3.14)},
		time.Date(2010, time.November, 10, 23, 0, 0, 0, time.UTC),
	)
	m2, _ := metric.New(
		"mymeasurement",
		map[string]string{"host": "192.168.0.1", "metric_type": "set"},
		map[string]interface{}{"value": float64(3.14)},
		time.Date(2010, time.November, 10, 23, 0, 0, 0, time.UTC),
	)

	metrics := []telegraf.Metric{m1, m2}
	i.Write(metrics)

	// Counter and Histogram are increments
	m3, _ := metric.New(
		"my_histogram",
		map[string]string{"host": "192.168.0.1", "metric_type": "histogram"},
		map[string]interface{}{"value": float64(3.14)},
		time.Date(2010, time.November, 10, 23, 0, 0, 0, time.UTC),
	)
	// We will modify metric names that won't be accepted by Instrumental
	m4, _ := metric.New(
		"bad_metric_name",
		map[string]string{"host": "192.168.0.1:8888::123", "metric_type": "counter"},
		map[string]interface{}{"value": 1},
		time.Date(2010, time.November, 10, 23, 0, 0, 0, time.UTC),
	)
	// We will drop metric values that won't be accepted by Instrumental
	m5, _ := metric.New(
		"bad_values",
		map[string]string{"host": "192.168.0.1", "metric_type": "counter"},
		map[string]interface{}{"value": "\" 3:30\""},
		time.Date(2010, time.November, 10, 23, 0, 0, 0, time.UTC),
	)
	m6, _ := metric.New(
		"my_counter",
		map[string]string{"host": "192.168.0.1", "metric_type": "counter"},
		map[string]interface{}{"value": float64(3.14)},
		time.Date(2010, time.November, 10, 23, 0, 0, 0, time.UTC),
	)

	metrics = []telegraf.Metric{m3, m4, m5, m6}
	i.Write(metrics)

	wg.Wait()
}

func TCPServer(t *testing.T, wg *sync.WaitGroup) {
	tcpServer, _ := net.Listen("tcp", "127.0.0.1:8000")
	defer wg.Done()
	conn, _ := tcpServer.Accept()
	conn.SetDeadline(time.Now().Add(1 * time.Second))
	reader := bufio.NewReader(conn)
	tp := textproto.NewReader(reader)

	hello, _ := tp.ReadLine()
	assert.Equal(t, "hello version go/telegraf/1.1", hello)
	auth, _ := tp.ReadLine()
	assert.Equal(t, "authenticate abc123token", auth)
	conn.Write([]byte("ok\nok\n"))

	data1, _ := tp.ReadLine()
	assert.Equal(t, "gauge my.prefix.192_168_0_1.mymeasurement.myfield 3.14 1289430000", data1)
	data2, _ := tp.ReadLine()
	assert.Equal(t, "gauge my.prefix.192_168_0_1.mymeasurement 3.14 1289430000", data2)

	conn, _ = tcpServer.Accept()
	conn.SetDeadline(time.Now().Add(1 * time.Second))
	reader = bufio.NewReader(conn)
	tp = textproto.NewReader(reader)

	hello, _ = tp.ReadLine()
	assert.Equal(t, "hello version go/telegraf/1.1", hello)
	auth, _ = tp.ReadLine()
	assert.Equal(t, "authenticate abc123token", auth)
	conn.Write([]byte("ok\nok\n"))

	data3, _ := tp.ReadLine()
	assert.Equal(t, "increment my.prefix.192_168_0_1.my_histogram 3.14 1289430000", data3)

	data4, _ := tp.ReadLine()
	assert.Equal(t, "increment my.prefix.192_168_0_1_8888_123.bad_metric_name 1 1289430000", data4)

	data5, _ := tp.ReadLine()
	assert.Equal(t, "increment my.prefix.192_168_0_1.my_counter 3.14 1289430000", data5)

	data6, _ := tp.ReadLine()
	assert.Equal(t, "", data6)

	conn.Close()
}