79 lines
1.8 KiB
Go
79 lines
1.8 KiB
Go
package lru
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestTtl(t *testing.T) {
|
|
cache := New[int](1)
|
|
|
|
cache.Set("first", 1, 500*time.Millisecond)
|
|
|
|
value, ok := cache.Get("first")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, 1, value)
|
|
|
|
time.Sleep(600 * time.Millisecond)
|
|
|
|
value, ok = cache.Get("first")
|
|
assert.False(t, ok)
|
|
assert.Equal(t, 0, value)
|
|
}
|
|
|
|
func TestEvictFirst(t *testing.T) {
|
|
cache := New[int](2)
|
|
|
|
assert.False(t, cache.Set("first", 1, 500*time.Millisecond))
|
|
assert.False(t, cache.Set("second", 2, 500*time.Millisecond))
|
|
assert.True(t, cache.Set("third", 3, 500*time.Millisecond))
|
|
|
|
_, ok := cache.Get("first")
|
|
assert.False(t, ok)
|
|
}
|
|
|
|
func TestEvictExpired(t *testing.T) {
|
|
cache := New[int](2)
|
|
|
|
assert.False(t, cache.Set("first", 1, 1000*time.Millisecond))
|
|
assert.False(t, cache.Set("second", 2, 100*time.Millisecond))
|
|
|
|
time.Sleep(200*time.Millisecond)
|
|
|
|
assert.True(t, cache.Set("third", 3, 1000*time.Millisecond))
|
|
|
|
var ok bool
|
|
_, ok = cache.Get("first")
|
|
assert.True(t, ok)
|
|
_, ok = cache.Get("third")
|
|
assert.True(t, ok)
|
|
_, ok = cache.Get("second")
|
|
assert.False(t, ok)
|
|
}
|
|
|
|
func TestSetExisted(t *testing.T) {
|
|
cache := New[int](2)
|
|
|
|
assert.False(t, cache.Set("first", 1, 200*time.Millisecond))
|
|
assert.False(t, cache.Set("second", 2, 500*time.Millisecond))
|
|
|
|
time.Sleep(100*time.Millisecond)
|
|
|
|
assert.False(t, cache.Set("first", 11, 1000*time.Millisecond))
|
|
assert.True(t, cache.Set("third", 3, 1000*time.Millisecond))
|
|
|
|
value, ok := cache.Get("first") // exists with updated ttl
|
|
assert.True(t, ok)
|
|
assert.Equal(t, 11, value)
|
|
_, ok = cache.Get("third")
|
|
assert.True(t, ok)
|
|
_, ok = cache.Get("second") // evicted
|
|
assert.False(t, ok)
|
|
}
|
|
|
|
func TestDummy(t *testing.T) { // for coverage 100% coverage
|
|
cache := New[int](0)
|
|
cache.evict()
|
|
}
|