Function TestAutoArchiveInactiveChats

coderd/database/dbpurge/dbpurge_test.go:2676–3527

nolint:paralleltest // It uses LockIDDBPurge.

(t *testing.T)

//nolint:paralleltest // It uses LockIDDBPurge.
func TestAutoArchiveInactiveChats(t *testing.T) {
	now := time.Date(2025, 6, 15, 12, 0, 0, 0, time.UTC)

	tests := []struct {
		name string
		run  func(t *testing.T)
	}{
		{
			name: "AutoArchiveDisabled",
			run: func(t *testing.T) {
				h := newArchiveHarness(t, now)
				ctx, clk, db, rawDB, logger, deps := h.ctx, h.clk, h.db, h.rawDB, h.logger, h.deps

				require.Zero(t, codersdk.DefaultChatAutoArchiveDays)
				require.NoError(t, db.UpsertChatAutoArchiveDays(ctx, codersdk.DefaultChatAutoArchiveDays))

				// Chat older than any reasonable cutoff.
				staleChat := createArchiveChat(ctx, t, db, rawDB, deps, "stale-chat", now.Add(-365*24*time.Hour))

				auditor := audit.NewMock()
				auditorPtr := mockAuditorPtr(auditor)
				enqueuer := notificationstest.NewFakeEnqueuer()
				done := awaitDoTick(ctx, t, clk)
				closer := dbpurge.New(ctx, logger, db, &codersdk.DeploymentValues{}, prometheus.NewRegistry(), auditorPtr, dbpurge.WithNotificationsEnqueuer(enqueuer), dbpurge.WithClock(clk))
				defer closer.Close()
				testutil.TryReceive(ctx, t, done)

				// Not archived, no audits, no digests.
				refreshed, err := db.GetChatByID(ctx, staleChat.ID)
				require.NoError(t, err)
				require.False(t, refreshed.Archived, "chat should stay active when auto-archive is disabled")

				require.Empty(t, auditor.AuditLogs(), "no audit log entries expected")
				require.Empty(t, enqueuer.Sent(), "no digest notifications expected")
			},
		},
		{
			name: "ArchivesInactiveRoot",
			run: func(t *testing.T) {
				h := newArchiveHarness(t, now)
				ctx, clk, db, rawDB, logger, deps := h.ctx, h.clk, h.db, h.rawDB, h.logger, h.deps

				// Regression guard: ensure that both auto-archive and retention
				// are both set to a distinct non-zero value.
				require.NoError(t, db.UpsertChatAutoArchiveDays(ctx, int32(90)))
				require.NoError(t, db.UpsertChatRetentionDays(ctx, int32(30)))

				// Inactive root: newest message 100 days old.
				staleChat := createArchiveChat(ctx, t, db, rawDB, deps, "stale-chat", now.Add(-120*24*time.Hour))
				insertTextMessage(ctx, t, db, rawDB, staleChat.ID, deps.user.ID, deps.modelConfig.ID, now.Add(-100*24*time.Hour))

				// Active root: message 10 days old, within cutoff.
				activeChat := createArchiveChat(ctx, t, db, rawDB, deps, "active-chat", now.Add(-120*24*time.Hour))
				insertTextMessage(ctx, t, db, rawDB, activeChat.ID, deps.user.ID, deps.modelConfig.ID, now.Add(-10*24*time.Hour))

				auditor := audit.NewMock()
				auditorPtr := mockAuditorPtr(auditor)
				enqueuer := notificationstest.NewFakeEnqueuer()