Function TestPrune

modules/backendscheduler/work/work_sharded_test.go:227–285

(t *testing.T)

}

func TestPrune(t *testing.T) {
	cfg := Config{
		PruneAge:       time.Hour,
		DeadJobTimeout: 30 * time.Minute,
	}
	work := New(cfg)
	ctx := context.Background()

	now := time.Now()

	var err error

	// Add old completed job (should be pruned)
	oldCompleted := createTestJob("old-completed", tempopb.JobType_JOB_TYPE_COMPACTION)
	err = work.AddJob(oldCompleted)
	require.NoError(t, err)
	work.CompleteJob(oldCompleted.ID) // Sets status to SUCCEEDED
	retrievedOldCompleted := work.GetJob(oldCompleted.ID)
	retrievedOldCompleted.EndTime = now.Add(-2 * time.Hour)

	// Add recent completed job (should not be pruned)
	recentCompleted := createTestJob("recent-completed", tempopb.JobType_JOB_TYPE_COMPACTION)
	err = work.AddJob(recentCompleted)
	require.NoError(t, err)
	work.CompleteJob(recentCompleted.ID) // Sets status to SUCCEEDED
	retrievedRecentCompleted := work.GetJob(recentCompleted.ID)
	retrievedRecentCompleted.EndTime = now.Add(-10 * time.Minute)

	// Add old running job (should be failed)
	oldRunning := createTestJob("old-running", tempopb.JobType_JOB_TYPE_COMPACTION)
	err = work.AddJob(oldRunning)
	require.NoError(t, err)
	work.StartJob(oldRunning.ID) // Sets status to RUNNING
	retrievedOldRunning := work.GetJob(oldRunning.ID)
	retrievedOldRunning.StartTime = now.Add(-time.Hour)

	// Add recent running job (should remain running)
	recentRunning := createTestJob("recent-running", tempopb.JobType_JOB_TYPE_COMPACTION)
	err = work.AddJob(recentRunning)
	require.NoError(t, err)
	work.StartJob(recentRunning.ID) // Sets status to RUNNING
	retrievedRecentRunning := work.GetJob(recentRunning.ID)
	retrievedRecentRunning.StartTime = now.Add(-10 * time.Minute)

	// Run prune
	work.Prune(ctx)

	// Verify results
	require.Nil(t, work.GetJob("old-completed"), "old completed job should be pruned")
	require.NotNil(t, work.GetJob("recent-completed"), "recent completed job should remain")

	oldRunningJob := work.GetJob("old-running")
	require.NotNil(t, oldRunningJob, "old running job should still exist")
	require.Equal(t, tempopb.JobStatus_JOB_STATUS_FAILED, oldRunningJob.Status, "old running job should be failed")

	recentRunningJob := work.GetJob("recent-running")
	require.NotNil(t, recentRunningJob, "recent running job should remain")
	require.Equal(t, tempopb.JobStatus_JOB_STATUS_RUNNING, recentRunningJob.Status, "recent running job should stay running")