telegraf/plugins/inputs/systemd_units/systemd_units_test.go

package systemd_units

import (
	"bytes"
	"reflect"
	"sort"
	"strings"
	"testing"

	"github.com/influxdata/telegraf/testutil"
)

// Global test definitions and structure.
// Tests are located within `subcommand_list_test.go` and
// `subcommand_show_test.go`.

type TestDef struct {
	Name   string
	Line   string
	Lines  []string
	Tags   map[string]string
	Fields map[string]interface{}
	Status int
	Err    error
}

func runParserTests(t *testing.T, tests []TestDef, dut *subCommandInfo) {
	for _, tt := range tests {
		t.Run(tt.Name, func(t *testing.T) {
			acc := new(testutil.Accumulator)

			var line string
			if len(tt.Lines) > 0 && len(tt.Line) == 0 {
				line = strings.Join(tt.Lines, "\n")
			} else if len(tt.Lines) == 0 && len(tt.Line) > 0 {
				line = tt.Line
			} else {
				t.Error("property Line and Lines set in test definition")
			}

			dut.parseResult(acc, bytes.NewBufferString(line))
			err := acc.FirstError()

			if !reflect.DeepEqual(tt.Err, err) {
				t.Errorf("%s: expected error '%#v' got '%#v'", tt.Name, tt.Err, err)
			}
			if len(acc.Metrics) > 0 {
				m := acc.Metrics[0]
				if !reflect.DeepEqual(m.Measurement, measurement) {
					t.Errorf("%s: expected measurement '%#v' got '%#v'\n", tt.Name, measurement, m.Measurement)
				}
				if !reflect.DeepEqual(m.Tags, tt.Tags) {
					t.Errorf("%s: expected tags\n%#v got\n%#v\n", tt.Name, tt.Tags, m.Tags)
				}
				if !reflect.DeepEqual(m.Fields, tt.Fields) {
					t.Errorf("%s: expected fields\n%#v got\n%#v\n", tt.Name, tt.Fields, m.Fields)
				}
			}
		})
	}
}

func runCommandLineTest(t *testing.T, paramsTemplate []string, dut *subCommandInfo, systemdUnits *SystemdUnits) {
	params := *dut.getParameters(systemdUnits)

	// Because we sort the params and the template array before comparison
	// we have to compare the positional parameters first.
	for i, v := range paramsTemplate {
		if strings.HasPrefix(v, "--") {
			break
		}

		if v != params[i] {
			t.Errorf("Positional parameter %d is '%s'. Expected '%s'.", i, params[i], v)
		}
	}
	// Because the maps do not lead to a stable order of the "--property"
	// arguments sort all the command line arguments and compare them.
	sort.Strings(params)
	sort.Strings(paramsTemplate)
	if !reflect.DeepEqual(params, paramsTemplate) {
		t.Errorf("Generated list of command line arguments '%#v' do not match expected list command line arguments '%#v'", params, paramsTemplate)
	}
}