mirror of
https://github.com/steveiliop56/tinyauth.git
synced 2026-05-17 09:40:14 +00:00
token cleanup consistency
This commit is contained in:
@@ -402,16 +402,10 @@ func TestMemoryStore(t *testing.T) {
|
|||||||
{
|
{
|
||||||
description: "Delete expired OIDC tokens",
|
description: "Delete expired OIDC tokens",
|
||||||
run: func(t *testing.T, s repository.Store) {
|
run: func(t *testing.T, s repository.Store) {
|
||||||
// expired by TokenExpiresAt
|
// both expiries past
|
||||||
_, err := s.CreateOidcToken(ctx, repository.CreateOidcTokenParams{
|
_, err := s.CreateOidcToken(ctx, repository.CreateOidcTokenParams{
|
||||||
Sub: "sub-1", AccessTokenHash: "at-1",
|
Sub: "sub-1", AccessTokenHash: "at-1",
|
||||||
TokenExpiresAt: 10, RefreshTokenExpiresAt: 100,
|
TokenExpiresAt: 10, RefreshTokenExpiresAt: 10,
|
||||||
})
|
|
||||||
require.NoError(t, err)
|
|
||||||
// expired by RefreshTokenExpiresAt
|
|
||||||
_, err = s.CreateOidcToken(ctx, repository.CreateOidcTokenParams{
|
|
||||||
Sub: "sub-2", AccessTokenHash: "at-2",
|
|
||||||
TokenExpiresAt: 100, RefreshTokenExpiresAt: 10,
|
|
||||||
})
|
})
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
// valid
|
// valid
|
||||||
@@ -426,7 +420,7 @@ func TestMemoryStore(t *testing.T) {
|
|||||||
RefreshTokenExpiresAt: 50,
|
RefreshTokenExpiresAt: 50,
|
||||||
})
|
})
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
assert.Len(t, deleted, 2)
|
assert.Len(t, deleted, 1)
|
||||||
|
|
||||||
_, err = s.GetOidcToken(ctx, "at-3")
|
_, err = s.GetOidcToken(ctx, "at-3")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|||||||
@@ -207,7 +207,7 @@ func (s *Store) DeleteExpiredOidcTokens(_ context.Context, arg repository.Delete
|
|||||||
defer s.mu.Unlock()
|
defer s.mu.Unlock()
|
||||||
var deleted []repository.OidcToken
|
var deleted []repository.OidcToken
|
||||||
for k, t := range s.oidcTokens {
|
for k, t := range s.oidcTokens {
|
||||||
if t.TokenExpiresAt < arg.TokenExpiresAt || t.RefreshTokenExpiresAt < arg.RefreshTokenExpiresAt {
|
if t.TokenExpiresAt < arg.TokenExpiresAt && t.RefreshTokenExpiresAt < arg.RefreshTokenExpiresAt {
|
||||||
deleted = append(deleted, t)
|
deleted = append(deleted, t)
|
||||||
delete(s.oidcTokens, k)
|
delete(s.oidcTokens, k)
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user