You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.6 KiB
56 lines
1.6 KiB
6 years ago
|
package metric
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestPointGaugeAdd(t *testing.T) {
|
||
|
|
opts := PointGaugeOpts{Size: 3}
|
||
|
|
pointGauge := NewPointGauge(opts)
|
||
|
|
listBuckets := func() [][]float64 {
|
||
|
|
buckets := make([][]float64, 0)
|
||
|
|
pointGauge.Reduce(func(i Iterator) float64 {
|
||
|
|
for i.Next() {
|
||
|
|
bucket := i.Bucket()
|
||
|
|
buckets = append(buckets, bucket.Points)
|
||
|
|
}
|
||
|
|
return 0.0
|
||
|
|
})
|
||
|
|
return buckets
|
||
|
|
}
|
||
|
|
assert.Equal(t, [][]float64{{}, {}, {}}, listBuckets(), "Empty Buckets")
|
||
|
|
pointGauge.Add(1)
|
||
|
|
assert.Equal(t, [][]float64{{}, {}, {1}}, listBuckets(), "Point 1")
|
||
|
|
pointGauge.Add(2)
|
||
|
|
assert.Equal(t, [][]float64{{}, {1}, {2}}, listBuckets(), "Point 1, 2")
|
||
|
|
pointGauge.Add(3)
|
||
|
|
assert.Equal(t, [][]float64{{1}, {2}, {3}}, listBuckets(), "Point 1, 2, 3")
|
||
|
|
pointGauge.Add(4)
|
||
|
|
assert.Equal(t, [][]float64{{2}, {3}, {4}}, listBuckets(), "Point 2, 3, 4")
|
||
|
|
pointGauge.Add(5)
|
||
|
|
assert.Equal(t, [][]float64{{3}, {4}, {5}}, listBuckets(), "Point 3, 4, 5")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPointGaugeReduce(t *testing.T) {
|
||
|
|
opts := PointGaugeOpts{Size: 10}
|
||
|
|
pointGauge := NewPointGauge(opts)
|
||
|
|
for i := 0; i < opts.Size; i++ {
|
||
|
|
pointGauge.Add(int64(i))
|
||
|
|
}
|
||
|
|
var _ = pointGauge.Reduce(func(i Iterator) float64 {
|
||
|
|
idx := 0
|
||
|
|
for i.Next() {
|
||
|
|
bucket := i.Bucket()
|
||
|
|
assert.Equal(t, bucket.Points[0], float64(idx), "validate points of pointGauge")
|
||
|
|
idx++
|
||
|
|
}
|
||
|
|
return 0.0
|
||
|
|
})
|
||
|
|
assert.Equal(t, float64(9), pointGauge.Max(), "validate max of pointGauge")
|
||
|
|
assert.Equal(t, float64(4.5), pointGauge.Avg(), "validate avg of pointGauge")
|
||
|
|
assert.Equal(t, float64(0), pointGauge.Min(), "validate min of pointGauge")
|
||
|
|
assert.Equal(t, float64(45), pointGauge.Sum(), "validate sum of pointGauge")
|
||
|
|
}
|