oppi.li
bnewbold.net
+12
-6
automod/countstore/countstore_test.go
+12
-6
automod/countstore/countstore_test.go
···
20
20
assert.Equal(0, c)
21
21
assert.NoError(cs.Increment(ctx, "test1", "val1"))
22
22
assert.NoError(cs.Increment(ctx, "test1", "val1"))
23
-
c, err = cs.GetCount(ctx, "test1", "val1", PeriodTotal)
24
-
assert.NoError(err)
25
-
assert.Equal(2, c)
23
+
24
+
for _, period := range []string{PeriodTotal, PeriodDay, PeriodHour} {
25
+
c, err = cs.GetCount(ctx, "test1", "val1", period)
26
+
assert.NoError(err)
27
+
assert.Equal(2, c)
28
+
}
26
29
27
30
c, err = cs.GetCountDistinct(ctx, "test2", "val2", PeriodTotal)
28
31
assert.NoError(err)
···
36
39
37
40
assert.NoError(cs.IncrementDistinct(ctx, "test2", "val2", "two"))
38
41
assert.NoError(cs.IncrementDistinct(ctx, "test2", "val2", "three"))
39
-
c, err = cs.GetCountDistinct(ctx, "test2", "val2", PeriodTotal)
40
-
assert.NoError(err)
41
-
assert.Equal(3, c)
42
+
43
+
for _, period := range []string{PeriodTotal, PeriodDay, PeriodHour} {
44
+
c, err = cs.GetCountDistinct(ctx, "test2", "val2", period)
45
+
assert.NoError(err)
46
+
assert.Equal(3, c)
47
+
}
42
48
}
43
49
44
50
func TestMemCountStoreConcurrent(t *testing.T) {