-
Notifications
You must be signed in to change notification settings - Fork 0
/
manager_test.go
124 lines (104 loc) · 2.55 KB
/
manager_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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
// Copyright © 2021 Elis Lulja
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package hitcount
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestManager(t *testing.T) {
a := assert.New(t)
key := "key"
newKey := "new-key"
res := Manager(key)
a.Equal(&mgr{counters: map[string]Counter{}}, res)
a.Len(managersList, 1)
m, exists := managersList[key]
a.True(exists)
a.Equal(m, res)
managersList[newKey] = &mgr{
counters: map[string]Counter{
"test": NewIntCounter(55, 100),
},
}
res = Manager(newKey)
a.Equal(managersList[newKey], res)
a.Len(managersList, 2)
}
func TestAdd(t *testing.T) {
a := assert.New(t)
key := "key"
m := &mgr{
counters: map[string]Counter{
"test": NewIntCounter(55, 100),
},
}
m.Add(key, nil)
a.Len(m.counters, 1)
counter := NewIntCounter(77, 100)
m.Add(key, counter)
a.Len(m.counters, 2)
counterRes := m.Get(key)
a.Equal(counter, counterRes)
newCounter := NewIntCounter(88, 100)
m.Add(key, newCounter)
a.Len(m.counters, 2)
counterRes = m.Get(key)
a.Equal(counter, counterRes)
}
func TestRemove(t *testing.T) {
a := assert.New(t)
key := "exists"
notExists := "not-exists"
start := map[string]Counter{
key: NewIntCounter(55, 100),
}
m := &mgr{
counters: start,
}
m.Remove("")
a.Equal(m.counters, start)
m.Remove(notExists)
a.Equal(m.counters, start)
m.Remove(key)
a.Empty(m.counters)
}
func TestManagerResetAllExcept(t *testing.T) {
a := assert.New(t)
exceptions := []string{"one", "two"}
start := map[string]Counter{
exceptions[0]: NewIntCounter(50, 100),
exceptions[1]: NewIntCounter(60, 100),
"stays": NewIntCounter(70, 100),
}
m := &mgr{
counters: start,
}
m.ResetAllExcept(exceptions...)
a.Equal(m.counters, map[string]Counter{
exceptions[0]: NewIntCounter(50, 100),
exceptions[1]: NewIntCounter(60, 100),
"stays": NewIntCounter(0, 100),
})
}
func TestGet(t *testing.T) {
a := assert.New(t)
start := map[string]Counter{
"one": NewIntCounter(70, 100),
}
m := &mgr{
counters: start,
}
a.NotNil(m.Get("one"))
a.Nil(m.Get("two"))
}