-
Notifications
You must be signed in to change notification settings - Fork 0
/
cache_test.go
125 lines (97 loc) · 2.69 KB
/
cache_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
125
package cache
import (
"reflect"
"strconv"
"testing"
"time"
)
func TestSetGetUpdate(t *testing.T) {
cache := NewCache()
testKey := "test"
testVal := []byte("This is a test!")
expiration := time.Now().Add(time.Hour)
cache.Set(testKey, testVal, expiration)
readVal := cache.Get(testKey)
if !reflect.DeepEqual(readVal, testVal) {
t.Error("Key put in store not equal to key read from store")
}
testVal = []byte("Different value")
cache.Set(testKey, testVal, expiration)
readVal = cache.Get(testKey)
if !reflect.DeepEqual(readVal, testVal) {
t.Error("Key put in store not equal to key read from store")
}
}
func TestNonexistentGet(t *testing.T) {
cache := NewCache()
nonexistent := cache.Get("lol")
if nonexistent != nil {
t.Error("Non-existent values should be nil")
}
}
func TestEviction(t *testing.T) {
cache := NewCache()
testKey := "test"
testVal := []byte("This is a test!")
expiration := time.Now().Add(time.Hour)
cache.Set(testKey, testVal, expiration)
// This is pretty jank, fix this testing by not relying on time
cache.StartEviction(0, time.Millisecond)
time.Sleep(5 * time.Millisecond)
cache.StopEviction()
shouldBeEvicted := cache.Get(testKey)
if shouldBeEvicted != nil {
t.Error("Expected the value to be evicted")
}
}
func TestStartEvictionTwice(t *testing.T) {
cache := NewCache()
err := cache.StartEviction(5000, time.Second)
if err != nil {
t.Error("Expected no error when starting eviction for the first time")
}
err = cache.StartEviction(5000, time.Second)
if err == nil {
t.Error("Expected error when starting eviction twice")
}
}
func TestExpiration(t *testing.T) {
cache := NewCache()
testKey := "test"
testVal := []byte("This is a test!")
expiration := time.Now().Add(time.Hour)
cache.Set(testKey, testVal, expiration)
cache.clock = &clock{instant: expiration.Add(1 * time.Second)}
shouldBeEvicted := cache.Get(testKey)
if shouldBeEvicted != nil {
t.Error("Expected the expired item to be evicted")
}
}
func TestBytesReferenced(t *testing.T) {
cache := NewCache()
testKey := "test"
testVal := []byte("This is a test!")
expiration := time.Now().Add(time.Hour)
cache.Set(testKey, testVal, expiration)
size := cache.BytesReferenced()
if size != uint64(len(testVal)) {
t.Error("Expected size of cache to be equal to sum of items")
}
}
func BenchmarkConcurrentInserts(b *testing.B) {
cache := NewCache()
expiration := time.Now().Add(time.Hour)
var strings []string
for i := 0; i < 100000; i++ {
strings = append(strings, strconv.Itoa(i))
}
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
counter := 0
for pb.Next() {
s := strings[counter%len(strings)]
cache.Set(s, []byte(s), expiration)
counter++
}
})
}