mirror of
https://github.com/luxfi/cache.git
synced 2026-09-24 00:24:18 +00:00
zeekayandHanzo Dev
d90e7473ee
luxfi/node required this module and also carried its own cache package, and the one file the two did not share was lru_sized_cache.go. It is here now, with the cachetest helper its test uses. It reached for utils.Zero, and luxfi/utils reaches constants, which reached geth, which reaches crypto — a chain no cache should pull in to name the zero value of a type parameter. The helper is four lines here instead. lru's TestCacheWithEvictionCallback fails before and after this commit; that directory is untouched. Co-authored-by: Hanzo Dev <[email protected]>
148 lines
2.9 KiB
Go
148 lines
2.9 KiB
Go
// Copyright (C) 2019-2025, Lux Industries Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package cachetest
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/luxfi/ids"
|
|
"github.com/luxfi/cache"
|
|
)
|
|
|
|
const IntSize = ids.IDLen + 8
|
|
|
|
func IntSizeFunc(ids.ID, int64) int {
|
|
return IntSize
|
|
}
|
|
|
|
// Tests is a list of all Cacher tests
|
|
var Tests = []struct {
|
|
Size int
|
|
Func func(t *testing.T, c cache.Cacher[ids.ID, int64])
|
|
}{
|
|
{Size: 1, Func: Basic},
|
|
{Size: 2, Func: Eviction},
|
|
}
|
|
|
|
func Basic(t *testing.T, cache cache.Cacher[ids.ID, int64]) {
|
|
require := require.New(t)
|
|
|
|
id1 := ids.ID{1}
|
|
_, found := cache.Get(id1)
|
|
require.False(found)
|
|
|
|
expectedValue1 := int64(1)
|
|
cache.Put(id1, expectedValue1)
|
|
value, found := cache.Get(id1)
|
|
require.True(found)
|
|
require.Equal(expectedValue1, value)
|
|
|
|
cache.Put(id1, expectedValue1)
|
|
value, found = cache.Get(id1)
|
|
require.True(found)
|
|
require.Equal(expectedValue1, value)
|
|
|
|
cache.Put(id1, expectedValue1)
|
|
value, found = cache.Get(id1)
|
|
require.True(found)
|
|
require.Equal(expectedValue1, value)
|
|
|
|
id2 := ids.ID{2}
|
|
|
|
expectedValue2 := int64(2)
|
|
cache.Put(id2, expectedValue2)
|
|
_, found = cache.Get(id1)
|
|
require.False(found)
|
|
|
|
value, found = cache.Get(id2)
|
|
require.True(found)
|
|
require.Equal(expectedValue2, value)
|
|
}
|
|
|
|
func Eviction(t *testing.T, cache cache.Cacher[ids.ID, int64]) {
|
|
require := require.New(t)
|
|
|
|
id1 := ids.ID{1}
|
|
id2 := ids.ID{2}
|
|
id3 := ids.ID{3}
|
|
|
|
expectedValue1 := int64(1)
|
|
expectedValue2 := int64(2)
|
|
expectedValue3 := int64(3)
|
|
|
|
require.Zero(cache.Len())
|
|
|
|
cache.Put(id1, expectedValue1)
|
|
|
|
require.Equal(1, cache.Len())
|
|
|
|
cache.Put(id2, expectedValue2)
|
|
|
|
require.Equal(2, cache.Len())
|
|
|
|
val, found := cache.Get(id1)
|
|
require.True(found)
|
|
require.Equal(expectedValue1, val)
|
|
|
|
val, found = cache.Get(id2)
|
|
require.True(found)
|
|
require.Equal(expectedValue2, val)
|
|
|
|
_, found = cache.Get(id3)
|
|
require.False(found)
|
|
|
|
cache.Put(id3, expectedValue3)
|
|
require.Equal(2, cache.Len())
|
|
|
|
_, found = cache.Get(id1)
|
|
require.False(found)
|
|
|
|
val, found = cache.Get(id2)
|
|
require.True(found)
|
|
require.Equal(expectedValue2, val)
|
|
|
|
val, found = cache.Get(id3)
|
|
require.True(found)
|
|
require.Equal(expectedValue3, val)
|
|
|
|
cache.Get(id2)
|
|
cache.Put(id1, expectedValue1)
|
|
|
|
val, found = cache.Get(id1)
|
|
require.True(found)
|
|
require.Equal(expectedValue1, val)
|
|
|
|
val, found = cache.Get(id2)
|
|
require.True(found)
|
|
require.Equal(expectedValue2, val)
|
|
|
|
_, found = cache.Get(id3)
|
|
require.False(found)
|
|
|
|
cache.Evict(id2)
|
|
cache.Put(id3, expectedValue3)
|
|
|
|
val, found = cache.Get(id1)
|
|
require.True(found)
|
|
require.Equal(expectedValue1, val)
|
|
|
|
_, found = cache.Get(id2)
|
|
require.False(found)
|
|
|
|
val, found = cache.Get(id3)
|
|
require.True(found)
|
|
require.Equal(expectedValue3, val)
|
|
|
|
cache.Flush()
|
|
|
|
_, found = cache.Get(id1)
|
|
require.False(found)
|
|
_, found = cache.Get(id2)
|
|
require.False(found)
|
|
_, found = cache.Get(id3)
|
|
require.False(found)
|
|
}
|