You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
640 B
35 lines
640 B
2 years ago
|
package lfu_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"cache/lfu"
|
||
|
|
"github.com/matryer/is"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSet(t *testing.T) {
|
||
|
|
is := is.New(t)
|
||
|
|
cache := lfu.New(24, nil)
|
||
|
|
cache.DelOldest()
|
||
|
|
cache.Set("k1", 1)
|
||
|
|
v := cache.Get("k1")
|
||
|
|
is.Equal(v, 1)
|
||
|
|
cache.Del("k1")
|
||
|
|
is.Equal(0, cache.Len())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestOnEvicted(t *testing.T) {
|
||
|
|
is := is.New(t)
|
||
|
|
keys := make([]string, 0, 8)
|
||
|
|
onEvicted := func(key string, value interface{}) {
|
||
|
|
keys = append(keys, key)
|
||
|
|
}
|
||
|
|
cache := lfu.New(32, onEvicted)
|
||
|
|
cache.Set("k1", 1)
|
||
|
|
cache.Set("k2", 2)
|
||
|
|
cache.Set("k3", 3)
|
||
|
|
cache.Set("k4", 4)
|
||
|
|
expected := []string{"k1", "k3"}
|
||
|
|
is.Equal(expected, keys)
|
||
|
|
is.Equal(2, cache.Len())
|
||
|
|
}
|