Improve the test coverage

cache/memory/memory_additional_test.go

@@ -0,0 +1,90 @@
+package libpack_cache_memory
+
+import (
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+// Default constants for testing
+const (
+	DefaultTestExpiration = 5 * time.Second
+)
+
+func TestMemoryCacheClear(t *testing.T) {
+	cache := New(DefaultTestExpiration)
+	
+	// Add some entries
+	cache.Set("key1", []byte("value1"), DefaultTestExpiration)
+	cache.Set("key2", []byte("value2"), DefaultTestExpiration)
+	
+	// Verify entries exist
+	_, found := cache.Get("key1")
+	assert.True(t, found, "Expected key1 to exist before clearing cache")
+	
+	// Clear the cache
+	cache.Clear()
+	
+	// Verify cache is empty
+	_, found = cache.Get("key1")
+	assert.False(t, found, "Expected key1 to be removed after clearing cache")
+	_, found = cache.Get("key2")
+	assert.False(t, found, "Expected key2 to be removed after clearing cache")
+	
+	// Check that counter was reset
+	assert.Equal(t, int64(0), cache.CountQueries(), "Expected count to be 0 after clearing cache")
+}
+
+func TestMemoryCacheCountQueries(t *testing.T) {
+	cache := New(DefaultTestExpiration)
+	
+	// Check initial count
+	assert.Equal(t, int64(0), cache.CountQueries(), "Expected initial count to be 0")
+	
+	// Add some entries
+	cache.Set("key1", []byte("value1"), DefaultTestExpiration)
+	cache.Set("key2", []byte("value2"), DefaultTestExpiration)
+	cache.Set("key3", []byte("value3"), DefaultTestExpiration)
+	
+	// Check count
+	assert.Equal(t, int64(3), cache.CountQueries(), "Expected count to be 3 after adding 3 entries")
+	
+	// Delete an entry
+	cache.Delete("key1")
+	
+	// Check count after deletion
+	assert.Equal(t, int64(2), cache.CountQueries(), "Expected count to be 2 after deleting 1 entry")
+}
+
+func TestMemoryCacheCleanExpiredEntries(t *testing.T) {
+	// Create a cache with default expiration
+	cache := New(10 * time.Second)
+	
+	// Add an entry that will expire quickly
+	cache.Set("expire-soon", []byte("value1"), 10*time.Millisecond)
+	
+	// Add an entry that will not expire during the test
+	cache.Set("expire-later", []byte("value3"), 10*time.Minute)
+	
+	// Initial count should be 2
+	assert.Equal(t, int64(2), cache.CountQueries(), "Expected count to be 2 after adding entries")
+	
+	// Wait for short expiration
+	time.Sleep(20 * time.Millisecond)
+	
+	// Get the expired key directly to verify it's expired
+	_, expiredFound := cache.Get("expire-soon")
+	assert.False(t, expiredFound, "Key 'expire-soon' should be expired now")
+	
+	// Verify the not-expired key is still there
+	val, nonExpiredFound := cache.Get("expire-later")
+	assert.True(t, nonExpiredFound, "Key 'expire-later' should not be expired")
+	assert.Equal(t, []byte("value3"), val, "Expected correct value for 'expire-later'")
+	
+	// Manually clean expired entries
+	cache.CleanExpiredEntries()
+	
+	// Count should be 1 now (only the non-expired entry)
+	assert.Equal(t, int64(1), cache.CountQueries(), "Expected count to be 1 after cleaning expired entries")
+}

cache/redis/redis_additional_test.go

@@ -0,0 +1,50 @@
+package libpack_cache_redis
+
+import (
+	"testing"
+	"time"
+
+	"github.com/alicebob/miniredis/v2"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRedisClear(t *testing.T) {
+	// Create a mock Redis server
+	s, err := miniredis.Run()
+	if err != nil {
+		t.Fatalf("Failed to create mock redis server: %v", err)
+	}
+	defer s.Close()
+
+	// Create a Redis client
+	redisConfig := New(&RedisClientConfig{
+		RedisServer:   s.Addr(),
+		RedisPassword: "",
+		RedisDB:       0,
+	})
+
+	// Add some test data
+	ttl := time.Duration(60) * time.Second
+	redisConfig.Set("key1", []byte("value1"), ttl)
+	redisConfig.Set("key2", []byte("value2"), ttl)
+	redisConfig.Set("key3", []byte("value3"), ttl)
+
+	// Verify keys exist
+	count := redisConfig.CountQueries()
+	assert.Equal(t, int64(3), count, "Expected 3 keys before clearing cache")
+
+	// Clear the cache
+	redisConfig.Clear()
+
+	// Verify all keys are gone
+	count = redisConfig.CountQueries()
+	assert.Equal(t, int64(0), count, "Expected 0 keys after clearing cache")
+
+	// Verify individual keys are gone
+	_, found := redisConfig.Get("key1")
+	assert.False(t, found, "Key1 should be deleted after Clear")
+	_, found = redisConfig.Get("key2")
+	assert.False(t, found, "Key2 should be deleted after Clear")
+	_, found = redisConfig.Get("key3")
+	assert.False(t, found, "Key3 should be deleted after Clear")
+}