64 lines
1.1 KiB
Go
64 lines
1.1 KiB
Go
package kvmemcache
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func testRunner(c *Cache, t *testing.T, done chan bool) {
|
|
for i := 0; i < 2000000; i++ {
|
|
x := rand.Int31n(50)
|
|
if rand.Float64() < 0.01 {
|
|
x = rand.Int31n(9999)
|
|
}
|
|
|
|
key := fmt.Sprintf("key-%v", x)
|
|
expectedVal := "value for " + key
|
|
|
|
val, err := c.Get(key)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if val.(string) != expectedVal {
|
|
t.Fatal(val)
|
|
}
|
|
}
|
|
done <- true
|
|
}
|
|
|
|
func testCache(name string, c *Cache, t *testing.T) {
|
|
N := 8
|
|
done := make(chan bool, N)
|
|
|
|
for i := 0; i < N; i++ {
|
|
go testRunner(c, t, done)
|
|
}
|
|
|
|
for i := 0; i < N; i++ {
|
|
<-done
|
|
}
|
|
|
|
stats := c.GetStats()
|
|
|
|
fmt.Println(name)
|
|
fmt.Printf(" Hits: %d\n", stats.Hits)
|
|
fmt.Printf(" Misses: %d\n", stats.Misses)
|
|
fmt.Printf(" Hit-rate: %.2f%%\n",
|
|
100*float64(stats.Hits)/float64(stats.Hits+stats.Misses))
|
|
}
|
|
|
|
func TestCache(t *testing.T) {
|
|
c := New(Config{
|
|
MaxSize: 2048,
|
|
TTL: time.Second,
|
|
Src: func(key string) (interface{}, error) {
|
|
return fmt.Sprintf("value for %s", key), nil
|
|
},
|
|
})
|
|
testCache("LRUTimeout", c, t)
|
|
}
|