feat(jobs): mcp_tokens_purge runner (spec §6.5)

- Periodic delete of mcp_tokens where (expired OR revoked) before retention - 24h interval default, 7d retention default - Mirrors acp_events_purge pattern: idempotent, error-tolerant Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-05-08 12:56:10 +08:00
parent 702daced7d
commit 146dca7738
2 changed files with 103 additions and 0 deletions
@@ -0,0 +1,62 @@
+package runners_test
+
+import (
+	"context"
+	"errors"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/yan1h/agent-coding-workflow/internal/jobs/runners"
+)
+
+type fakeMCPTokensPurgeRepo struct {
+	purged         int64
+	purgeErr       error
+	capturedCutoff time.Time
+	called         bool
+}
+
+func (f *fakeMCPTokensPurgeRepo) PurgeExpired(_ context.Context, before time.Time) (int64, error) {
+	f.called = true
+	f.capturedCutoff = before
+	if f.purgeErr != nil {
+		return 0, f.purgeErr
+	}
+	return f.purged, nil
+}
+
+func TestMCPTokensPurge_RunCallsPurgeWithCorrectCutoff(t *testing.T) {
+	t.Parallel()
+	repo := &fakeMCPTokensPurgeRepo{purged: 4}
+	retention := 168 * time.Hour
+	r := runners.NewMCPTokensPurge(repo, retention, discardLogger())
+
+	before := time.Now()
+	r.Run(context.Background())
+	after := time.Now()
+
+	require.True(t, repo.called)
+	expectedFloor := before.Add(-retention).Add(-time.Second)
+	expectedCeil := after.Add(-retention).Add(time.Second)
+	assert.True(t, !repo.capturedCutoff.Before(expectedFloor) && !repo.capturedCutoff.After(expectedCeil),
+		"cutoff %s outside [%s, %s]", repo.capturedCutoff, expectedFloor, expectedCeil)
+}
+
+func TestMCPTokensPurge_RunErrorDoesNotPanic(t *testing.T) {
+	t.Parallel()
+	repo := &fakeMCPTokensPurgeRepo{purgeErr: errors.New("db down")}
+	r := runners.NewMCPTokensPurge(repo, time.Hour, discardLogger())
+	require.NotPanics(t, func() { r.Run(context.Background()) })
+	assert.True(t, repo.called)
+}
+
+func TestMCPTokensPurge_NoRowsIsNoOp(t *testing.T) {
+	t.Parallel()
+	repo := &fakeMCPTokensPurgeRepo{purged: 0}
+	r := runners.NewMCPTokensPurge(repo, time.Hour, discardLogger())
+	require.NotPanics(t, func() { r.Run(context.Background()) })
+	assert.True(t, repo.called)
+}