119 lines
3.1 KiB
Go
119 lines
3.1 KiB
Go
package config
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/influxdb/telegraf/plugins"
|
|
"github.com/influxdb/telegraf/plugins/exec"
|
|
"github.com/influxdb/telegraf/plugins/memcached"
|
|
"github.com/influxdb/telegraf/plugins/procstat"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestConfig_LoadSinglePlugin(t *testing.T) {
|
|
c := NewConfig()
|
|
c.LoadConfig("./testdata/single_plugin.toml")
|
|
|
|
memcached := plugins.Plugins["memcached"]().(*memcached.Memcached)
|
|
memcached.Servers = []string{"localhost"}
|
|
|
|
mConfig := &PluginConfig{
|
|
Name: "memcached",
|
|
Drop: []string{"other", "stuff"},
|
|
Pass: []string{"some", "strings"},
|
|
TagDrop: []TagFilter{
|
|
TagFilter{
|
|
Name: "badtag",
|
|
Filter: []string{"othertag"},
|
|
},
|
|
},
|
|
TagPass: []TagFilter{
|
|
TagFilter{
|
|
Name: "goodtag",
|
|
Filter: []string{"mytag"},
|
|
},
|
|
},
|
|
Interval: 5 * time.Second,
|
|
}
|
|
|
|
assert.Equal(t, memcached, c.Plugins[0].Plugin,
|
|
"Testdata did not produce a correct memcached struct.")
|
|
assert.Equal(t, mConfig, c.Plugins[0].Config,
|
|
"Testdata did not produce correct memcached metadata.")
|
|
}
|
|
|
|
func TestConfig_LoadDirectory(t *testing.T) {
|
|
c := NewConfig()
|
|
err := c.LoadConfig("./testdata/single_plugin.toml")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
err = c.LoadDirectory("./testdata/subconfig")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
memcached := plugins.Plugins["memcached"]().(*memcached.Memcached)
|
|
memcached.Servers = []string{"localhost"}
|
|
|
|
mConfig := &PluginConfig{
|
|
Name: "memcached",
|
|
Drop: []string{"other", "stuff"},
|
|
Pass: []string{"some", "strings"},
|
|
TagDrop: []TagFilter{
|
|
TagFilter{
|
|
Name: "badtag",
|
|
Filter: []string{"othertag"},
|
|
},
|
|
},
|
|
TagPass: []TagFilter{
|
|
TagFilter{
|
|
Name: "goodtag",
|
|
Filter: []string{"mytag"},
|
|
},
|
|
},
|
|
Interval: 5 * time.Second,
|
|
}
|
|
assert.Equal(t, memcached, c.Plugins[0].Plugin,
|
|
"Testdata did not produce a correct memcached struct.")
|
|
assert.Equal(t, mConfig, c.Plugins[0].Config,
|
|
"Testdata did not produce correct memcached metadata.")
|
|
|
|
ex := plugins.Plugins["exec"]().(*exec.Exec)
|
|
ex.Commands = []*exec.Command{
|
|
&exec.Command{
|
|
Command: "/usr/bin/myothercollector --foo=bar",
|
|
Name: "myothercollector",
|
|
},
|
|
}
|
|
eConfig := &PluginConfig{Name: "exec"}
|
|
assert.Equal(t, ex, c.Plugins[1].Plugin,
|
|
"Merged Testdata did not produce a correct exec struct.")
|
|
assert.Equal(t, eConfig, c.Plugins[1].Config,
|
|
"Merged Testdata did not produce correct exec metadata.")
|
|
|
|
memcached.Servers = []string{"192.168.1.1"}
|
|
assert.Equal(t, memcached, c.Plugins[2].Plugin,
|
|
"Testdata did not produce a correct memcached struct.")
|
|
assert.Equal(t, mConfig, c.Plugins[2].Config,
|
|
"Testdata did not produce correct memcached metadata.")
|
|
|
|
pstat := plugins.Plugins["procstat"]().(*procstat.Procstat)
|
|
pstat.Specifications = []*procstat.Specification{
|
|
&procstat.Specification{
|
|
PidFile: "/var/run/grafana-server.pid",
|
|
},
|
|
&procstat.Specification{
|
|
PidFile: "/var/run/influxdb/influxd.pid",
|
|
},
|
|
}
|
|
|
|
pConfig := &PluginConfig{Name: "procstat"}
|
|
|
|
assert.Equal(t, pstat, c.Plugins[3].Plugin,
|
|
"Merged Testdata did not produce a correct procstat struct.")
|
|
assert.Equal(t, pConfig, c.Plugins[3].Config,
|
|
"Merged Testdata did not produce correct procstat metadata.")
|
|
}
|