telegraf/plugins/exec/exec_test.go

99 lines
1.9 KiB
Go
Raw Normal View History

2015-08-06 00:29:27 +00:00
package exec
import (
"fmt"
"testing"
2015-08-06 00:29:27 +00:00
"github.com/influxdb/telegraf/testutil"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
2015-09-23 18:21:42 +00:00
// Midnight 9/22/2015
const baseTimeSeconds = 1442905200
2015-08-06 00:29:27 +00:00
const validJson = `
{
"status": "green",
"num_processes": 82,
"cpu": {
"status": "red",
"nil_status": null,
2015-08-06 00:29:27 +00:00
"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 {
2015-09-23 18:21:42 +00:00
return &runnerMock{
out: out,
err: err,
}
2015-08-06 00:29:27 +00:00
}
func (r runnerMock) Run(e *Exec) ([]byte, error) {
2015-08-06 00:29:27 +00:00
if r.err != nil {
return nil, r.err
}
return r.out, nil
}
func TestExec(t *testing.T) {
2015-09-23 18:21:42 +00:00
e := &Exec{
runner: newRunnerMock([]byte(validJson), nil),
Command: "testcommand arg1",
Name: "mycollector",
2015-09-23 18:21:42 +00:00
}
2015-08-06 00:29:27 +00:00
var acc testutil.Accumulator
err := e.Gather(&acc)
require.NoError(t, err)
assert.Equal(t, acc.NFields(), 4, "non-numeric measurements should be ignored")
2015-08-06 00:29:27 +00:00
fields := map[string]interface{}{
"num_processes": float64(82),
"cpu_used": float64(8234),
"cpu_free": float64(32),
"percent": float64(0.81),
2015-08-06 00:29:27 +00:00
}
acc.AssertContainsFields(t, "exec_mycollector", fields)
2015-08-06 00:29:27 +00:00
}
func TestExecMalformed(t *testing.T) {
2015-09-23 18:21:42 +00:00
e := &Exec{
runner: newRunnerMock([]byte(malformedJson), nil),
Command: "badcommand arg1",
Name: "mycollector",
2015-09-23 18:21:42 +00:00
}
2015-08-06 00:29:27 +00:00
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")
2015-08-06 00:29:27 +00:00
}
func TestCommandError(t *testing.T) {
2015-09-23 18:21:42 +00:00
e := &Exec{
runner: newRunnerMock(nil, fmt.Errorf("exit status code 1")),
Command: "badcommand",
Name: "mycollector",
2015-09-23 18:21:42 +00:00
}
2015-08-06 00:29:27 +00:00
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")
2015-08-06 00:29:27 +00:00
}