mirror of
https://github.com/hanzoai/deploy.git
synced 2026-10-07 08:44:50 +00:00
30 lines
609 B
Go
30 lines
609 B
Go
package session
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/argoproj/argo-cd/v3/test"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUserStateStorage_LoadRevokedTokens(t *testing.T) {
|
|
redis, closer := test.NewInMemoryRedis()
|
|
defer closer()
|
|
|
|
err := redis.Set(t.Context(), revokedTokenPrefix+"abc", "", time.Hour).Err()
|
|
require.NoError(t, err)
|
|
|
|
ctx, cancel := context.WithCancel(t.Context())
|
|
defer cancel()
|
|
|
|
storage := NewUserStateStorage(redis)
|
|
storage.Init(ctx)
|
|
time.Sleep(time.Millisecond * 100)
|
|
|
|
assert.True(t, storage.IsTokenRevoked("abc"))
|
|
}
|