21add2c799
closes #1412 separate hello and authenticate functions, force connection close at end of write cycle so we don't hold open idle connections, which has the benefit of mostly removing the chance of getting hopelessly connection lost bump instrumental agent version fix test to deal with better better connect/reconnect logic and changed ident & auth handshake Update CHANGELOG.md correct URL from instrumental fork to origin and put the change in the correct part of the file go fmt undo split hello and auth commands, to reduce roundtrips
109 lines
3.0 KiB
Go
109 lines
3.0 KiB
Go
package instrumental
|
|
|
|
import (
|
|
"bufio"
|
|
"net"
|
|
"net/textproto"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
"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, _ := telegraf.NewMetric(
|
|
"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, _ := telegraf.NewMetric(
|
|
"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, _ := telegraf.NewMetric(
|
|
"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 drop metrics that simply won't be accepted by Instrumental
|
|
m4, _ := telegraf.NewMetric(
|
|
"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),
|
|
)
|
|
m5, _ := telegraf.NewMetric(
|
|
"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}
|
|
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.my_counter 3.14 1289430000", data4)
|
|
|
|
conn.Close()
|
|
}
|