Function TestDeleteOldAIBridgeRecords

coderd/database/dbpurge/dbpurge_test.go:1357–1591

(t *testing.T)

}

func TestDeleteOldAIBridgeRecords(t *testing.T) {
	t.Parallel()

	now := time.Date(2025, 1, 15, 7, 30, 0, 0, time.UTC)
	retentionPeriod := 30 * 24 * time.Hour                                // 30 days
	afterThreshold := now.Add(-retentionPeriod).Add(-24 * time.Hour)      // 31 days ago (older than threshold)
	beforeThreshold := now.Add(-15 * 24 * time.Hour)                      // 15 days ago (newer than threshold)
	closeBeforeThreshold := now.Add(-retentionPeriod).Add(24 * time.Hour) // 29 days ago

	type testFixtures struct {
		oldInterception            database.AIBridgeInterception
		oldInterceptionWithRelated database.AIBridgeInterception
		recentInterception         database.AIBridgeInterception
		nearThresholdInterception  database.AIBridgeInterception
	}

	testCases := []struct {
		name      string
		retention time.Duration
		verify    func(t *testing.T, ctx context.Context, db database.Store, fixtures testFixtures)
	}{
		{
			name:      "RetentionEnabled",
			retention: retentionPeriod,
			verify: func(t *testing.T, ctx context.Context, db database.Store, fixtures testFixtures) {
				t.Helper()

				interceptions, err := db.GetAIBridgeInterceptions(ctx)
				require.NoError(t, err)
				require.Len(t, interceptions, 2, "expected 2 interceptions remaining")

				interceptionIDs := make([]uuid.UUID, len(interceptions))
				for i, interception := range interceptions {
					interceptionIDs[i] = interception.ID
				}

				require.NotContains(t, interceptionIDs, fixtures.oldInterception.ID, "old interception should be deleted")
				require.NotContains(t, interceptionIDs, fixtures.oldInterceptionWithRelated.ID, "old interception with related records should be deleted")
				require.Contains(t, interceptionIDs, fixtures.recentInterception.ID, "recent interception should be kept")
				require.Contains(t, interceptionIDs, fixtures.nearThresholdInterception.ID, "near threshold interception should be kept")

				// Verify related records were deleted for old interception.
				oldTokenUsages, err := db.GetAIBridgeTokenUsagesByInterceptionID(ctx, fixtures.oldInterceptionWithRelated.ID)
				require.NoError(t, err)
				require.Empty(t, oldTokenUsages, "old token usages should be deleted")

				oldUserPrompts, err := db.GetAIBridgeUserPromptsByInterceptionID(ctx, fixtures.oldInterceptionWithRelated.ID)
				require.NoError(t, err)
				require.Empty(t, oldUserPrompts, "old user prompts should be deleted")

				oldToolUsages, err := db.GetAIBridgeToolUsagesByInterceptionID(ctx, fixtures.oldInterceptionWithRelated.ID)
				require.NoError(t, err)
				require.Empty(t, oldToolUsages, "old tool usages should be deleted")

				// Verify related records were NOT deleted for near-threshold interception.
				newTokenUsages, err := db.GetAIBridgeTokenUsagesByInterceptionID(ctx, fixtures.nearThresholdInterception.ID)
				require.NoError(t, err)
				require.Len(t, newTokenUsages, 1, "near threshold token usages should not be deleted")