119 lines
3.3 KiB
Go
119 lines
3.3 KiB
Go
package json_v2_test
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/config"
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
|
"github.com/influxdata/telegraf/plugins/inputs/file"
|
|
"github.com/influxdata/telegraf/plugins/parsers/influx"
|
|
"github.com/influxdata/telegraf/testutil"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMultipleConfigs(t *testing.T) {
|
|
// Get all directories in testdata
|
|
folders, err := ioutil.ReadDir("testdata")
|
|
require.NoError(t, err)
|
|
// Make sure testdata contains data
|
|
require.Greater(t, len(folders), 0)
|
|
|
|
expectedErrors := []struct {
|
|
Name string
|
|
Error string
|
|
}{
|
|
{
|
|
Name: "wrong_path",
|
|
Error: "wrong",
|
|
},
|
|
}
|
|
|
|
for _, f := range folders {
|
|
t.Run(f.Name(), func(t *testing.T) {
|
|
// Process the telegraf config file for the test
|
|
buf, err := os.ReadFile(fmt.Sprintf("testdata/%s/telegraf.conf", f.Name()))
|
|
require.NoError(t, err)
|
|
inputs.Add("file", func() telegraf.Input {
|
|
return &file.File{}
|
|
})
|
|
cfg := config.NewConfig()
|
|
err = cfg.LoadConfigData(buf)
|
|
require.NoError(t, err)
|
|
|
|
// Gather the metrics from the input file configure
|
|
acc := testutil.Accumulator{}
|
|
for _, input := range cfg.Inputs {
|
|
err = input.Init()
|
|
require.NoError(t, err)
|
|
err = input.Gather(&acc)
|
|
// If the test has an expected error then require one was received
|
|
var expectedError bool
|
|
for _, e := range expectedErrors {
|
|
if e.Name == f.Name() {
|
|
require.Contains(t, err.Error(), e.Error)
|
|
expectedError = true
|
|
break
|
|
}
|
|
}
|
|
if !expectedError {
|
|
require.NoError(t, err)
|
|
}
|
|
}
|
|
|
|
// Process expected metrics and compare with resulting metrics
|
|
expectedOutputs, err := readMetricFile(t, fmt.Sprintf("testdata/%s/expected.out", f.Name()))
|
|
require.NoError(t, err)
|
|
resultingMetrics := acc.GetTelegrafMetrics()
|
|
testutil.RequireMetricsEqual(t, expectedOutputs, resultingMetrics, testutil.IgnoreTime())
|
|
|
|
// Folder with timestamp prefixed will also check for matching timestamps to make sure they are parsed correctly
|
|
// The milliseconds weren't matching, seemed like a rounding difference between the influx parser
|
|
// Compares each metrics times separately and ignores milliseconds
|
|
if strings.HasPrefix(f.Name(), "timestamp") {
|
|
require.Equal(t, len(expectedOutputs), len(resultingMetrics))
|
|
for i, m := range resultingMetrics {
|
|
require.Equal(t, expectedOutputs[i].Time().Truncate(time.Second), m.Time().Truncate(time.Second))
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func readMetricFile(t *testing.T, path string) ([]telegraf.Metric, error) {
|
|
var metrics []telegraf.Metric
|
|
expectedFile, err := os.Open(path)
|
|
if err != nil {
|
|
return metrics, err
|
|
}
|
|
defer expectedFile.Close()
|
|
|
|
parser := &influx.Parser{}
|
|
require.NoError(t, parser.Init())
|
|
scanner := bufio.NewScanner(expectedFile)
|
|
for scanner.Scan() {
|
|
line := scanner.Text()
|
|
if line != "" {
|
|
m, err := parser.ParseLine(line)
|
|
// The timezone needs to be UTC to match the timestamp test results
|
|
m.SetTime(m.Time().UTC())
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unable to parse metric in %q failed: %v", line, err)
|
|
}
|
|
metrics = append(metrics, m)
|
|
}
|
|
}
|
|
err = expectedFile.Close()
|
|
if err != nil {
|
|
return metrics, err
|
|
}
|
|
|
|
return metrics, nil
|
|
}
|