2019-04-11 12:59:53 +03:00
|
|
|
package metrics
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
2019-04-11 14:03:30 +03:00
|
|
|
"fmt"
|
2019-04-11 12:59:53 +03:00
|
|
|
"strings"
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestSummarySerial(t *testing.T) {
|
|
|
|
name := `TestSummarySerial`
|
|
|
|
s := NewSummary(name)
|
|
|
|
|
|
|
|
// Verify that the summary isn't visible in the output of WritePrometheus,
|
|
|
|
// since it doesn't contain any values yet.
|
|
|
|
var bb bytes.Buffer
|
|
|
|
WritePrometheus(&bb, false)
|
|
|
|
result := bb.String()
|
|
|
|
if strings.Contains(result, name) {
|
|
|
|
t.Fatalf("summary %s shouldn't be visible in the WritePrometheus output; got\n%s", name, result)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Write data to summary
|
|
|
|
for i := 0; i < 2000; i++ {
|
|
|
|
s.Update(float64(i))
|
|
|
|
t := time.Now()
|
|
|
|
s.UpdateDuration(t.Add(-time.Millisecond * time.Duration(i)))
|
|
|
|
}
|
|
|
|
|
|
|
|
// Make sure the summary doesn't print anything on marshalTo call
|
|
|
|
// and updates s.quantileValues.
|
|
|
|
testMarshalTo(t, s, "prefix", "")
|
|
|
|
|
|
|
|
// Verify s.quantileValues
|
|
|
|
if s.quantileValues[len(s.quantileValues)-1] != 1999 {
|
|
|
|
t.Fatalf("unexpected quantileValues[last]; got %v; want %v", s.quantileValues[len(s.quantileValues)-1], 1999)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Make sure the summary 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 summary %s in the WritePrometheus output; got\n%s", name, result)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSummaryConcurrent(t *testing.T) {
|
|
|
|
name := "SummaryConcurrent"
|
|
|
|
s := NewSummary(name)
|
|
|
|
err := testConcurrent(func() error {
|
|
|
|
for i := 0; i < 10; i++ {
|
|
|
|
s.Update(float64(i))
|
|
|
|
}
|
|
|
|
testMarshalTo(t, s, "prefix", "")
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSummaryWithTags(t *testing.T) {
|
|
|
|
name := `TestSummary{tag="foo"}`
|
|
|
|
s := NewSummary(name)
|
|
|
|
s.Update(123)
|
|
|
|
|
|
|
|
var bb bytes.Buffer
|
|
|
|
WritePrometheus(&bb, false)
|
|
|
|
result := bb.String()
|
|
|
|
namePrefixWithTag := `TestSummary{tag="foo",quantile="`
|
|
|
|
if !strings.Contains(result, namePrefixWithTag) {
|
|
|
|
t.Fatalf("missing summary prefix %s in the WritePrometheus output; got\n%s", namePrefixWithTag, result)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSummaryInvalidQuantiles(t *testing.T) {
|
|
|
|
name := "SummaryInvalidQuantiles"
|
|
|
|
expectPanic(t, name, func() {
|
|
|
|
NewSummaryExt(name, time.Minute, []float64{123, -234})
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSummarySmallWindow(t *testing.T) {
|
|
|
|
name := "SummarySmallWindow"
|
|
|
|
window := time.Millisecond * 20
|
|
|
|
quantiles := []float64{0.1, 0.2, 0.3}
|
|
|
|
s := NewSummaryExt(name, window, quantiles)
|
|
|
|
for i := 0; i < 2000; i++ {
|
|
|
|
s.Update(123)
|
|
|
|
}
|
|
|
|
// Wait for window update and verify that the summary has been cleared.
|
|
|
|
time.Sleep(5 * window)
|
|
|
|
var bb bytes.Buffer
|
|
|
|
WritePrometheus(&bb, false)
|
|
|
|
result := bb.String()
|
|
|
|
if strings.Contains(result, name) {
|
|
|
|
t.Fatalf("summary %s cannot be present in the WritePrometheus output; got\n%s", name, result)
|
|
|
|
}
|
|
|
|
}
|
2019-04-11 14:03:30 +03:00
|
|
|
|
|
|
|
func TestGetOrCreateSummaryInvalidWindow(t *testing.T) {
|
|
|
|
name := "GetOrCreateSummaryInvalidWindow"
|
|
|
|
GetOrCreateSummaryExt(name, defaultSummaryWindow, defaultSummaryQuantiles)
|
|
|
|
expectPanic(t, name, func() {
|
|
|
|
GetOrCreateSummaryExt(name, defaultSummaryWindow/2, defaultSummaryQuantiles)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestGetOrCreateSummaryInvalidQuantiles(t *testing.T) {
|
|
|
|
name := "GetOrCreateSummaryInvalidQuantiles"
|
|
|
|
GetOrCreateSummaryExt(name, defaultSummaryWindow, defaultSummaryQuantiles)
|
|
|
|
expectPanic(t, name, func() {
|
|
|
|
GetOrCreateSummaryExt(name, defaultSummaryWindow, []float64{0.1, 0.2})
|
|
|
|
})
|
|
|
|
quantiles := append([]float64{}, defaultSummaryQuantiles...)
|
|
|
|
quantiles[len(quantiles)-1] /= 2
|
|
|
|
expectPanic(t, name, func() {
|
|
|
|
GetOrCreateSummaryExt(name, defaultSummaryWindow, quantiles)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestGetOrCreateSummarySerial(t *testing.T) {
|
|
|
|
name := "GetOrCreateSummarySerial"
|
|
|
|
if err := testGetOrCreateSummary(name); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestGetOrCreateSummaryConcurrent(t *testing.T) {
|
|
|
|
name := "GetOrCreateSummaryConcurrent"
|
|
|
|
err := testConcurrent(func() error {
|
|
|
|
return testGetOrCreateSummary(name)
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func testGetOrCreateSummary(name string) error {
|
|
|
|
s1 := GetOrCreateSummary(name)
|
|
|
|
for i := 0; i < 10; i++ {
|
|
|
|
s2 := GetOrCreateSummary(name)
|
|
|
|
if s1 != s2 {
|
|
|
|
return fmt.Errorf("unexpected summary returned; got %p; want %p", s2, s1)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|