Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/repository/batch_image_download_limiter_test.go

@@ -0,0 +1,43 @@
+//go:build unit
+
+package repository
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/Wei-Shaw/sub2api/internal/service"
+	"github.com/alicebob/miniredis/v2"
+	"github.com/redis/go-redis/v9"
+	"github.com/stretchr/testify/require"
+)
+
+func TestBatchImageDownloadLimiter_AcquireDenyReleaseAndTTL(t *testing.T) {
+	ctx := context.Background()
+	mr := miniredis.RunT(t)
+	rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
+	t.Cleanup(func() { _ = rdb.Close() })
+	limiter := &batchImageDownloadLimiter{
+		rdb:          rdb,
+		activePrefix: defaultBatchImageDownloadActivePrefix,
+		maxActive:    1,
+		ttl:          time.Minute,
+	}
+
+	permit, err := limiter.Acquire(ctx, "11", "zip")
+	require.NoError(t, err)
+	require.NotNil(t, permit)
+	require.True(t, mr.TTL(limiter.activeKey("11")) > 0)
+
+	_, err = limiter.Acquire(ctx, "11", "zip")
+	require.ErrorIs(t, err, service.ErrBatchImageDownloadLimited)
+
+	require.NoError(t, permit.Release(ctx))
+	require.NoError(t, permit.Release(ctx))
+	require.False(t, mr.Exists(limiter.activeKey("11")))
+
+	permit, err = limiter.Acquire(ctx, "11", "zip")
+	require.NoError(t, err)
+	require.NotNil(t, permit)
+}