shanshanzhong
9f4d71770b
Some checks failed
Build docker and publish / build (20.15.1) (push) Failing after 9m9s
110 lines
3.8 KiB
Go
110 lines
3.8 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/alicebob/miniredis/v2"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
func TestClearAllSessions_RemovesUserSessionsAndDeviceMappings(t *testing.T) {
|
|
logic, redisClient, cleanup := newDeleteAccountRedisTestLogic(t)
|
|
defer cleanup()
|
|
|
|
mustRedisSet(t, redisClient, "auth:session_id:sid-user-1", "1001")
|
|
mustRedisSet(t, redisClient, "auth:session_id:sid-user-2", "1001")
|
|
mustRedisSet(t, redisClient, "auth:session_id:sid-other", "2002")
|
|
|
|
mustRedisSet(t, redisClient, "auth:session_id:detail:sid-user-1", "detail")
|
|
mustRedisSet(t, redisClient, "auth:session_id:detail:sid-other", "detail")
|
|
|
|
mustRedisSet(t, redisClient, "auth:device_identifier:dev-user-1", "sid-user-1")
|
|
mustRedisSet(t, redisClient, "auth:device_identifier:dev-user-2", "sid-user-2")
|
|
mustRedisSet(t, redisClient, "auth:device_identifier:dev-other", "sid-other")
|
|
|
|
mustRedisZAdd(t, redisClient, "auth:user_sessions:1001", "sid-user-3", 1)
|
|
mustRedisSet(t, redisClient, "auth:session_id:sid-user-3", "1001")
|
|
|
|
logic.clearAllSessions(1001)
|
|
|
|
mustRedisNotExist(t, redisClient, "auth:session_id:sid-user-1")
|
|
mustRedisNotExist(t, redisClient, "auth:session_id:sid-user-2")
|
|
mustRedisNotExist(t, redisClient, "auth:session_id:sid-user-3")
|
|
mustRedisNotExist(t, redisClient, "auth:session_id:detail:sid-user-1")
|
|
mustRedisNotExist(t, redisClient, "auth:user_sessions:1001")
|
|
mustRedisNotExist(t, redisClient, "auth:device_identifier:dev-user-1")
|
|
mustRedisNotExist(t, redisClient, "auth:device_identifier:dev-user-2")
|
|
|
|
mustRedisExist(t, redisClient, "auth:session_id:sid-other")
|
|
mustRedisExist(t, redisClient, "auth:session_id:detail:sid-other")
|
|
mustRedisExist(t, redisClient, "auth:device_identifier:dev-other")
|
|
}
|
|
|
|
func TestClearAllSessions_ScanFallbackWorksWithoutUserSessionIndex(t *testing.T) {
|
|
logic, redisClient, cleanup := newDeleteAccountRedisTestLogic(t)
|
|
defer cleanup()
|
|
|
|
mustRedisSet(t, redisClient, "auth:session_id:sid-a", "3003")
|
|
mustRedisSet(t, redisClient, "auth:session_id:sid-b", "3003")
|
|
mustRedisSet(t, redisClient, "auth:session_id:sid-c", "4004")
|
|
|
|
logic.clearAllSessions(3003)
|
|
|
|
mustRedisNotExist(t, redisClient, "auth:session_id:sid-a")
|
|
mustRedisNotExist(t, redisClient, "auth:session_id:sid-b")
|
|
mustRedisExist(t, redisClient, "auth:session_id:sid-c")
|
|
}
|
|
|
|
func newDeleteAccountRedisTestLogic(t *testing.T) (*DeleteAccountLogic, *redis.Client, func()) {
|
|
t.Helper()
|
|
|
|
miniRedis := miniredis.RunT(t)
|
|
redisClient := redis.NewClient(&redis.Options{Addr: miniRedis.Addr()})
|
|
logic := NewDeleteAccountLogic(context.Background(), &svc.ServiceContext{Redis: redisClient})
|
|
|
|
cleanup := func() {
|
|
_ = redisClient.Close()
|
|
miniRedis.Close()
|
|
}
|
|
return logic, redisClient, cleanup
|
|
}
|
|
|
|
func mustRedisSet(t *testing.T, redisClient *redis.Client, key, value string) {
|
|
t.Helper()
|
|
if err := redisClient.Set(context.Background(), key, value, time.Hour).Err(); err != nil {
|
|
t.Fatalf("redis set %s failed: %v", key, err)
|
|
}
|
|
}
|
|
|
|
func mustRedisZAdd(t *testing.T, redisClient *redis.Client, key, member string, score float64) {
|
|
t.Helper()
|
|
if err := redisClient.ZAdd(context.Background(), key, redis.Z{Member: member, Score: score}).Err(); err != nil {
|
|
t.Fatalf("redis zadd %s failed: %v", key, err)
|
|
}
|
|
}
|
|
|
|
func mustRedisExist(t *testing.T, redisClient *redis.Client, key string) {
|
|
t.Helper()
|
|
exists, err := redisClient.Exists(context.Background(), key).Result()
|
|
if err != nil {
|
|
t.Fatalf("redis exists %s failed: %v", key, err)
|
|
}
|
|
if exists == 0 {
|
|
t.Fatalf("expected redis key %s to exist", key)
|
|
}
|
|
}
|
|
|
|
func mustRedisNotExist(t *testing.T, redisClient *redis.Client, key string) {
|
|
t.Helper()
|
|
exists, err := redisClient.Exists(context.Background(), key).Result()
|
|
if err != nil {
|
|
t.Fatalf("redis exists %s failed: %v", key, err)
|
|
}
|
|
if exists != 0 {
|
|
t.Fatalf("expected redis key %s to be deleted", key)
|
|
}
|
|
}
|