forked from flashcat/categraf
48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
// Copyright 2019 Google Inc. All Rights Reserved.
|
|
// This file is available under the Apache license.
|
|
|
|
package datum_test
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
"testing/quick"
|
|
"time"
|
|
|
|
"flashcat.cloud/categraf/inputs/mtail/internal/metrics/datum"
|
|
)
|
|
|
|
func TestBucketContains(t *testing.T) {
|
|
if err := quick.Check(func(min, max, val float64) bool {
|
|
r := &datum.Range{Min: min, Max: max}
|
|
truth := val < max && val >= min
|
|
return truth == r.Contains(val)
|
|
}, nil); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestMakeBucket(t *testing.T) {
|
|
r := []datum.Range{
|
|
{0, 1},
|
|
{1, 2},
|
|
{2, 4},
|
|
}
|
|
b := datum.MakeBuckets(r, time.Unix(37, 42))
|
|
ts := time.Unix(37, 31)
|
|
datum.Observe(b, 2, ts)
|
|
if r := datum.GetBucketsSum(b); r != 2 {
|
|
t.Errorf("sum not 2, got %v", r)
|
|
}
|
|
if r := datum.GetBucketsCount(b); r != 1 {
|
|
t.Errorf("count not 1, got %v", r)
|
|
}
|
|
bs := datum.GetBucketsCumByMax(b)
|
|
if r := datum.GetBucketsCount(b); r != bs[math.Inf(+1)] {
|
|
t.Errorf("Inf bucket des not equal total observation count: %v vs %v", bs[math.Inf(+1)], r)
|
|
}
|
|
if len(bs) != len(r)+1 {
|
|
t.Errorf("missing buckets from BucketsByMax: expected %d, got %v", len(r)+1, len(bs))
|
|
}
|
|
}
|