62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
|
package inputs_io_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf/config"
|
||
|
|
_ "github.com/influxdata/telegraf/migrations/inputs_io" // register migration
|
||
|
|
_ "github.com/influxdata/telegraf/plugins/inputs/diskio" // register plugin
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCases(t *testing.T) {
|
||
|
|
// Get all directories in testdata
|
||
|
|
folders, err := os.ReadDir("testcases")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
for _, f := range folders {
|
||
|
|
// Only handle folders
|
||
|
|
if !f.IsDir() {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
t.Run(f.Name(), func(t *testing.T) {
|
||
|
|
testcasePath := filepath.Join("testcases", f.Name())
|
||
|
|
inputFile := filepath.Join(testcasePath, "telegraf.conf")
|
||
|
|
expectedFile := filepath.Join(testcasePath, "expected.conf")
|
||
|
|
|
||
|
|
// Read the expected output
|
||
|
|
expected := config.NewConfig()
|
||
|
|
require.NoError(t, expected.LoadConfig(expectedFile))
|
||
|
|
require.NotEmpty(t, expected.Inputs)
|
||
|
|
|
||
|
|
// Read the input data
|
||
|
|
input, remote, err := config.LoadConfigFile(inputFile)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.False(t, remote)
|
||
|
|
require.NotEmpty(t, input)
|
||
|
|
|
||
|
|
// Migrate
|
||
|
|
output, n, err := config.ApplyMigrations(input)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotEmpty(t, output)
|
||
|
|
require.GreaterOrEqual(t, n, uint64(1))
|
||
|
|
actual := config.NewConfig()
|
||
|
|
require.NoError(t, actual.LoadConfigData(output, config.EmptySourcePath))
|
||
|
|
|
||
|
|
// Test the output
|
||
|
|
require.Len(t, actual.Inputs, len(expected.Inputs))
|
||
|
|
actualIDs := make([]string, 0, len(expected.Inputs))
|
||
|
|
expectedIDs := make([]string, 0, len(expected.Inputs))
|
||
|
|
for i := range actual.Inputs {
|
||
|
|
actualIDs = append(actualIDs, actual.Inputs[i].ID())
|
||
|
|
expectedIDs = append(expectedIDs, expected.Inputs[i].ID())
|
||
|
|
}
|
||
|
|
require.ElementsMatch(t, expectedIDs, actualIDs, string(output))
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|