Function TestListCommand_LimitFlag

apps/cli/internal/cli/list_test.go:394–444

(t *testing.T)

}

func TestListCommand_LimitFlag(t *testing.T) {
	tasks := []*model.Task{
		{ID: "001", Title: "A", Status: model.StatusPending, Priority: model.PriorityLow, FilePath: "a.md"},
		{ID: "002", Title: "B", Status: model.StatusPending, Priority: model.PriorityHigh, FilePath: "b.md"},
		{ID: "003", Title: "C", Status: model.StatusPending, Priority: model.PriorityCritical, FilePath: "c.md"},
		{ID: "004", Title: "D", Status: model.StatusPending, Priority: model.PriorityMedium, FilePath: "d.md"},
		{ID: "005", Title: "E", Status: model.StatusPending, Priority: model.PriorityLow, FilePath: "e.md"},
	}

	tests := []struct {
		name          string
		limit         int
		sort          string
		expectedCount int
		firstID       string
	}{
		{"no limit", 0, "", 5, "001"},
		{"limit less than total", 3, "", 3, "001"},
		{"limit equal to total", 5, "", 5, "001"},
		{"limit greater than total", 10, "", 5, "001"},
		{"limit 1", 1, "", 1, "001"},
		{"limit with sort by priority", 2, "priority", 2, "003"},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			resetListFlags()
			listLimit = tt.limit

			tasksCopy := make([]*model.Task, len(tasks))
			copy(tasksCopy, tasks)

			if tt.sort != "" {
				if err := sortTasks(tasksCopy, tt.sort); err != nil {
					t.Fatalf("sortTasks() error: %v", err)
				}
			}

			if listLimit > 0 && listLimit < len(tasksCopy) {
				tasksCopy = tasksCopy[:listLimit]
			}

			if len(tasksCopy) != tt.expectedCount {
				t.Errorf("got %d tasks, want %d", len(tasksCopy), tt.expectedCount)
			}
			if tasksCopy[0].ID != tt.firstID {
				t.Errorf("first task ID = %s, want %s", tasksCopy[0].ID, tt.firstID)
			}
		})
	}
}

func TestListCommand_LimitTableOutput(t *testing.T) {
	resetListFlags()