李建琦
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
80 lines
2.5 KiB
Go
80 lines
2.5 KiB
Go
//go:build unit
|
|
|
|
package repository
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/alicebob/miniredis/v2"
|
|
"github.com/redis/go-redis/v9"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func newLeaderLockTestCache(t *testing.T) (*leaderLockCache, *miniredis.Miniredis) {
|
|
t.Helper()
|
|
mr := miniredis.RunT(t)
|
|
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
|
|
t.Cleanup(func() { _ = rdb.Close() })
|
|
return &leaderLockCache{rdb: rdb}, mr
|
|
}
|
|
|
|
func TestLeaderLockCache_AcquireContendedRelease(t *testing.T) {
|
|
cache, _ := newLeaderLockTestCache(t)
|
|
ctx := context.Background()
|
|
const key = "dashboard:aggregation:leader"
|
|
|
|
ok, err := cache.TryAcquireLeaderLock(ctx, key, "A", time.Minute)
|
|
require.NoError(t, err)
|
|
require.True(t, ok, "first owner should acquire")
|
|
|
|
ok, err = cache.TryAcquireLeaderLock(ctx, key, "B", time.Minute)
|
|
require.NoError(t, err)
|
|
require.False(t, ok, "peer must be locked out while held")
|
|
|
|
require.NoError(t, cache.ReleaseLeaderLock(ctx, key, "A"))
|
|
|
|
ok, err = cache.TryAcquireLeaderLock(ctx, key, "B", time.Minute)
|
|
require.NoError(t, err)
|
|
require.True(t, ok, "peer should acquire after release")
|
|
}
|
|
|
|
// A stale owner whose lock expired and was re-acquired by a peer must not delete
|
|
// the peer's lock when its late release fires (compare-and-delete by owner).
|
|
func TestLeaderLockCache_ReleaseIsCompareAndDelete(t *testing.T) {
|
|
cache, _ := newLeaderLockTestCache(t)
|
|
ctx := context.Background()
|
|
const key = "payment:order:expiry:leader"
|
|
|
|
ok, err := cache.TryAcquireLeaderLock(ctx, key, "A", time.Minute)
|
|
require.NoError(t, err)
|
|
require.True(t, ok)
|
|
|
|
// Simulate A's lock expiring and peer B taking ownership.
|
|
require.NoError(t, cache.rdb.Set(ctx, leaderLockKeyPrefix+key, "B", time.Minute).Err())
|
|
|
|
// A's stale release must be a no-op against B's lock.
|
|
require.NoError(t, cache.ReleaseLeaderLock(ctx, key, "A"))
|
|
|
|
val, err := cache.rdb.Get(ctx, leaderLockKeyPrefix+key).Result()
|
|
require.NoError(t, err)
|
|
require.Equal(t, "B", val, "stale owner must not delete the new owner's lock")
|
|
}
|
|
|
|
func TestLeaderLockCache_TTLExpires(t *testing.T) {
|
|
cache, mr := newLeaderLockTestCache(t)
|
|
ctx := context.Background()
|
|
const key = "subscription:expiry:reminder:leader"
|
|
|
|
ok, err := cache.TryAcquireLeaderLock(ctx, key, "A", time.Minute)
|
|
require.NoError(t, err)
|
|
require.True(t, ok)
|
|
|
|
mr.FastForward(2 * time.Minute)
|
|
|
|
ok, err = cache.TryAcquireLeaderLock(ctx, key, "B", time.Minute)
|
|
require.NoError(t, err)
|
|
require.True(t, ok, "lock should be re-acquirable after the TTL expires")
|
|
}
|