forked from influxdata/influxdb
-
Notifications
You must be signed in to change notification settings - Fork 0
/
stats_test.go
88 lines (70 loc) · 1.79 KB
/
stats_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
package influxdb_test
import (
"testing"
"github.com/influxdb/influxdb"
)
func TestStats_SetAndGet(t *testing.T) {
s := influxdb.NewStats("foo")
s.Set("a", 100)
if s.Get("a") != 100 {
t.Fatalf("stats set failed, expected 100, got %d", s.Get("a"))
}
}
func TestStats_Add(t *testing.T) {
s := influxdb.NewStats("foo")
s.Add("a", 200)
if s.Get("a") != 200 {
t.Fatalf("stats set failed, expected 200, got %d", s.Get("a"))
}
}
func TestStats_Inc(t *testing.T) {
s := influxdb.NewStats("foo")
s.Set("a", 100)
s.Inc("a")
if s.Get("a") != 101 {
t.Fatalf("stats Inc failed, expected 101, got %d", s.Get("a"))
}
s.Inc("b")
if s.Get("b") != 1 {
t.Fatalf("stats Inc failed, expected 1, got %d", s.Get("b"))
}
}
func TestStats_AddNegative(t *testing.T) {
s := influxdb.NewStats("foo")
s.Add("a", -200)
if s.Get("a") != -200 {
t.Fatalf("stats set failed, expected -200, got %d", s.Get("a"))
}
}
func TestStats_SetAndAdd(t *testing.T) {
s := influxdb.NewStats("foo")
s.Set("a", 100)
s.Add("a", 200)
if s.Get("a") != 300 {
t.Fatalf("stats set failed, expected 300, got %d", s.Get("a"))
}
}
func TestStats_Diff(t *testing.T) {
foo := influxdb.NewStats("server")
bar := influxdb.NewStats("server")
foo.Set("a", 100)
foo.Set("b", 600)
bar.Set("a", 450)
bar.Set("b", 525)
qux := bar.Diff(foo)
if qux.Name() != "server" {
t.Fatalf("stats diff has unexpected name: %s", qux.Name())
}
if qux.Get("a") != 350 || qux.Get("b") != -75 {
t.Fatalf("stats diff returned unexpected result: %v", qux)
}
}
func TestStats_Snapshot(t *testing.T) {
foo := influxdb.NewStats("server")
foo.Set("a", 100)
foo.Set("b", 600)
bar := foo.Snapshot()
if bar.Name() != "server" || bar.Get("a") != 100 || bar.Get("b") != 600 {
t.Fatalf("stats snapshot returned unexpected result: %#v", bar)
}
}