79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
// 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()
|
|
}
|
|
}
|