99 lines
1.9 KiB
Go
99 lines
1.9 KiB
Go
package exec
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/influxdb/telegraf/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Midnight 9/22/2015
|
|
const baseTimeSeconds = 1442905200
|
|
|
|
const validJson = `
|
|
{
|
|
"status": "green",
|
|
"num_processes": 82,
|
|
"cpu": {
|
|
"status": "red",
|
|
"nil_status": null,
|
|
"used": 8234,
|
|
"free": 32
|
|
},
|
|
"percent": 0.81,
|
|
"users": [0, 1, 2, 3]
|
|
}`
|
|
|
|
const malformedJson = `
|
|
{
|
|
"status": "green",
|
|
`
|
|
|
|
type runnerMock struct {
|
|
out []byte
|
|
err error
|
|
}
|
|
|
|
func newRunnerMock(out []byte, err error) Runner {
|
|
return &runnerMock{
|
|
out: out,
|
|
err: err,
|
|
}
|
|
}
|
|
|
|
func (r runnerMock) Run(e *Exec) ([]byte, error) {
|
|
if r.err != nil {
|
|
return nil, r.err
|
|
}
|
|
return r.out, nil
|
|
}
|
|
|
|
func TestExec(t *testing.T) {
|
|
e := &Exec{
|
|
runner: newRunnerMock([]byte(validJson), nil),
|
|
Command: "testcommand arg1",
|
|
Name: "mycollector",
|
|
}
|
|
|
|
var acc testutil.Accumulator
|
|
err := e.Gather(&acc)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, acc.NFields(), 4, "non-numeric measurements should be ignored")
|
|
|
|
fields := map[string]interface{}{
|
|
"num_processes": float64(82),
|
|
"cpu_used": float64(8234),
|
|
"cpu_free": float64(32),
|
|
"percent": float64(0.81),
|
|
}
|
|
acc.AssertContainsFields(t, "exec_mycollector", fields)
|
|
}
|
|
|
|
func TestExecMalformed(t *testing.T) {
|
|
e := &Exec{
|
|
runner: newRunnerMock([]byte(malformedJson), nil),
|
|
Command: "badcommand arg1",
|
|
Name: "mycollector",
|
|
}
|
|
|
|
var acc testutil.Accumulator
|
|
err := e.Gather(&acc)
|
|
require.Error(t, err)
|
|
assert.Equal(t, acc.NFields(), 0, "No new points should have been added")
|
|
}
|
|
|
|
func TestCommandError(t *testing.T) {
|
|
e := &Exec{
|
|
runner: newRunnerMock(nil, fmt.Errorf("exit status code 1")),
|
|
Command: "badcommand",
|
|
Name: "mycollector",
|
|
}
|
|
|
|
var acc testutil.Accumulator
|
|
err := e.Gather(&acc)
|
|
require.Error(t, err)
|
|
assert.Equal(t, acc.NFields(), 0, "No new points should have been added")
|
|
}
|