-
Notifications
You must be signed in to change notification settings - Fork 298
/
store_test.go
79 lines (73 loc) · 1.7 KB
/
store_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
// Copyright 2011 Dmitry Chestnykh. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package captcha
import (
"bytes"
"testing"
)
func TestSetGet(t *testing.T) {
s := NewMemoryStore(CollectNum, Expiration)
id := "captcha id"
d := RandomDigits(10)
s.Set(id, d)
d2 := s.Get(id, false)
if d2 == nil || !bytes.Equal(d, d2) {
t.Errorf("saved %v, getDigits returned got %v", d, d2)
}
}
func TestGetClear(t *testing.T) {
s := NewMemoryStore(CollectNum, Expiration)
id := "captcha id"
d := RandomDigits(10)
s.Set(id, d)
d2 := s.Get(id, true)
if d2 == nil || !bytes.Equal(d, d2) {
t.Errorf("saved %v, getDigitsClear returned got %v", d, d2)
}
d2 = s.Get(id, false)
if d2 != nil {
t.Errorf("getDigitClear didn't clear (%q=%v)", id, d2)
}
}
func TestCollect(t *testing.T) {
//TODO(dchest): can't test automatic collection when saving, because
//it's currently launched in a different goroutine.
s := NewMemoryStore(10, -1)
// create 10 ids
ids := make([]string, 10)
d := RandomDigits(10)
for i := range ids {
ids[i] = randomId()
s.Set(ids[i], d)
}
s.(*memoryStore).collect()
// Must be already collected
nc := 0
for i := range ids {
d2 := s.Get(ids[i], false)
if d2 != nil {
t.Errorf("%d: not collected", i)
nc++
}
}
if nc > 0 {
t.Errorf("= not collected %d out of %d captchas", nc, len(ids))
}
}
func BenchmarkSetCollect(b *testing.B) {
b.StopTimer()
d := RandomDigits(10)
s := NewMemoryStore(9999, -1)
ids := make([]string, 1000)
for i := range ids {
ids[i] = randomId()
}
b.StartTimer()
for i := 0; i < b.N; i++ {
for j := 0; j < 1000; j++ {
s.Set(ids[j], d)
}
s.(*memoryStore).collect()
}
}