Adding upstream version 1.34.4.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
e393c3af3f
commit
4978089aab
GPG key ID:
FBB4F0E80A80222F
4963 changed files with 677545 additions and 0 deletions
676
plugins/inputs/procstat/procstat_test.go
Normal file
676
plugins/inputs/procstat/procstat_test.go
Normal file
|
|
@ -0,0 +1,676 @@
|
|||
package procstat
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
gopsprocess "github.com/shirou/gopsutil/v4/process"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/influxdata/telegraf"
|
||||
"github.com/influxdata/telegraf/metric"
|
||||
"github.com/influxdata/telegraf/testutil"
|
||||
)
|
||||
|
||||
func init() {
|
||||
execCommand = mockExecCommand
|
||||
}
|
||||
func mockExecCommand(arg0 string, args ...string) *exec.Cmd {
|
||||
args = append([]string{"-test.run=TestMockExecCommand", "--", arg0}, args...)
|
||||
cmd := exec.Command(os.Args[0], args...)
|
||||
cmd.Stderr = os.Stderr
|
||||
return cmd
|
||||
}
|
||||
func TestMockExecCommand(_ *testing.T) {
|
||||
var cmd []string //nolint:prealloc // Pre-allocated this slice would break the algorithm
|
||||
for _, arg := range os.Args {
|
||||
if arg == "--" {
|
||||
cmd = make([]string, 0)
|
||||
continue
|
||||
}
|
||||
if cmd == nil {
|
||||
continue
|
||||
}
|
||||
cmd = append(cmd, arg)
|
||||
}
|
||||
if cmd == nil {
|
||||
return
|
||||
}
|
||||
cmdline := strings.Join(cmd, " ")
|
||||
|
||||
if cmdline == "systemctl show TestGather_systemdUnitPIDs" {
|
||||
fmt.Printf(`PIDFile=
|
||||
GuessMainPID=yes
|
||||
MainPID=11408
|
||||
ControlPID=0
|
||||
ExecMainPID=11408
|
||||
`)
|
||||
//nolint:revive // error code is important for this "test"
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
if cmdline == "supervisorctl status TestGather_supervisorUnitPIDs" {
|
||||
fmt.Printf(`TestGather_supervisorUnitPIDs RUNNING pid 7311, uptime 0:00:19
|
||||
`)
|
||||
//nolint:revive // error code is important for this "test"
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
if cmdline == "supervisorctl status TestGather_STARTINGsupervisorUnitPIDs TestGather_FATALsupervisorUnitPIDs" {
|
||||
fmt.Printf(`TestGather_FATALsupervisorUnitPIDs FATAL Exited too quickly (process log may have details)
|
||||
TestGather_STARTINGsupervisorUnitPIDs STARTING`)
|
||||
//nolint:revive // error code is important for this "test"
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
fmt.Printf("command not found\n")
|
||||
//nolint:revive // error code is important for this "test"
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
type testPgrep struct {
|
||||
pids []pid
|
||||
err error
|
||||
}
|
||||
|
||||
func newTestFinder(pids []pid) pidFinder {
|
||||
return &testPgrep{
|
||||
pids: pids,
|
||||
err: nil,
|
||||
}
|
||||
}
|
||||
|
||||
func (pg *testPgrep) pidFile(_ string) ([]pid, error) {
|
||||
return pg.pids, pg.err
|
||||
}
|
||||
|
||||
func (pg *testPgrep) pattern(_ string) ([]pid, error) {
|
||||
return pg.pids, pg.err
|
||||
}
|
||||
|
||||
func (pg *testPgrep) uid(_ string) ([]pid, error) {
|
||||
return pg.pids, pg.err
|
||||
}
|
||||
|
||||
func (pg *testPgrep) fullPattern(_ string) ([]pid, error) {
|
||||
return pg.pids, pg.err
|
||||
}
|
||||
|
||||
func (pg *testPgrep) children(_ pid) ([]pid, error) {
|
||||
pids := []pid{7311, 8111, 8112}
|
||||
return pids, pg.err
|
||||
}
|
||||
|
||||
type testProc struct {
|
||||
procID pid
|
||||
tags map[string]string
|
||||
}
|
||||
|
||||
func newTestProc(pid pid) (process, error) {
|
||||
proc := &testProc{
|
||||
procID: pid,
|
||||
tags: make(map[string]string),
|
||||
}
|
||||
return proc, nil
|
||||
}
|
||||
|
||||
func (p *testProc) pid() pid {
|
||||
return p.procID
|
||||
}
|
||||
|
||||
func (*testProc) Name() (string, error) {
|
||||
return "test_proc", nil
|
||||
}
|
||||
|
||||
func (p *testProc) setTag(k, v string) {
|
||||
p.tags[k] = v
|
||||
}
|
||||
|
||||
func (*testProc) MemoryMaps(bool) (*[]gopsprocess.MemoryMapsStat, error) {
|
||||
stats := make([]gopsprocess.MemoryMapsStat, 0)
|
||||
return &stats, nil
|
||||
}
|
||||
|
||||
func (p *testProc) metrics(prefix string, cfg *collectionConfig, t time.Time) ([]telegraf.Metric, error) {
|
||||
if prefix != "" {
|
||||
prefix += "_"
|
||||
}
|
||||
|
||||
fields := map[string]interface{}{
|
||||
prefix + "num_fds": int32(0),
|
||||
prefix + "num_threads": int32(0),
|
||||
prefix + "voluntary_context_switches": int64(0),
|
||||
prefix + "involuntary_context_switches": int64(0),
|
||||
prefix + "minor_faults": uint64(0),
|
||||
prefix + "major_faults": uint64(0),
|
||||
prefix + "child_major_faults": uint64(0),
|
||||
prefix + "child_minor_faults": uint64(0),
|
||||
prefix + "read_bytes": uint64(0),
|
||||
prefix + "read_count": uint64(0),
|
||||
prefix + "write_bytes": uint64(0),
|
||||
prefix + "write_count": uint64(0),
|
||||
prefix + "created_at": int64(0),
|
||||
}
|
||||
if cfg.features["cpu"] {
|
||||
fields[prefix+"cpu_time_user"] = float64(0)
|
||||
fields[prefix+"cpu_time_system"] = float64(0)
|
||||
fields[prefix+"cpu_time_iowait"] = float64(0)
|
||||
fields[prefix+"cpu_usage"] = float64(0)
|
||||
}
|
||||
if cfg.features["memory"] {
|
||||
fields[prefix+"memory_rss"] = uint64(0)
|
||||
fields[prefix+"memory_vms"] = uint64(0)
|
||||
fields[prefix+"memory_usage"] = float32(0)
|
||||
}
|
||||
|
||||
tags := map[string]string{
|
||||
"process_name": "test_proc",
|
||||
}
|
||||
for k, v := range p.tags {
|
||||
tags[k] = v
|
||||
}
|
||||
|
||||
// Add the tags as requested by the user
|
||||
if cfg.tagging["cmdline"] {
|
||||
tags["cmdline"] = "test_proc"
|
||||
} else {
|
||||
fields[prefix+"cmdline"] = "test_proc"
|
||||
}
|
||||
|
||||
if cfg.tagging["pid"] {
|
||||
tags["pid"] = strconv.Itoa(int(p.procID))
|
||||
} else {
|
||||
fields["pid"] = int32(p.procID)
|
||||
}
|
||||
|
||||
if cfg.tagging["ppid"] {
|
||||
tags["ppid"] = "0"
|
||||
} else {
|
||||
fields[prefix+"ppid"] = int32(0)
|
||||
}
|
||||
|
||||
if cfg.tagging["status"] {
|
||||
tags["status"] = "running"
|
||||
} else {
|
||||
fields[prefix+"status"] = "running"
|
||||
}
|
||||
|
||||
if cfg.tagging["user"] {
|
||||
tags["user"] = "testuser"
|
||||
} else {
|
||||
fields[prefix+"user"] = "testuser"
|
||||
}
|
||||
|
||||
return []telegraf.Metric{metric.New("procstat", tags, fields, t)}, nil
|
||||
}
|
||||
|
||||
var processID = pid(42)
|
||||
var exe = "foo"
|
||||
|
||||
func TestInitInvalidFinder(t *testing.T) {
|
||||
plugin := Procstat{
|
||||
PidFinder: "foo",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.Error(t, plugin.Init())
|
||||
}
|
||||
|
||||
func TestInitRequiresChildDarwin(t *testing.T) {
|
||||
if runtime.GOOS != "darwin" {
|
||||
t.Skip("Skipping test on non-darwin platform")
|
||||
}
|
||||
|
||||
p := Procstat{
|
||||
Pattern: "somepattern",
|
||||
SupervisorUnits: []string{"a_unit"},
|
||||
PidFinder: "native",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
}
|
||||
require.ErrorContains(t, p.Init(), "requires 'pgrep' finder")
|
||||
}
|
||||
|
||||
func TestInitMissingPidMethod(t *testing.T) {
|
||||
p := Procstat{
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.ErrorContains(t, p.Init(), "require filter option but none set")
|
||||
}
|
||||
|
||||
func TestGather_CreateProcessErrorOk(t *testing.T) {
|
||||
expected := []telegraf.Metric{
|
||||
testutil.MustMetric(
|
||||
"procstat_lookup",
|
||||
map[string]string{
|
||||
"exe": "foo",
|
||||
"pid_finder": "test",
|
||||
"result": "success",
|
||||
},
|
||||
map[string]interface{}{
|
||||
"pid_count": int64(1),
|
||||
"result_code": int64(0),
|
||||
"running": int64(0),
|
||||
},
|
||||
time.Unix(0, 0),
|
||||
telegraf.Untyped,
|
||||
),
|
||||
}
|
||||
|
||||
p := Procstat{
|
||||
Exe: exe,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: func(pid) (process, error) {
|
||||
return nil, errors.New("createProcess error")
|
||||
},
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
testutil.RequireMetricsEqual(t, expected, acc.GetTelegrafMetrics(), testutil.IgnoreTime())
|
||||
}
|
||||
|
||||
func TestGather_ProcessName(t *testing.T) {
|
||||
expected := []telegraf.Metric{
|
||||
testutil.MustMetric(
|
||||
"procstat",
|
||||
map[string]string{
|
||||
"exe": "foo",
|
||||
"process_name": "custom_name",
|
||||
},
|
||||
map[string]interface{}{
|
||||
"child_major_faults": uint64(0),
|
||||
"child_minor_faults": uint64(0),
|
||||
"cmdline": "test_proc",
|
||||
"cpu_time_iowait": float64(0),
|
||||
"cpu_time_system": float64(0),
|
||||
"cpu_time_user": float64(0),
|
||||
"cpu_usage": float64(0),
|
||||
"created_at": int64(0),
|
||||
"involuntary_context_switches": int64(0),
|
||||
"major_faults": uint64(0),
|
||||
"memory_rss": uint64(0),
|
||||
"memory_usage": float32(0),
|
||||
"memory_vms": uint64(0),
|
||||
"minor_faults": uint64(0),
|
||||
"num_fds": int32(0),
|
||||
"num_threads": int32(0),
|
||||
"pid": int32(42),
|
||||
"ppid": int32(0),
|
||||
"read_bytes": uint64(0),
|
||||
"read_count": uint64(0),
|
||||
"status": "running",
|
||||
"user": "testuser",
|
||||
"voluntary_context_switches": int64(0),
|
||||
"write_bytes": uint64(0),
|
||||
"write_count": uint64(0),
|
||||
},
|
||||
time.Unix(0, 0),
|
||||
telegraf.Untyped,
|
||||
),
|
||||
testutil.MustMetric(
|
||||
"procstat_lookup",
|
||||
map[string]string{
|
||||
"exe": "foo",
|
||||
"pid_finder": "test",
|
||||
"result": "success",
|
||||
},
|
||||
map[string]interface{}{
|
||||
"pid_count": int64(1),
|
||||
"result_code": int64(0),
|
||||
"running": int64(1),
|
||||
},
|
||||
time.Unix(0, 0),
|
||||
telegraf.Untyped,
|
||||
),
|
||||
}
|
||||
|
||||
p := Procstat{
|
||||
Exe: exe,
|
||||
ProcessName: "custom_name",
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
require.Equal(t, "custom_name", acc.TagValue("procstat", "process_name"))
|
||||
testutil.RequireMetricsEqual(t, expected, acc.GetTelegrafMetrics(), testutil.IgnoreTime())
|
||||
}
|
||||
|
||||
func TestGather_NoProcessNameUsesReal(t *testing.T) {
|
||||
processID := pid(os.Getpid())
|
||||
|
||||
p := Procstat{
|
||||
Exe: exe,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
require.True(t, acc.HasTag("procstat", "process_name"))
|
||||
}
|
||||
|
||||
func TestGather_NoPidTag(t *testing.T) {
|
||||
p := Procstat{
|
||||
Exe: exe,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
require.True(t, acc.HasInt64Field("procstat", "pid"))
|
||||
require.False(t, acc.HasTag("procstat", "pid"))
|
||||
}
|
||||
|
||||
func TestGather_PidTag(t *testing.T) {
|
||||
p := Procstat{
|
||||
Exe: exe,
|
||||
PidTag: true,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
require.Equal(t, "42", acc.TagValue("procstat", "pid"))
|
||||
require.False(t, acc.HasInt32Field("procstat", "pid"))
|
||||
}
|
||||
|
||||
func TestGather_Prefix(t *testing.T) {
|
||||
p := Procstat{
|
||||
Exe: exe,
|
||||
Prefix: "custom_prefix",
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
require.True(t, acc.HasInt64Field("procstat", "custom_prefix_num_fds"))
|
||||
}
|
||||
|
||||
func TestGather_Exe(t *testing.T) {
|
||||
p := Procstat{
|
||||
Exe: exe,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
require.Equal(t, exe, acc.TagValue("procstat", "exe"))
|
||||
}
|
||||
|
||||
func TestGather_User(t *testing.T) {
|
||||
user := "ada"
|
||||
|
||||
p := Procstat{
|
||||
User: user,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
require.Equal(t, user, acc.TagValue("procstat", "user"))
|
||||
}
|
||||
|
||||
func TestGather_Pattern(t *testing.T) {
|
||||
pattern := "foo"
|
||||
|
||||
p := Procstat{
|
||||
Pattern: pattern,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
require.Equal(t, pattern, acc.TagValue("procstat", "pattern"))
|
||||
}
|
||||
|
||||
func TestGather_PidFile(t *testing.T) {
|
||||
pidfile := "/path/to/pidfile"
|
||||
|
||||
p := Procstat{
|
||||
PidFile: pidfile,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
require.Equal(t, pidfile, acc.TagValue("procstat", "pidfile"))
|
||||
}
|
||||
|
||||
func TestGather_PercentFirstPass(t *testing.T) {
|
||||
processID := pid(os.Getpid())
|
||||
|
||||
p := Procstat{
|
||||
Pattern: "foo",
|
||||
PidTag: true,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
require.True(t, acc.HasFloatField("procstat", "cpu_time_user"))
|
||||
require.False(t, acc.HasFloatField("procstat", "cpu_usage"))
|
||||
}
|
||||
|
||||
func TestGather_PercentSecondPass(t *testing.T) {
|
||||
processID := pid(os.Getpid())
|
||||
|
||||
p := Procstat{
|
||||
Pattern: "foo",
|
||||
PidTag: true,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
require.True(t, acc.HasFloatField("procstat", "cpu_time_user"))
|
||||
require.True(t, acc.HasFloatField("procstat", "cpu_usage"))
|
||||
}
|
||||
|
||||
func TestGather_systemdUnitPIDs(t *testing.T) {
|
||||
p := Procstat{
|
||||
SystemdUnit: "TestGather_systemdUnitPIDs",
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
pidsTags, err := p.findPids()
|
||||
require.NoError(t, err)
|
||||
|
||||
for _, pidsTag := range pidsTags {
|
||||
require.Equal(t, []pid{11408}, pidsTag.PIDs)
|
||||
require.Equal(t, "TestGather_systemdUnitPIDs", pidsTag.Tags["systemd_unit"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestGather_cgroupPIDs(t *testing.T) {
|
||||
// no cgroups in windows
|
||||
if runtime.GOOS == "windows" {
|
||||
t.Skip("no cgroups in windows")
|
||||
}
|
||||
td := t.TempDir()
|
||||
err := os.WriteFile(filepath.Join(td, "cgroup.procs"), []byte("1234\n5678\n"), 0640)
|
||||
require.NoError(t, err)
|
||||
|
||||
p := Procstat{
|
||||
CGroup: td,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
pidsTags, err := p.findPids()
|
||||
require.NoError(t, err)
|
||||
for _, pidsTag := range pidsTags {
|
||||
require.Equal(t, []pid{1234, 5678}, pidsTag.PIDs)
|
||||
require.Equal(t, td, pidsTag.Tags["cgroup"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestProcstatLookupMetric(t *testing.T) {
|
||||
p := Procstat{
|
||||
Exe: "-Gsys",
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{543}),
|
||||
createProcess: newProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
require.NotEmpty(t, acc.GetTelegrafMetrics())
|
||||
}
|
||||
|
||||
func TestGather_SameTimestamps(t *testing.T) {
|
||||
pidfile := "/path/to/pidfile"
|
||||
|
||||
p := Procstat{
|
||||
PidFile: pidfile,
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
createProcess: newTestProc,
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, p.Gather(&acc))
|
||||
|
||||
procstat, _ := acc.Get("procstat")
|
||||
procstatLookup, _ := acc.Get("procstat_lookup")
|
||||
|
||||
require.Equal(t, procstat.Time, procstatLookup.Time)
|
||||
}
|
||||
|
||||
func TestGather_supervisorUnitPIDs(t *testing.T) {
|
||||
p := Procstat{
|
||||
SupervisorUnits: []string{"TestGather_supervisorUnitPIDs"},
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
pidsTags, err := p.findPids()
|
||||
require.NoError(t, err)
|
||||
for _, pidsTag := range pidsTags {
|
||||
require.Equal(t, []pid{7311, 8111, 8112}, pidsTag.PIDs)
|
||||
require.Equal(t, "TestGather_supervisorUnitPIDs", pidsTag.Tags["supervisor_unit"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestGather_MoresupervisorUnitPIDs(t *testing.T) {
|
||||
p := Procstat{
|
||||
SupervisorUnits: []string{"TestGather_STARTINGsupervisorUnitPIDs", "TestGather_FATALsupervisorUnitPIDs"},
|
||||
PidFinder: "test",
|
||||
Properties: []string{"cpu", "memory", "mmap"},
|
||||
Log: testutil.Logger{},
|
||||
finder: newTestFinder([]pid{processID}),
|
||||
}
|
||||
require.NoError(t, p.Init())
|
||||
|
||||
pidsTags, err := p.findPids()
|
||||
require.NoError(t, err)
|
||||
for _, pidsTag := range pidsTags {
|
||||
require.Empty(t, pidsTag.PIDs)
|
||||
switch pidsTag.Tags["supervisor_unit"] {
|
||||
case "TestGather_STARTINGsupervisorUnitPIDs":
|
||||
require.Equal(t, "STARTING", pidsTag.Tags["status"])
|
||||
case "TestGather_FATALsupervisorUnitPIDs":
|
||||
require.Equal(t, "FATAL", pidsTag.Tags["status"])
|
||||
require.Equal(t, "Exited too quickly (process log may have details)", pidsTag.Tags["error"])
|
||||
default:
|
||||
t.Fatalf("unexpected value for tag 'supervisor_unit': %q", pidsTag.Tags["supervisor_unit"])
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue