✨ feat(user): magic-link expired-token cleanup loop (ADR-0028 Phase A consequence) (#65)
Co-authored-by: Gabriel Radureau <arcodange@gmail.com> Co-committed-by: Gabriel Radureau <arcodange@gmail.com>
This commit was merged in pull request #65.
This commit is contained in:
64
pkg/user/magic_link_cleanup_test.go
Normal file
64
pkg/user/magic_link_cleanup_test.go
Normal file
@@ -0,0 +1,64 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
type fakeMLRepo struct {
|
||||
deleteN int64
|
||||
deleteErr error
|
||||
cutoffSeen time.Time
|
||||
}
|
||||
|
||||
func (r *fakeMLRepo) CreateMagicLinkToken(_ context.Context, _ *MagicLinkToken) error { return nil }
|
||||
func (r *fakeMLRepo) GetMagicLinkTokenByHash(_ context.Context, _ string) (*MagicLinkToken, error) {
|
||||
return nil, nil
|
||||
}
|
||||
func (r *fakeMLRepo) MarkMagicLinkTokenConsumed(_ context.Context, _ uint, _ time.Time) error {
|
||||
return nil
|
||||
}
|
||||
func (r *fakeMLRepo) DeleteExpiredMagicLinkTokens(_ context.Context, before time.Time) (int64, error) {
|
||||
r.cutoffSeen = before
|
||||
return r.deleteN, r.deleteErr
|
||||
}
|
||||
|
||||
func TestRunOnce_ReturnsCount(t *testing.T) {
|
||||
repo := &fakeMLRepo{deleteN: 7}
|
||||
r := NewMagicLinkCleanupRunner(repo)
|
||||
n, err := r.runOnce(context.Background())
|
||||
require.NoError(t, err)
|
||||
assert.EqualValues(t, 7, n)
|
||||
assert.WithinDuration(t, time.Now(), repo.cutoffSeen, time.Second)
|
||||
}
|
||||
|
||||
func TestRunOnce_PropagatesError(t *testing.T) {
|
||||
repo := &fakeMLRepo{deleteErr: errors.New("simulated")}
|
||||
r := NewMagicLinkCleanupRunner(repo)
|
||||
_, err := r.runOnce(context.Background())
|
||||
require.Error(t, err)
|
||||
}
|
||||
|
||||
func TestStartCleanupLoop_StopsOnContextCancel(t *testing.T) {
|
||||
repo := &fakeMLRepo{}
|
||||
r := NewMagicLinkCleanupRunner(repo)
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
r.StartCleanupLoop(ctx, 10*time.Millisecond)
|
||||
time.Sleep(25 * time.Millisecond) // 2 ticks
|
||||
cancel()
|
||||
time.Sleep(15 * time.Millisecond) // give the goroutine time to exit
|
||||
// Implicit assertion: no goroutine leak (test would hang in -race mode otherwise).
|
||||
}
|
||||
|
||||
func TestStartCleanupLoop_NoOpWhenIntervalZero(t *testing.T) {
|
||||
repo := &fakeMLRepo{}
|
||||
r := NewMagicLinkCleanupRunner(repo)
|
||||
r.StartCleanupLoop(context.Background(), 0)
|
||||
// Just make sure no goroutine is started ; nothing observable to assert
|
||||
// beyond "no panic, returns immediately".
|
||||
}
|
||||
Reference in New Issue
Block a user