added tests for eviction strategy

Author: orr-a-security

pkg/storage/storagedb_test.go

@@ -133,3 +133,49 @@ func doStuffWithOtherCache(cache cache.Cache) {
 		_, _ = cache.GetIfPresent(strconv.Itoa(i))
 	}
 }
+
+func TestSlidingEvictionStrategy(t *testing.T) {
+	testTTL := 100 * time.Millisecond
+	smallDelay := 10 * time.Millisecond
+	mem, err := New(&Options{EvictionTTL: testTTL, EvictionStrategy: EvictionStrategySliding})
+	require.Nil(t, err)
+	defer mem.Close()
+
+	err = mem.SetID("test-sliding")
+	require.Nil(t, err)
+
+	// Access after half TTL - should extend expiration
+	time.Sleep(testTTL / 2)
+	_, ok := mem.cache.GetIfPresent("test-sliding")
+	require.True(t, ok)
+
+	// Still present after original TTL due to sliding window
+	time.Sleep(testTTL / 2 + smallDelay)
+	_, ok = mem.cache.GetIfPresent("test-sliding")
+	require.True(t, ok)
+
+	// Should be expired after full TTL despite access
+	time.Sleep(testTTL + smallDelay)
+	_, ok = mem.cache.GetIfPresent("test-sliding")
+	require.False(t, ok)
+}
+
+func TestFixedEvictionStrategy(t *testing.T) {
+	testTTL := 100 * time.Millisecond
+	mem, err := New(&Options{EvictionTTL: testTTL, EvictionStrategy: EvictionStrategyFixed})
+	require.Nil(t, err)
+	defer mem.Close()
+
+	err = mem.SetID("test-fixed")
+	require.Nil(t, err)
+
+	// Access after half TTL - should NOT extend expiration
+	time.Sleep(testTTL / 2)
+	_, ok := mem.cache.GetIfPresent("test-fixed")
+	require.True(t, ok)
+
+	// Should be expired after full TTL despite access
+	time.Sleep(testTTL / 2 + 10 * time.Millisecond)
+	_, ok = mem.cache.GetIfPresent("test-fixed")
+	require.False(t, ok)
+}