graphql-monitoring-proxy/cache/redis/redis_additional_test.go

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, err := New(&RedisClientConfig{
		RedisServer:   s.Addr(),
		RedisPassword: "",
		RedisDB:       0,
	})
	if err != nil {
		t.Fatalf("Failed to create Redis client: %v", err)
	}

	// Add some test data
	ttl := time.Duration(60) * time.Second
	err = redisConfig.Set("key1", []byte("value1"), ttl)
	assert.NoError(t, err)
	err = redisConfig.Set("key2", []byte("value2"), ttl)
	assert.NoError(t, err)
	err = redisConfig.Set("key3", []byte("value3"), ttl)
	assert.NoError(t, err)

	// Verify keys exist
	count, err := redisConfig.CountQueries()
	assert.NoError(t, err)
	assert.Equal(t, int64(3), count, "Expected 3 keys before clearing cache")

	// Clear the cache
	err = redisConfig.Clear()
	assert.NoError(t, err)

	// Verify all keys are gone
	count, err = redisConfig.CountQueries()
	assert.NoError(t, err)
	assert.Equal(t, int64(0), count, "Expected 0 keys after clearing cache")

	// Verify individual keys are gone
	_, found, err := redisConfig.Get("key1")
	assert.NoError(t, err)
	assert.False(t, found, "Key1 should be deleted after Clear")
	_, found, err = redisConfig.Get("key2")
	assert.NoError(t, err)
	assert.False(t, found, "Key2 should be deleted after Clear")
	_, found, err = redisConfig.Get("key3")
	assert.NoError(t, err)
	assert.False(t, found, "Key3 should be deleted after Clear")
}