102 lines
2.2 KiB
Go
102 lines
2.2 KiB
Go
package shim
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
tgConfig "github.com/influxdata/telegraf/config"
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
|
"github.com/influxdata/telegraf/plugins/processors"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestLoadConfig(t *testing.T) {
|
|
os.Setenv("SECRET_TOKEN", "xxxxxxxxxx")
|
|
os.Setenv("SECRET_VALUE", `test"\test`)
|
|
|
|
inputs.Add("test", func() telegraf.Input {
|
|
return &serviceInput{}
|
|
})
|
|
|
|
c := "./testdata/plugin.conf"
|
|
conf, err := LoadConfig(&c)
|
|
require.NoError(t, err)
|
|
|
|
inp := conf.Input.(*serviceInput)
|
|
|
|
require.Equal(t, "awesome name", inp.ServiceName)
|
|
require.Equal(t, "xxxxxxxxxx", inp.SecretToken)
|
|
require.Equal(t, `test"\test`, inp.SecretValue)
|
|
}
|
|
|
|
func TestDefaultImportedPluginsSelfRegisters(t *testing.T) {
|
|
inputs.Add("test", func() telegraf.Input {
|
|
return &testInput{}
|
|
})
|
|
|
|
cfg, err := LoadConfig(nil)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "test", cfg.Input.Description())
|
|
}
|
|
|
|
func TestLoadingSpecialTypes(t *testing.T) {
|
|
inputs.Add("test", func() telegraf.Input {
|
|
return &testDurationInput{}
|
|
})
|
|
|
|
c := "./testdata/special.conf"
|
|
conf, err := LoadConfig(&c)
|
|
require.NoError(t, err)
|
|
|
|
inp := conf.Input.(*testDurationInput)
|
|
|
|
require.EqualValues(t, 3*time.Second, inp.Duration)
|
|
require.EqualValues(t, 3*1000*1000, inp.Size)
|
|
}
|
|
|
|
func TestLoadingProcessorWithConfig(t *testing.T) {
|
|
proc := &testConfigProcessor{}
|
|
processors.Add("test_config_load", func() telegraf.Processor {
|
|
return proc
|
|
})
|
|
|
|
c := "./testdata/processor.conf"
|
|
_, err := LoadConfig(&c)
|
|
require.NoError(t, err)
|
|
|
|
require.EqualValues(t, "yep", proc.Loaded)
|
|
}
|
|
|
|
type testDurationInput struct {
|
|
Duration tgConfig.Duration `toml:"duration"`
|
|
Size tgConfig.Size `toml:"size"`
|
|
}
|
|
|
|
func (i *testDurationInput) SampleConfig() string {
|
|
return ""
|
|
}
|
|
|
|
func (i *testDurationInput) Description() string {
|
|
return ""
|
|
}
|
|
func (i *testDurationInput) Gather(_ telegraf.Accumulator) error {
|
|
return nil
|
|
}
|
|
|
|
type testConfigProcessor struct {
|
|
Loaded string `toml:"loaded"`
|
|
}
|
|
|
|
func (p *testConfigProcessor) SampleConfig() string {
|
|
return ""
|
|
}
|
|
|
|
func (p *testConfigProcessor) Description() string {
|
|
return ""
|
|
}
|
|
func (p *testConfigProcessor) Apply(metrics ...telegraf.Metric) []telegraf.Metric {
|
|
return metrics
|
|
}
|