Function TestTaskStorage_AddBeforeAfter

pkg/task/queue/task_storage_test.go:108–148

TestTaskStorage_AddBeforeAfter tests AddBefore and AddAfter operations

(t *testing.T)

// TestTaskStorage_AddBeforeAfter tests AddBefore and AddAfter operations
func TestTaskStorage_AddBeforeAfter(t *testing.T) {
	ts := newTaskStorage()

	task1 := task.NewTask("test1")
	task2 := task.NewTask("test2")
	task3 := task.NewTask("test3")
	task4 := task.NewTask("test4")

	// Add initial tasks
	ts.AddLast(task1, task2)

	// Test AddAfter
	ts.AddAfter(task1.GetId(), task3)
	snapshot := ts.GetSnapshot()
	if len(snapshot) != 3 {
		t.Errorf("Expected 3 tasks, got %d", len(snapshot))
	}

	if snapshot[0] != task1 || snapshot[1] != task3 || snapshot[2] != task2 {
		t.Error("AddAfter failed: incorrect order")
	}

	// Test AddBefore
	ts.AddBefore(task2.GetId(), task4)
	snapshot = ts.GetSnapshot()
	if len(snapshot) != 4 {
		t.Errorf("Expected 4 tasks, got %d", len(snapshot))
	}

	if snapshot[0] != task1 || snapshot[1] != task3 || snapshot[2] != task4 || snapshot[3] != task2 {
		t.Error("AddBefore failed: incorrect order")
	}

	// Test AddAfter/AddBefore with non-existent ID
	ts.AddAfter("nonexistent", task.NewTask("should-not-be-added"))
	ts.AddBefore("nonexistent", task.NewTask("should-not-be-added"))

	if ts.Length() != 4 {
		t.Errorf("Expected length 4 (no tasks should be added for non-existent IDs), got %d", ts.Length())
	}
}

// TestTaskStorage_DeleteFunc tests the DeleteFunc operation
func TestTaskStorage_DeleteFunc(t *testing.T) {