token cleanup consistency

This commit is contained in:
Scott McKendry
2026-05-17 18:42:07 +12:00
parent 225041126e
commit 6d6b53270d
2 changed files with 4 additions and 10 deletions
+3 -9
View File
@@ -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)
+1 -1
View File
@@ -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)
} }