Function TestJobOperations

modules/backendscheduler/work/work_sharded_test.go:105–158

(t *testing.T)

}

func TestJobOperations(t *testing.T) {
	work := New(Config{})
	job := createTestJob("ops-test", tempopb.JobType_JOB_TYPE_COMPACTION)
	err := work.AddJob(job)
	require.NoError(t, err)

	t.Run("start job", func(t *testing.T) {
		work.StartJob("ops-test")
		retrievedJob := work.GetJob("ops-test")
		require.Equal(t, tempopb.JobStatus_JOB_STATUS_RUNNING, retrievedJob.Status)
		require.False(t, retrievedJob.StartTime.IsZero())
	})

	t.Run("complete job", func(t *testing.T) {
		work.CompleteJob("ops-test")
		retrievedJob := work.GetJob("ops-test")
		require.Equal(t, tempopb.JobStatus_JOB_STATUS_SUCCEEDED, retrievedJob.Status)
		require.False(t, retrievedJob.EndTime.IsZero())
	})

	t.Run("set compaction output", func(t *testing.T) {
		output := []string{"output1", "output2"}
		work.SetJobCompactionOutput("ops-test", output)
		retrievedJob := work.GetJob("ops-test")
		require.Equal(t, output, retrievedJob.GetCompactionOutput())
	})

	t.Run("fail job", func(t *testing.T) {
		failJob := createTestJob("fail-test", tempopb.JobType_JOB_TYPE_COMPACTION)
		err = work.AddJob(failJob)
		require.NoError(t, err)
		work.StartJob("fail-test")

		work.FailJob("fail-test")
		retrievedJob := work.GetJob("fail-test")
		require.Equal(t, tempopb.JobStatus_JOB_STATUS_FAILED, retrievedJob.Status)
		require.False(t, retrievedJob.EndTime.IsZero())
	})

	t.Run("remove job", func(t *testing.T) {
		removeJob := createTestJob("remove-test", tempopb.JobType_JOB_TYPE_COMPACTION)
		err = work.AddJob(removeJob)
		require.NoError(t, err)

		// Verify job exists
		require.NotNil(t, work.GetJob("remove-test"))

		// Remove job
		work.RemoveJob("remove-test")

		// Verify job is gone
		require.Nil(t, work.GetJob("remove-test"))
	})
}

func TestListJobs(t *testing.T) {
	work := New(Config{})