195 lines
6.2 KiB
Go
195 lines
6.2 KiB
Go
|
package metrics
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"math"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestHistogramUpdateNegativeValue(t *testing.T) {
|
||
|
|
h := NewHistogram("TestHisogramUpdateNegativeValue")
|
||
|
|
expectPanic(t, "negative value", func() {
|
||
|
|
h.Update(-123)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetBucketIdx(t *testing.T) {
|
||
|
|
f := func(v float64, idxExpected uint) {
|
||
|
|
t.Helper()
|
||
|
|
idx := getBucketIdx(v)
|
||
|
|
if idx != idxExpected {
|
||
|
|
t.Fatalf("unexpected getBucketIdx(%g); got %d; want %d", v, idx, idxExpected)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
f(0, 0)
|
||
|
|
f(math.Pow10(e10Min-10), 1)
|
||
|
|
f(math.Pow10(e10Min-1), 1)
|
||
|
|
f(1.5*math.Pow10(e10Min-1), 1)
|
||
|
|
f(2*math.Pow10(e10Min-1), 1)
|
||
|
|
f(3*math.Pow10(e10Min-1), 1)
|
||
|
|
f(9*math.Pow10(e10Min-1), 1)
|
||
|
|
f(9.999*math.Pow10(e10Min-1), 1)
|
||
|
|
f(math.Pow10(e10Min), 1)
|
||
|
|
f(1.00001*math.Pow10(e10Min), 2)
|
||
|
|
f(1.5*math.Pow10(e10Min), 2)
|
||
|
|
f(1.999999*math.Pow10(e10Min), 2)
|
||
|
|
f(2*math.Pow10(e10Min), 2)
|
||
|
|
f(2.0000001*math.Pow10(e10Min), 3)
|
||
|
|
f(2.999*math.Pow10(e10Min), 3)
|
||
|
|
f(8.999*math.Pow10(e10Min), 9)
|
||
|
|
f(9*math.Pow10(e10Min), 9)
|
||
|
|
f(9.01*math.Pow10(e10Min), 10)
|
||
|
|
f(9.99999*math.Pow10(e10Min), 10)
|
||
|
|
f(math.Pow10(e10Min+1), 10)
|
||
|
|
f(1.9*math.Pow10(e10Min+1), 11)
|
||
|
|
f(9.9*math.Pow10(e10Min+1), 19)
|
||
|
|
f(math.Pow10(e10Min+2), 19)
|
||
|
|
f(math.Pow10(e10Min+3), 28)
|
||
|
|
f(5*math.Pow10(e10Min+3), 32)
|
||
|
|
f(0.1, 1-9*(e10Min+1))
|
||
|
|
f(0.11, 2-9*(e10Min+1))
|
||
|
|
f(0.95, 1-9*e10Min)
|
||
|
|
f(1, 1-9*e10Min)
|
||
|
|
f(2, 2-9*e10Min)
|
||
|
|
f(math.Pow10(e10Max), 1+9*(e10Max-e10Min))
|
||
|
|
f(2*math.Pow10(e10Max), 2+9*(e10Max-e10Min))
|
||
|
|
f(9.999*math.Pow10(e10Max), 10+9*(e10Max-e10Min))
|
||
|
|
f(math.Pow10(e10Max+1), 10+9*(e10Max-e10Min))
|
||
|
|
f(2*math.Pow10(e10Max+1), 11+9*(e10Max-e10Min))
|
||
|
|
f(9*math.Pow10(e10Max+1), 11+9*(e10Max-e10Min))
|
||
|
|
f(math.Pow10(e10Max+5), 11+9*(e10Max-e10Min))
|
||
|
|
f(12.34*math.Pow10(e10Max+7), 11+9*(e10Max-e10Min))
|
||
|
|
f(math.Inf(1), 11+9*(e10Max-e10Min))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRangeEndFromBucketIdx(t *testing.T) {
|
||
|
|
f := func(idx uint, endExpected float64) {
|
||
|
|
t.Helper()
|
||
|
|
end := getRangeEndFromBucketIdx(idx)
|
||
|
|
if end != endExpected {
|
||
|
|
t.Fatalf("unexpected getRangeEndFromBucketIdx(%d); got %g; want %g", idx, end, endExpected)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
f(0, 0)
|
||
|
|
f(1, math.Pow10(e10Min))
|
||
|
|
f(2, 2*math.Pow10(e10Min))
|
||
|
|
f(3, 3*math.Pow10(e10Min))
|
||
|
|
f(9, 9*math.Pow10(e10Min))
|
||
|
|
f(10, math.Pow10(e10Min+1))
|
||
|
|
f(11, 2*math.Pow10(e10Min+1))
|
||
|
|
f(16, 7*math.Pow10(e10Min+1))
|
||
|
|
f(17, 8*math.Pow10(e10Min+1))
|
||
|
|
f(18, 9*math.Pow10(e10Min+1))
|
||
|
|
f(19, math.Pow10(e10Min+2))
|
||
|
|
f(20, 2*math.Pow10(e10Min+2))
|
||
|
|
f(21, 3*math.Pow10(e10Min+2))
|
||
|
|
f(bucketsCount-21, 9*math.Pow10(e10Max-2))
|
||
|
|
f(bucketsCount-20, math.Pow10(e10Max-1))
|
||
|
|
f(bucketsCount-16, 5*math.Pow10(e10Max-1))
|
||
|
|
f(bucketsCount-13, 8*math.Pow10(e10Max-1))
|
||
|
|
f(bucketsCount-12, 9*math.Pow10(e10Max-1))
|
||
|
|
f(bucketsCount-11, math.Pow10(e10Max))
|
||
|
|
f(bucketsCount-10, 2*math.Pow10(e10Max))
|
||
|
|
f(bucketsCount-4, 8*math.Pow10(e10Max))
|
||
|
|
f(bucketsCount-3, 9*math.Pow10(e10Max))
|
||
|
|
f(bucketsCount-2, math.Pow10(e10Max+1))
|
||
|
|
f(bucketsCount-1, math.Inf(1))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHistogramSerial(t *testing.T) {
|
||
|
|
name := `TestHistogramSerial`
|
||
|
|
h := NewHistogram(name)
|
||
|
|
|
||
|
|
// Verify that the histogram is invisible in the output of WritePrometheus when it has no data.
|
||
|
|
var bb bytes.Buffer
|
||
|
|
WritePrometheus(&bb, false)
|
||
|
|
result := bb.String()
|
||
|
|
if strings.Contains(result, name) {
|
||
|
|
t.Fatalf("histogram %s shouldn't be visible in the WritePrometheus output; got\n%s", name, result)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Write data to histogram
|
||
|
|
for i := 84; i < 324; i++ {
|
||
|
|
h.Update(float64(i))
|
||
|
|
}
|
||
|
|
|
||
|
|
// Make sure the histogram prints <prefix>_xbucket on marshalTo call
|
||
|
|
testMarshalTo(t, h, "prefix", "prefix_vmbucket{vmrange=\"80...90\"} 7\nprefix_vmbucket{vmrange=\"90...100\"} 10\nprefix_vmbucket{vmrange=\"100...200\"} 100\nprefix_vmbucket{vmrange=\"200...300\"} 100\nprefix_vmbucket{vmrange=\"300...400\"} 23\nprefix_sum 48840\nprefix_count 240\n")
|
||
|
|
testMarshalTo(t, h, ` m{foo="bar"}`, "\t m_vmbucket{foo=\"bar\",vmrange=\"80...90\"} 7\n\t m_vmbucket{foo=\"bar\",vmrange=\"90...100\"} 10\n\t m_vmbucket{foo=\"bar\",vmrange=\"100...200\"} 100\n\t m_vmbucket{foo=\"bar\",vmrange=\"200...300\"} 100\n\t m_vmbucket{foo=\"bar\",vmrange=\"300...400\"} 23\n\t m_sum{foo=\"bar\"} 48840\n\t m_count{foo=\"bar\"} 240\n")
|
||
|
|
|
||
|
|
// Verify supported ranges
|
||
|
|
for i := -100; i < 100; i++ {
|
||
|
|
h.Update(1.23 * math.Pow10(i))
|
||
|
|
}
|
||
|
|
h.UpdateDuration(time.Now().Add(-time.Minute))
|
||
|
|
|
||
|
|
// Make sure the histogram becomes visible in the output of WritePrometheus,
|
||
|
|
// since now it contains values.
|
||
|
|
bb.Reset()
|
||
|
|
WritePrometheus(&bb, false)
|
||
|
|
result = bb.String()
|
||
|
|
if !strings.Contains(result, name) {
|
||
|
|
t.Fatalf("missing histogram %s in the WritePrometheus output; got\n%s", name, result)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHistogramConcurrent(t *testing.T) {
|
||
|
|
name := "HistogramConcurrent"
|
||
|
|
h := NewHistogram(name)
|
||
|
|
err := testConcurrent(func() error {
|
||
|
|
for i := 0; i < 10; i++ {
|
||
|
|
h.Update(float64(i))
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
testMarshalTo(t, h, "prefix", "prefix_vmbucket{vmrange=\"0...0\"} 5\nprefix_vmbucket{vmrange=\"0.9...1\"} 5\nprefix_vmbucket{vmrange=\"1...2\"} 5\nprefix_vmbucket{vmrange=\"2...3\"} 5\nprefix_vmbucket{vmrange=\"3...4\"} 5\nprefix_vmbucket{vmrange=\"4...5\"} 5\nprefix_vmbucket{vmrange=\"5...6\"} 5\nprefix_vmbucket{vmrange=\"6...7\"} 5\nprefix_vmbucket{vmrange=\"7...8\"} 5\nprefix_vmbucket{vmrange=\"8...9\"} 5\nprefix_sum 225\nprefix_count 50\n")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHistogramWithTags(t *testing.T) {
|
||
|
|
name := `TestHistogram{tag="foo"}`
|
||
|
|
h := NewHistogram(name)
|
||
|
|
h.Update(123)
|
||
|
|
|
||
|
|
var bb bytes.Buffer
|
||
|
|
WritePrometheus(&bb, false)
|
||
|
|
result := bb.String()
|
||
|
|
namePrefixWithTag := `TestHistogram_vmbucket{tag="foo",vmrange="100...200"} 1` + "\n"
|
||
|
|
if !strings.Contains(result, namePrefixWithTag) {
|
||
|
|
t.Fatalf("missing histogram %s in the WritePrometheus output; got\n%s", namePrefixWithTag, result)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetOrCreateHistogramSerial(t *testing.T) {
|
||
|
|
name := "GetOrCreateHistogramSerial"
|
||
|
|
if err := testGetOrCreateHistogram(name); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetOrCreateHistogramConcurrent(t *testing.T) {
|
||
|
|
name := "GetOrCreateHistogramConcurrent"
|
||
|
|
err := testConcurrent(func() error {
|
||
|
|
return testGetOrCreateHistogram(name)
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func testGetOrCreateHistogram(name string) error {
|
||
|
|
h1 := GetOrCreateHistogram(name)
|
||
|
|
for i := 0; i < 10; i++ {
|
||
|
|
h2 := GetOrCreateHistogram(name)
|
||
|
|
if h1 != h2 {
|
||
|
|
return fmt.Errorf("unexpected histogram returned; got %p; want %p", h2, h1)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|