-
Notifications
You must be signed in to change notification settings - Fork 0
/
inmem_test.go
109 lines (87 loc) · 2.13 KB
/
inmem_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
package inmem_test
import (
"testing"
"time"
inmem "github.com/mavericksy/inMem"
"github.com/facebookgo/ensure"
)
func testManyThings(t *testing.T, c inmem.Cache) {
const (
k = 1
v1 = 2
v2 = 3
)
// it's empty
ensure.DeepEqual(t, c.Len(), 0)
// not there to start with
actual, found := c.Get(k)
ensure.False(t, found)
ensure.Nil(t, actual)
// add it
c.Add(k, v1, time.Now().Add(time.Hour))
// now it's there
actual, found = c.Get(k)
ensure.True(t, found)
ensure.DeepEqual(t, actual, v1)
// we have some items
ensure.DeepEqual(t, c.Len(), 1)
// replace it
c.Add(k, v2, time.Now().Add(time.Hour))
// now find the new value
actual, found = c.Get(k)
ensure.True(t, found)
ensure.DeepEqual(t, actual, v2)
// we still only have 1 item
ensure.DeepEqual(t, c.Len(), 1)
// remove it
c.Remove(k)
// not there any more
actual, found = c.Get(k)
ensure.False(t, found)
ensure.Nil(t, actual)
// it's empty again
ensure.DeepEqual(t, c.Len(), 0)
}
func TestManyThingsUnlocked(t *testing.T) {
testManyThings(t, inmem.NewUnlocked(10, true))
}
func TestManyThingsLocked(t *testing.T) {
testManyThings(t, inmem.NewLocked(10, true))
}
func TestPanicNewUnlockedSizeZero(t *testing.T) {
defer ensure.PanicDeepEqual(t, "inmem: must provide a positive size")
_ = inmem.NewUnlocked(0, true)
}
func TestPanicNewLockedSizeZero(t *testing.T) {
defer ensure.PanicDeepEqual(t, "inmem: must provide a positive size")
_ = inmem.NewLocked(0, true)
}
func TestCacheSize(t *testing.T) {
c := inmem.NewUnlocked(2, true)
e := time.Now().Add(time.Hour)
c.Add(1, 1, e)
c.Add(2, 2, e)
c.Add(3, 3, e)
ensure.DeepEqual(t, c.Len(), 2)
_, found := c.Get(1)
ensure.False(t, found)
}
func TestTTLExpired(t *testing.T) {
c := inmem.NewUnlocked(2, true)
e := time.Now().Add(-time.Hour)
c.Add(1, 1, e)
ensure.DeepEqual(t, c.Len(), 1)
_, found := c.Get(1)
ensure.False(t, found)
ensure.DeepEqual(t, c.Len(), 0)
}
//
func TestTTLNotEnforced(t *testing.T) {
c := inmem.NewUnlocked(2, false)
e := time.Now().Add(-time.Hour)
c.Add(1, 1, e)
ensure.DeepEqual(t, c.Len(), 1)
_, found := c.Get(1)
ensure.True(t, found)
ensure.DeepEqual(t, c.Len(), 1)
}