127 lines
2.4 KiB
Go
127 lines
2.4 KiB
Go
package valuecounter
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/metric"
|
|
"github.com/influxdata/telegraf/testutil"
|
|
)
|
|
|
|
// Create a valuecounter with config
|
|
func NewTestValueCounter(fields []string) telegraf.Aggregator {
|
|
vc := &ValueCounter{
|
|
Fields: fields,
|
|
}
|
|
vc.Reset()
|
|
|
|
return vc
|
|
}
|
|
|
|
var m1, _ = metric.New("m1",
|
|
map[string]string{"foo": "bar"},
|
|
map[string]interface{}{
|
|
"status": 200,
|
|
"somefield": 20.1,
|
|
"foobar": "bar",
|
|
},
|
|
time.Now(),
|
|
)
|
|
|
|
var m2, _ = metric.New("m1",
|
|
map[string]string{"foo": "bar"},
|
|
map[string]interface{}{
|
|
"status": "OK",
|
|
"ignoreme": "string",
|
|
"andme": true,
|
|
"boolfield": false,
|
|
},
|
|
time.Now(),
|
|
)
|
|
|
|
func BenchmarkApply(b *testing.B) {
|
|
vc := NewTestValueCounter([]string{"status"})
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
vc.Add(m1)
|
|
vc.Add(m2)
|
|
}
|
|
}
|
|
|
|
// Test basic functionality
|
|
func TestBasic(t *testing.T) {
|
|
vc := NewTestValueCounter([]string{"status"})
|
|
acc := testutil.Accumulator{}
|
|
|
|
vc.Add(m1)
|
|
vc.Add(m2)
|
|
vc.Add(m1)
|
|
vc.Push(&acc)
|
|
|
|
expectedFields := map[string]interface{}{
|
|
"status_200": 2,
|
|
"status_OK": 1,
|
|
}
|
|
expectedTags := map[string]string{
|
|
"foo": "bar",
|
|
}
|
|
acc.AssertContainsTaggedFields(t, "m1", expectedFields, expectedTags)
|
|
}
|
|
|
|
// Test with multiple fields to count
|
|
func TestMultipleFields(t *testing.T) {
|
|
vc := NewTestValueCounter([]string{"status", "somefield", "boolfield"})
|
|
acc := testutil.Accumulator{}
|
|
|
|
vc.Add(m1)
|
|
vc.Add(m2)
|
|
vc.Add(m2)
|
|
vc.Add(m1)
|
|
vc.Push(&acc)
|
|
|
|
expectedFields := map[string]interface{}{
|
|
"status_200": 2,
|
|
"status_OK": 2,
|
|
"boolfield_false": 2,
|
|
}
|
|
expectedTags := map[string]string{
|
|
"foo": "bar",
|
|
}
|
|
acc.AssertContainsTaggedFields(t, "m1", expectedFields, expectedTags)
|
|
}
|
|
|
|
// Test with a reset between two runs
|
|
func TestWithReset(t *testing.T) {
|
|
vc := NewTestValueCounter([]string{"status"})
|
|
acc := testutil.Accumulator{}
|
|
|
|
vc.Add(m1)
|
|
vc.Add(m1)
|
|
vc.Add(m2)
|
|
vc.Push(&acc)
|
|
|
|
expectedFields := map[string]interface{}{
|
|
"status_200": 2,
|
|
"status_OK": 1,
|
|
}
|
|
expectedTags := map[string]string{
|
|
"foo": "bar",
|
|
}
|
|
acc.AssertContainsTaggedFields(t, "m1", expectedFields, expectedTags)
|
|
|
|
acc.ClearMetrics()
|
|
vc.Reset()
|
|
|
|
vc.Add(m2)
|
|
vc.Add(m2)
|
|
vc.Add(m1)
|
|
vc.Push(&acc)
|
|
|
|
expectedFields = map[string]interface{}{
|
|
"status_200": 1,
|
|
"status_OK": 2,
|
|
}
|
|
acc.AssertContainsTaggedFields(t, "m1", expectedFields, expectedTags)
|
|
}
|