196 lines
6.1 KiB
Go
196 lines
6.1 KiB
Go
|
package etcd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"golang.org/x/net/context"
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"syscall"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf/internal/config"
|
||
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
||
|
|
"github.com/influxdata/telegraf/plugins/outputs"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf/plugins/inputs/system"
|
||
|
|
"github.com/influxdata/telegraf/plugins/outputs/influxdb"
|
||
|
|
|
||
|
|
eclient "github.com/coreos/etcd/client"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test1Write(t *testing.T) {
|
||
|
|
// Delete hostname conf file
|
||
|
|
hostname, _ := os.Hostname()
|
||
|
|
os.Remove("./testdata/test1/hosts/" + hostname + ".conf")
|
||
|
|
// Get etcd client
|
||
|
|
e := NewEtcdClient("http://localhost:2379", "/telegraf")
|
||
|
|
// Delete old conf from etcd
|
||
|
|
delOptions := &eclient.DeleteOptions{
|
||
|
|
Recursive: true,
|
||
|
|
Dir: true,
|
||
|
|
}
|
||
|
|
e.Kapi.Delete(context.Background(), "/telegraf", delOptions)
|
||
|
|
|
||
|
|
// Test write dir
|
||
|
|
err := e.WriteConfigDir("./testdata/test1")
|
||
|
|
require.NoError(t, err)
|
||
|
|
resp, _ := e.Kapi.Get(context.Background(), "/telegraf/main", nil)
|
||
|
|
assert.Equal(t,
|
||
|
|
"[tags]\n dc = \"us-east-1\"\n\n[agent]\n interval = \"2s\"\n round_interval = true\n flush_interval = \"10s\"\n flush_jitter = \"0s\"\n debug = false\n hostname = \"\"\n",
|
||
|
|
resp.Node.Value)
|
||
|
|
resp, _ = e.Kapi.Get(context.Background(), "/telegraf/hosts/localhost", nil)
|
||
|
|
assert.Equal(t,
|
||
|
|
"\n[agent]\n interval = \"2s\"\n labels = [\"influx\"]\n\n[[inputs.cpu]]\n percpu = true\n totalcpu = true\n drop = [\"cpu_time*\"]\n",
|
||
|
|
resp.Node.Value)
|
||
|
|
|
||
|
|
// Test write conf
|
||
|
|
err = e.WriteLabelConfig("mylabel", "./testdata/test1/labels/network.conf")
|
||
|
|
require.NoError(t, err)
|
||
|
|
resp, _ = e.Kapi.Get(context.Background(), "/telegraf/labels/mylabel", nil)
|
||
|
|
assert.Equal(t,
|
||
|
|
"[[inputs.net]]\n\n",
|
||
|
|
resp.Node.Value)
|
||
|
|
|
||
|
|
// Test read
|
||
|
|
c := config.NewConfig()
|
||
|
|
var inputFilters []string
|
||
|
|
var outputFilters []string
|
||
|
|
c.OutputFilters = outputFilters
|
||
|
|
c.InputFilters = inputFilters
|
||
|
|
|
||
|
|
net := inputs.Inputs["net"]().(*system.NetIOStats)
|
||
|
|
influx := outputs.Outputs["influxdb"]().(*influxdb.InfluxDB)
|
||
|
|
influx.URLs = []string{"http://localhost:8086"}
|
||
|
|
influx.Database = "telegraf"
|
||
|
|
influx.Precision = "s"
|
||
|
|
|
||
|
|
c, err = e.ReadConfig(c, "mylabel,influx")
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Equal(t, net, c.Inputs[0].Input,
|
||
|
|
"Testdata did not produce a correct net struct.")
|
||
|
|
assert.Equal(t, influx, c.Outputs[0].Output,
|
||
|
|
"Testdata did not produce a correct influxdb struct.")
|
||
|
|
|
||
|
|
// Test reload
|
||
|
|
shutdown := make(chan struct{})
|
||
|
|
signals := make(chan os.Signal)
|
||
|
|
go e.LaunchWatcher(shutdown, signals)
|
||
|
|
// Test write conf
|
||
|
|
err = e.WriteLabelConfig("mylabel", "./testdata/test1/labels/network2.conf")
|
||
|
|
require.NoError(t, err)
|
||
|
|
resp, _ = e.Kapi.Get(context.Background(), "/telegraf/labels/mylabel", nil)
|
||
|
|
assert.Equal(t,
|
||
|
|
"[[inputs.net]]\n\n interfaces = [\"eth0\"]\n\n",
|
||
|
|
resp.Node.Value)
|
||
|
|
// TODO found a way to test reload ....
|
||
|
|
sig := <-signals
|
||
|
|
assert.Equal(t, syscall.SIGHUP, sig)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test2Error(t *testing.T) {
|
||
|
|
e := NewEtcdClient("http://localhost:2379", "/telegraf")
|
||
|
|
|
||
|
|
// Test write dir
|
||
|
|
err := e.WriteConfigDir("./testdata/test2")
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test3Write(t *testing.T) {
|
||
|
|
// Delete old hostname conf file
|
||
|
|
hostname, _ := os.Hostname()
|
||
|
|
os.Remove("./testdata/test1/hosts/" + hostname + ".conf")
|
||
|
|
// Write host file
|
||
|
|
if hostname != "" {
|
||
|
|
r, err := os.Open("./testdata/test1/hosts/localhost.conf")
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
defer r.Close()
|
||
|
|
|
||
|
|
w, err := os.Create("./testdata/test1/hosts/" + hostname + ".conf")
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
defer w.Close()
|
||
|
|
|
||
|
|
// do the actual work
|
||
|
|
_, err = io.Copy(w, r)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
// Get tcd client
|
||
|
|
e := NewEtcdClient("http://localhost:2379", "/telegraf")
|
||
|
|
// Delete old conf from etcd
|
||
|
|
delOptions := &eclient.DeleteOptions{
|
||
|
|
Recursive: true,
|
||
|
|
Dir: true,
|
||
|
|
}
|
||
|
|
e.Kapi.Delete(context.Background(), "/telegraf", delOptions)
|
||
|
|
|
||
|
|
// Test write dir
|
||
|
|
err := e.WriteConfigDir("./testdata/test1")
|
||
|
|
require.NoError(t, err)
|
||
|
|
resp, _ := e.Kapi.Get(context.Background(), "/telegraf/main", nil)
|
||
|
|
assert.Equal(t,
|
||
|
|
"[tags]\n dc = \"us-east-1\"\n\n[agent]\n interval = \"2s\"\n round_interval = true\n flush_interval = \"10s\"\n flush_jitter = \"0s\"\n debug = false\n hostname = \"\"\n",
|
||
|
|
resp.Node.Value)
|
||
|
|
resp, _ = e.Kapi.Get(context.Background(), "/telegraf/hosts/localhost", nil)
|
||
|
|
assert.Equal(t,
|
||
|
|
"\n[agent]\n interval = \"2s\"\n labels = [\"influx\"]\n\n[[inputs.cpu]]\n percpu = true\n totalcpu = true\n drop = [\"cpu_time*\"]\n",
|
||
|
|
resp.Node.Value)
|
||
|
|
|
||
|
|
// Test write conf
|
||
|
|
err = e.WriteLabelConfig("mylabel", "./testdata/test1/labels/network.conf")
|
||
|
|
require.NoError(t, err)
|
||
|
|
resp, _ = e.Kapi.Get(context.Background(), "/telegraf/labels/mylabel", nil)
|
||
|
|
assert.Equal(t,
|
||
|
|
"[[inputs.net]]\n\n",
|
||
|
|
resp.Node.Value)
|
||
|
|
|
||
|
|
// Test read
|
||
|
|
c := config.NewConfig()
|
||
|
|
var inputFilters []string
|
||
|
|
var outputFilters []string
|
||
|
|
c.OutputFilters = outputFilters
|
||
|
|
c.InputFilters = inputFilters
|
||
|
|
|
||
|
|
cpu := inputs.Inputs["cpu"]().(*system.CPUStats)
|
||
|
|
cpu.PerCPU = true
|
||
|
|
cpu.TotalCPU = true
|
||
|
|
net := inputs.Inputs["net"]().(*system.NetIOStats)
|
||
|
|
influx := outputs.Outputs["influxdb"]().(*influxdb.InfluxDB)
|
||
|
|
influx.URLs = []string{"http://localhost:8086"}
|
||
|
|
influx.Database = "telegraf"
|
||
|
|
influx.Precision = "s"
|
||
|
|
|
||
|
|
c, err = e.ReadConfig(c, "mylabel,influx")
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Equal(t, cpu, c.Inputs[0].Input,
|
||
|
|
"Testdata did not produce a correct net struct.")
|
||
|
|
assert.Equal(t, net, c.Inputs[1].Input,
|
||
|
|
"Testdata did not produce a correct net struct.")
|
||
|
|
assert.Equal(t, influx, c.Outputs[0].Output,
|
||
|
|
"Testdata did not produce a correct influxdb struct.")
|
||
|
|
|
||
|
|
// Test reload
|
||
|
|
shutdown := make(chan struct{})
|
||
|
|
signals := make(chan os.Signal)
|
||
|
|
go e.LaunchWatcher(shutdown, signals)
|
||
|
|
// Test write conf
|
||
|
|
err = e.WriteLabelConfig("mylabel", "./testdata/test1/labels/network2.conf")
|
||
|
|
require.NoError(t, err)
|
||
|
|
resp, _ = e.Kapi.Get(context.Background(), "/telegraf/labels/mylabel", nil)
|
||
|
|
assert.Equal(t,
|
||
|
|
"[[inputs.net]]\n\n interfaces = [\"eth0\"]\n\n",
|
||
|
|
resp.Node.Value)
|
||
|
|
// TODO found a way to test reload ....
|
||
|
|
sig := <-signals
|
||
|
|
assert.Equal(t, syscall.SIGHUP, sig)
|
||
|
|
// Delete hostname conf file
|
||
|
|
os.Remove("./testdata/test1/hosts/" + hostname + ".conf")
|
||
|
|
}
|