Function TestTaskStorage_ProcessResult

pkg/task/queue/task_storage_test.go:185–234

TestTaskStorage_ProcessResult tests the ProcessResult operation

(t *testing.T)

// TestTaskStorage_ProcessResult tests the ProcessResult operation
func TestTaskStorage_ProcessResult(t *testing.T) {
	ts := newTaskStorage()

	task1 := task.NewTask("test1")
	task2 := task.NewTask("test2")
	headTask := task.NewTask("head")
	tailTask := task.NewTask("tail")
	afterTask := task.NewTask("after")

	ts.AddLast(task1, task2)

	// Create a TaskResult for successful processing
	result := &TaskResult{
		Status: Success,
	}
	result.AddHeadTasks(headTask)
	result.AddTailTasks(tailTask)
	result.AddAfterTasks(afterTask)

	// Process the result for task1
	ts.ProcessResult(*result, task1)

	snapshot := ts.GetSnapshot()
	if len(snapshot) != 4 {
		t.Errorf("Expected 4 tasks after ProcessResult, got %d", len(snapshot))
	}

	// Order should be: headTask, afterTask, task2, tailTask
	if snapshot[0] != headTask || snapshot[1] != afterTask || snapshot[2] != task2 || snapshot[3] != tailTask {
		t.Error("ProcessResult failed: incorrect task order")
	}

	// Test with Fail status (task should not be removed)
	ts = newTaskStorage()
	ts.AddLast(task1)

	failResult := &TaskResult{Status: Fail}
	failResult.AddHeadTasks(task.NewTask("fail-head"))

	ts.ProcessResult(*failResult, task1)

	snapshot = ts.GetSnapshot()
	if len(snapshot) != 2 {
		t.Errorf("Expected 2 tasks after Fail ProcessResult, got %d", len(snapshot))
	}

	if snapshot[0].GetType() != "fail-head" || snapshot[1] != task1 {
		t.Error("ProcessResult with Fail status failed")
	}
}

// TestTaskStorage_SnapshotAndIteration tests GetSnapshot and Iterate operations
func TestTaskStorage_SnapshotAndIteration(t *testing.T) {