-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
metric_test.go
101 lines (80 loc) · 1.77 KB
/
metric_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// +build !js
// Copyright (c) 2009-2020 Misakai Ltd. and contributors. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for details.
package stats
import (
"sync"
"testing"
"time"
"github.com/stretchr/testify/assert"
)
func BenchmarkMetricSnapshot(b *testing.B) {
h := NewMetric("x")
for i := int32(0); i < 50000; i++ {
h.Update(i)
}
b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
h.Snapshot()
}
}
func BenchmarkMetricUpdate(b *testing.B) {
m := NewMetric("x")
b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
m.Update(15423)
}
}
func TestMetricConcurrency(t *testing.T) {
h := NewMetric("x")
var wg sync.WaitGroup
wg.Add(2)
go func() {
for i := int32(0); i < 50000; i++ {
h.Update(i)
}
wg.Done()
}()
go func() {
for i := int32(0); i < 50000; i++ {
h.Reset()
}
wg.Done()
}()
wg.Wait()
}
func TestMetric(t *testing.T) {
h := NewMetric("x")
for i := int32(0); i < 100; i++ {
h.Update(i)
}
h.UpdateTag("test")
assert.Equal(t, "test", h.Tag())
// Create a snapshot
assert.Equal(t, 100, h.Count())
assert.Equal(t, 99, h.Max())
assert.Equal(t, 0, h.Min())
assert.True(t, h.Mean() > 49)
assert.True(t, h.StdDev() > 28)
assert.Equal(t, "x", h.Name())
assert.Equal(t, float64(49.5), h.Quantile(50)[0])
assert.Equal(t, 833.25, h.Variance())
assert.NotZero(t, h.Rate())
t0, t1 := h.Window()
assert.NotEqual(t, time.Unix(0, 0), t0)
assert.NotEqual(t, time.Unix(0, 0), t1)
assert.Len(t, h.Histogram(0, 50, 100), 2)
h.Reset()
assert.Equal(t, 0, h.Count())
assert.Equal(t, 0, h.Max())
}
func TestSampleClamp(t *testing.T) {
h := NewMetric("x")
for i := int32(0); i < 2000; i++ {
h.Update(i)
}
sample := h.sample()
assert.Equal(t, reservoirSize, len(sample))
}