aboutsummaryrefslogtreecommitdiffstats
path: root/metrics/meter_test.go
blob: e88922260145ebb422b468c508537a94f2e52766 (plain) (blame)
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
package metrics

import (
    "testing"
    "time"
)

func BenchmarkMeter(b *testing.B) {
    m := NewMeter()
    b.ResetTimer()
    for i := 0; i < b.N; i++ {
        m.Mark(1)
    }
}

func TestGetOrRegisterMeter(t *testing.T) {
    r := NewRegistry()
    NewRegisteredMeter("foo", r).Mark(47)
    if m := GetOrRegisterMeter("foo", r); 47 != m.Count() {
        t.Fatal(m)
    }
}

func TestMeterDecay(t *testing.T) {
    ma := meterArbiter{
        ticker: time.NewTicker(time.Millisecond),
        meters: make(map[*StandardMeter]struct{}),
    }
    m := newStandardMeter()
    ma.meters[m] = struct{}{}
    go ma.tick()
    m.Mark(1)
    rateMean := m.RateMean()
    time.Sleep(100 * time.Millisecond)
    if m.RateMean() >= rateMean {
        t.Error("m.RateMean() didn't decrease")
    }
}

func TestMeterNonzero(t *testing.T) {
    m := NewMeter()
    m.Mark(3)
    if count := m.Count(); 3 != count {
        t.Errorf("m.Count(): 3 != %v\n", count)
    }
}

func TestMeterStop(t *testing.T) {
    l := len(arbiter.meters)
    m := NewMeter()
    if len(arbiter.meters) != l+1 {
        t.Errorf("arbiter.meters: %d != %d\n", l+1, len(arbiter.meters))
    }
    m.Stop()
    if len(arbiter.meters) != l {
        t.Errorf("arbiter.meters: %d != %d\n", l, len(arbiter.meters))
    }
}

func TestMeterSnapshot(t *testing.T) {
    m := NewMeter()
    m.Mark(1)
    if snapshot := m.Snapshot(); m.RateMean() != snapshot.RateMean() {
        t.Fatal(snapshot)
    }
}

func TestMeterZero(t *testing.T) {
    m := NewMeter()
    if count := m.Count(); 0 != count {
        t.Errorf("m.Count(): 0 != %v\n", count)
    }
}