李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
//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)
|
|
}
|