Function TestTaskStorage_BasicOperations

pkg/task/queue/task_storage_test.go:13–105

TestTaskStorage_BasicOperations tests basic add, get, and remove operations

(t *testing.T)

// TestTaskStorage_BasicOperations tests basic add, get, and remove operations
func TestTaskStorage_BasicOperations(t *testing.T) {
	ts := newTaskStorage()

	// Test empty storage
	if ts.Length() != 0 {
		t.Errorf("Expected length 0, got %d", ts.Length())
	}

	if ts.GetFirst() != nil {
		t.Error("Expected GetFirst() to return nil for empty storage")
	}

	if ts.GetLast() != nil {
		t.Error("Expected GetLast() to return nil for empty storage")
	}

	// Create some tasks
	task1 := task.NewTask("test1")
	task2 := task.NewTask("test2")
	task3 := task.NewTask("test3")

	// Test AddLast
	ts.AddLast(task1, task2)
	if ts.Length() != 2 {
		t.Errorf("Expected length 2, got %d", ts.Length())
	}

	if ts.GetFirst() != task1 {
		t.Error("Expected first task to be task1")
	}

	if ts.GetLast() != task2 {
		t.Error("Expected last task to be task2")
	}

	// Test Get by ID
	if ts.Get(task1.GetId()) != task1 {
		t.Error("Expected to get task1 by ID")
	}

	if ts.Get("nonexistent") != nil {
		t.Error("Expected Get(nonexistent) to return nil")
	}

	// Test AddFirst
	ts.AddFirst(task3)
	if ts.Length() != 3 {
		t.Errorf("Expected length 3, got %d", ts.Length())
	}

	if ts.GetFirst() != task3 {
		t.Error("Expected first task to be task3 after AddFirst")
	}

	// Test RemoveFirst
	removed := ts.RemoveFirst()
	if removed != task3 {
		t.Error("Expected RemoveFirst to return task3")