lru: init

lru/lru_test.go

@@ -0,0 +1,78 @@
+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()
+}