snapshot of https://github.com/centrifugal/centrifugo.git (master, 2026-09-16; blobs > 131072B omitted)

internal/jwks/cache_ttl_test.go

@@ -0,0 +1,161 @@
+package jwks
+
+import (
+	"fmt"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestTTLCacheInit(t *testing.T) {
+	cache := NewTTLCache(5 * time.Minute)
+	require.NotNil(t, cache)
+}
+
+func TestTTLCacheAdd(t *testing.T) {
+	testCases := []struct {
+		Name string
+		TTL  time.Duration
+		Ops  int
+	}{
+		{
+			Name: "OK",
+			TTL:  5 * time.Second,
+			Ops:  100,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.Name, func(t *testing.T) {
+			cache := NewTTLCache(tc.TTL)
+			require.NotNil(t, cache)
+
+			for i := 0; i < tc.Ops; i++ {
+				kid := fmt.Sprintf("key-%d", i+1)
+				require.NoError(t, cache.Add(kid, &JWK{
+					Kid: kid,
+					Kty: "RSA",
+					Alg: "RS256",
+					Use: "sig",
+				}))
+			}
+		})
+	}
+}
+
+func TestTTLCacheGet(t *testing.T) {
+	testCases := []struct {
+		Name  string
+		Key   *JWK
+		Kid   string
+		Error error
+	}{
+		{
+			Name: "OK",
+			Key: &JWK{
+				Kid: "202101",
+				Kty: "RSA",
+				Alg: "RS256",
+				Use: "sig",
+			},
+			Kid: "202101",
+		},
+		{
+			Name: "NotFound",
+			Key: &JWK{
+				Kid: "202101",
+				Kty: "RSA",
+				Alg: "RS256",
+				Use: "sig",
+			},
+			Kid:   "202102",
+			Error: ErrCacheNotFound,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.Name, func(t *testing.T) {
+			cache := NewTTLCache(5 * time.Minute)
+			require.NotNil(t, cache)
+			require.NoError(t, cache.Add(tc.Key.Kid, tc.Key))
+
+			key, err := cache.Get(tc.Kid)
+			if tc.Error != nil {
+				require.Error(t, err)
+				require.ErrorIs(t, err, tc.Error)
+			} else {
+				require.NoError(t, err)
+				require.EqualValues(t, tc.Key, key)
+			}
+		})
+	}
+}
+
+func TestTTLCacheRemove(t *testing.T) {
+	testCases := []struct {
+		Name      string
+		NumAdd    int
+		NumDelete int
+		Len       int
+	}{
+		{
+			Name:      "OK",
+			NumAdd:    75,
+			NumDelete: 50,
+			Len:       25,
+		},
+		{
+			Name:      "RemoveUntilEmpty",
+			NumAdd:    75,
+			NumDelete: 100,
+			Len:       0,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.Name, func(t *testing.T) {
+			cache := NewTTLCache(5 * time.Minute)
+			require.NotNil(t, cache)
+
+			for i := 0; i < tc.NumAdd; i++ {
+				kid := fmt.Sprintf("key-%d", i+1)
+				require.NoError(t, cache.Add(kid, &JWK{
+					Kid: kid,
+					Kty: "RSA",
+					Alg: "RS256",
+					Use: "sig",
+				}))
+			}
+
+			for i := 0; i < tc.NumDelete; i++ {
+				kid := fmt.Sprintf("key-%d", i+1)
+				require.NoError(t, cache.remove(kid))
+			}
+
+			n, err := cache.Len()
+			require.NoError(t, err)
+			require.Equal(t, tc.Len, n)
+		})
+	}
+}
+
+func TestTTLCacheCleanup(t *testing.T) {
+	cache := NewTTLCache(1 * time.Millisecond)
+
+	for i := 0; i < 10; i++ {
+		kid := fmt.Sprintf("key-%d", i+1)
+		require.NoError(t, cache.Add(kid, &JWK{
+			Kid: kid,
+			Kty: "RSA",
+			Alg: "RS256",
+			Use: "sig",
+		}))
+	}
+
+	time.Sleep(2 * time.Second)
+
+	n, err := cache.Len()
+	require.NoError(t, err)
+	require.Equal(t, 0, n)
+}