99 lines
2.3 KiB
Go
99 lines
2.3 KiB
Go
|
package procstat
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"os/user"
|
||
|
|
"runtime"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func BenchmarkPattern(b *testing.B) {
|
||
|
|
finder := &NativeFinder{}
|
||
|
|
for n := 0; n < b.N; n++ {
|
||
|
|
_, err := finder.pattern(".*")
|
||
|
|
require.NoError(b, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkFullPattern(b *testing.B) {
|
||
|
|
finder := &NativeFinder{}
|
||
|
|
for n := 0; n < b.N; n++ {
|
||
|
|
_, err := finder.fullPattern(".*")
|
||
|
|
require.NoError(b, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestChildPattern(t *testing.T) {
|
||
|
|
if runtime.GOOS == "windows" || runtime.GOOS == "darwin" {
|
||
|
|
t.Skip("Skipping test on unsupported platform")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get our own process name
|
||
|
|
parentName, err := os.Executable()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Spawn two child processes and get their PIDs
|
||
|
|
expected := make([]pid, 0, 2)
|
||
|
|
ctx, cancel := context.WithCancel(t.Context())
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
// First process
|
||
|
|
cmd1 := exec.CommandContext(ctx, "/bin/sh")
|
||
|
|
require.NoError(t, cmd1.Start(), "starting first command failed")
|
||
|
|
expected = append(expected, pid(cmd1.Process.Pid))
|
||
|
|
|
||
|
|
// Second process
|
||
|
|
cmd2 := exec.CommandContext(ctx, "/bin/sh")
|
||
|
|
require.NoError(t, cmd2.Start(), "starting first command failed")
|
||
|
|
expected = append(expected, pid(cmd2.Process.Pid))
|
||
|
|
|
||
|
|
// Use the plugin to find the children
|
||
|
|
finder := &NativeFinder{}
|
||
|
|
parent, err := finder.pattern(parentName)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Len(t, parent, 1)
|
||
|
|
children, err := finder.children(parent[0])
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.ElementsMatch(t, expected, children)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGather_RealPatternIntegration(t *testing.T) {
|
||
|
|
if testing.Short() {
|
||
|
|
t.Skip("Skipping integration test in short mode")
|
||
|
|
}
|
||
|
|
pg := &NativeFinder{}
|
||
|
|
pids, err := pg.pattern(`procstat`)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotEmpty(t, pids)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGather_RealFullPatternIntegration(t *testing.T) {
|
||
|
|
if testing.Short() {
|
||
|
|
t.Skip("Skipping integration test in short mode")
|
||
|
|
}
|
||
|
|
if runtime.GOOS != "windows" {
|
||
|
|
t.Skip("Skipping integration test on Non-Windows OS")
|
||
|
|
}
|
||
|
|
pg := &NativeFinder{}
|
||
|
|
pids, err := pg.fullPattern(`%procstat%`)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotEmpty(t, pids)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGather_RealUserIntegration(t *testing.T) {
|
||
|
|
if testing.Short() {
|
||
|
|
t.Skip("Skipping integration test in short mode")
|
||
|
|
}
|
||
|
|
currentUser, err := user.Current()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
pg := &NativeFinder{}
|
||
|
|
pids, err := pg.uid(currentUser.Username)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotEmpty(t, pids)
|
||
|
|
}
|