82 lines
1.5 KiB
Go
82 lines
1.5 KiB
Go
|
package mock
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf/testutil"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGather(t *testing.T) {
|
||
|
|
testRandom := &random{
|
||
|
|
Name: "random",
|
||
|
|
Min: 1.0,
|
||
|
|
Max: 6.0,
|
||
|
|
}
|
||
|
|
testSineWave := &sineWave{
|
||
|
|
Name: "sine",
|
||
|
|
Amplitude: 1.0,
|
||
|
|
Period: 0.5,
|
||
|
|
}
|
||
|
|
testStep := &step{
|
||
|
|
Name: "step",
|
||
|
|
Start: 0.0,
|
||
|
|
Step: 1.0,
|
||
|
|
}
|
||
|
|
testStock := &stock{
|
||
|
|
Name: "abc",
|
||
|
|
Price: 50.00,
|
||
|
|
Volatility: 0.2,
|
||
|
|
}
|
||
|
|
|
||
|
|
tags := map[string]string{
|
||
|
|
"buildling": "tbd",
|
||
|
|
"site": "nowhere",
|
||
|
|
}
|
||
|
|
|
||
|
|
m := &Mock{
|
||
|
|
MetricName: "test",
|
||
|
|
Tags: tags,
|
||
|
|
|
||
|
|
Random: []*random{testRandom},
|
||
|
|
SineWave: []*sineWave{testSineWave},
|
||
|
|
Step: []*step{testStep},
|
||
|
|
Stock: []*stock{testStock},
|
||
|
|
}
|
||
|
|
|
||
|
|
var acc testutil.Accumulator
|
||
|
|
require.NoError(t, m.Gather(&acc))
|
||
|
|
|
||
|
|
require.Len(t, acc.Metrics, 1)
|
||
|
|
|
||
|
|
metric := acc.Metrics[0]
|
||
|
|
require.Equal(t, "test", metric.Measurement)
|
||
|
|
require.Equal(t, tags, metric.Tags)
|
||
|
|
for k, v := range metric.Fields {
|
||
|
|
switch k {
|
||
|
|
case "abc":
|
||
|
|
require.Equal(t, 50.0, v)
|
||
|
|
case "random":
|
||
|
|
require.GreaterOrEqual(t, 6.0, v)
|
||
|
|
require.LessOrEqual(t, 1.0, v)
|
||
|
|
case "sine":
|
||
|
|
require.Equal(t, 0.0, v)
|
||
|
|
case "step":
|
||
|
|
require.Equal(t, 0.0, v)
|
||
|
|
default:
|
||
|
|
require.Failf(t, "unexpected field %q", k)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGatherEmpty(t *testing.T) {
|
||
|
|
m := &Mock{
|
||
|
|
MetricName: "test_empty",
|
||
|
|
}
|
||
|
|
|
||
|
|
var acc testutil.Accumulator
|
||
|
|
require.NoError(t, m.Gather(&acc))
|
||
|
|
|
||
|
|
acc.AssertDoesNotContainMeasurement(t, "test_empty")
|
||
|
|
}
|