mirror of
https://github.com/SagerNet/sing.git
synced 2025-04-03 11:57:39 +03:00
95 lines
2.7 KiB
Go
95 lines
2.7 KiB
Go
package freelru_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/sagernet/sing/contrab/freelru"
|
|
"github.com/sagernet/sing/contrab/maphash"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUpdateLifetimeOnGet(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.SetUpdateLifetimeOnGet(true)
|
|
lru.AddWithLifetime("hello", "world", 2*time.Second)
|
|
time.Sleep(time.Second)
|
|
_, ok := lru.Get("hello")
|
|
require.True(t, ok)
|
|
time.Sleep(time.Second + time.Millisecond*100)
|
|
_, ok = lru.Get("hello")
|
|
require.True(t, ok)
|
|
}
|
|
|
|
func TestUpdateLifetimeOnGet1(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.SetUpdateLifetimeOnGet(true)
|
|
lru.AddWithLifetime("hello", "world", 2*time.Second)
|
|
time.Sleep(time.Second)
|
|
lru.Peek("hello")
|
|
time.Sleep(time.Second + time.Millisecond*100)
|
|
_, ok := lru.Get("hello")
|
|
require.False(t, ok)
|
|
}
|
|
|
|
func TestUpdateLifetime(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.Add("hello", "world")
|
|
require.True(t, lru.UpdateLifetime("hello", "world", 2*time.Second))
|
|
time.Sleep(time.Second)
|
|
_, ok := lru.Get("hello")
|
|
require.True(t, ok)
|
|
time.Sleep(time.Second + time.Millisecond*100)
|
|
_, ok = lru.Get("hello")
|
|
require.False(t, ok)
|
|
}
|
|
|
|
func TestUpdateLifetime1(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.Add("hello", "world")
|
|
require.False(t, lru.UpdateLifetime("hello", "not world", 2*time.Second))
|
|
time.Sleep(2*time.Second + time.Millisecond*100)
|
|
_, ok := lru.Get("hello")
|
|
require.True(t, ok)
|
|
}
|
|
|
|
func TestUpdateLifetime2(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.AddWithLifetime("hello", "world", 2*time.Second)
|
|
time.Sleep(time.Second)
|
|
require.True(t, lru.UpdateLifetime("hello", "world", 2*time.Second))
|
|
time.Sleep(time.Second + time.Millisecond*100)
|
|
_, ok := lru.Get("hello")
|
|
require.True(t, ok)
|
|
time.Sleep(time.Second + time.Millisecond*100)
|
|
_, ok = lru.Get("hello")
|
|
require.False(t, ok)
|
|
}
|
|
|
|
func TestPeekWithLifetime(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.SetLifetime(time.Second)
|
|
lru.Add("1", "")
|
|
time.Sleep(300 * time.Millisecond)
|
|
lru.Add("2", "")
|
|
time.Sleep(300 * time.Millisecond)
|
|
lru.Add("3", "")
|
|
time.Sleep(300 * time.Millisecond)
|
|
lru.Add("4", "")
|
|
time.Sleep(time.Second)
|
|
lru.PurgeExpired()
|
|
require.Equal(t, 0, lru.Len())
|
|
}
|