DEVPROD-11189 Resolve execution and display tasks correctly for API Patches

model/project.go

@@ -1683,6 +1683,28 @@ func (p *Project) IgnoresAllFiles(files []string) bool {
 // unmatched requester (e.g. a patch-only task for a mainline commit).
 func (p *Project) BuildProjectTVPairs(patchDoc *patch.Patch, alias string) {
 	patchDoc.BuildVariants, patchDoc.Tasks, patchDoc.VariantsTasks = p.ResolvePatchVTs(patchDoc, patchDoc.GetRequester(), alias, true)
+
+	// Connect the execution tasks to the display tasks.
+	displayTasksToExecTasks := map[string][]string{}
+	for _, bv := range p.BuildVariants {
+		for _, dt := range bv.DisplayTasks {
+			displayTasksToExecTasks[dt.Name] = dt.ExecTasks
+		}
+	}
+
+	vts := []patch.VariantTasks{}
+	for _, vt := range patchDoc.VariantsTasks {
+		dts := []patch.DisplayTask{}
+		for _, dt := range vt.DisplayTasks {
+			if ets, ok := displayTasksToExecTasks[dt.Name]; ok {
+				dt.ExecTasks = ets
+			}
+			dts = append(dts, dt)
+		}
+		vt.DisplayTasks = dts
+		vts = append(vts, vt)
+	}
+	patchDoc.VariantsTasks = vts
 }
 
 // ResolvePatchVTs resolves a list of build variants and tasks into a list of

model/project_test.go

@@ -1371,7 +1371,10 @@ func (s *projectSuite) TestBuildProjectTVPairsWithDisplayTaskWithDependencies()
 			s.Contains(vt.Tasks, "very_task")
 			s.Require().Len(vt.DisplayTasks, 1)
 			s.Equal("memes", vt.DisplayTasks[0].Name)
-			s.Empty(vt.DisplayTasks[0].ExecTasks)
+			s.Len(vt.DisplayTasks[0].ExecTasks, 3)
+			s.Contains(vt.DisplayTasks[0].ExecTasks, "9001_task")
+			s.Contains(vt.DisplayTasks[0].ExecTasks, "very_task")
+			s.Contains(vt.DisplayTasks[0].ExecTasks, "another_disabled_task")
 		} else if vt.Variant == "bv_2" {
 			s.Len(vt.Tasks, 1)
 			s.Contains(vt.Tasks, "a_task_2")

rest/model/patch.go

@@ -77,6 +77,7 @@ type ChildPatch struct {
 	PatchID *string `json:"patch_id"`
 	Status  *string `json:"status"`
 }
+
 type VariantTask struct {
 	// Name of build variant
 	Name *string `json:"name"`
@@ -149,6 +150,7 @@ type APIPatchArgs struct {
 }
 
 // BuildFromService converts from service level structs to an APIPatch.
+// An APIPatch expects the VariantTasks to be populated with only non-execution tasks and display tasks.
 // If args are set, includes identifier, commit queue position, and/or child patches from the DB, if applicable.
 func (apiPatch *APIPatch) BuildFromService(p patch.Patch, args *APIPatchArgs) error {
 	apiPatch.buildBasePatch(p)
@@ -242,16 +244,39 @@ func (apiPatch *APIPatch) buildBasePatch(p patch.Patch) {
 	}
 	apiPatch.Tasks = tasks
 	variantTasks := []VariantTask{}
+
+	// We remove the execution tasks from selected display tasks to avoid duplication.
+	execTasksToRemove := []string{}
 	for _, vt := range p.VariantsTasks {
 		vtasks := make([]*string, 0)
 		for _, task := range vt.Tasks {
 			vtasks = append(vtasks, utility.ToStringPtr(task))
 		}
+		for _, task := range vt.DisplayTasks {
+			vtasks = append(vtasks, utility.ToStringPtr(task.Name))
+			execTasksToRemove = append(execTasksToRemove, task.ExecTasks...)
+		}
 		variantTasks = append(variantTasks, VariantTask{
 			Name:  utility.ToStringPtr(vt.Variant),
 			Tasks: vtasks,
 		})
 	}
+	for i, vt := range variantTasks {
+		tasks := []*string{}
+		for j, t := range vt.Tasks {
+			keepTask := true
+			for _, task := range execTasksToRemove {
+				if utility.FromStringPtr(t) == task {
+					keepTask = false
+					break
+				}
+			}
+			if keepTask {
+				tasks = append(tasks, vt.Tasks[j])
+			}
+		}
+		variantTasks[i].Tasks = tasks
+	}
 	apiPatch.VariantsTasks = variantTasks
 	apiPatch.Activated = p.Activated
 	apiPatch.Alias = utility.ToStringPtr(p.Alias)

rest/model/patch_test.go

@@ -224,3 +224,46 @@ func TestDownstreamTasks(t *testing.T) {
 	assert.Len(a.DownstreamTasks[0].Tasks, 2)
 	assert.Len(a.DownstreamTasks[0].VariantTasks, 1)
 }
+
+func TestPreselectedDisplayTasks(t *testing.T) {
+	require.NoError(t, db.ClearCollections(patch.Collection, model.ProjectRefCollection))
+
+	p := patch.Patch{
+		Id:          mgobson.NewObjectId(),
+		Description: "test",
+		Project:     "mci",
+		Tasks:       []string{"variant_task_1", "variant_task_2", "exec1", "exec2"},
+		VariantsTasks: []patch.VariantTasks{
+			{
+				Variant: "coverage",
+				Tasks:   []string{"variant_task_1", "variant_task_2", "exec1", "exec2"},
+				DisplayTasks: []patch.DisplayTask{
+					{
+						Name:      "display_task",
+						ExecTasks: []string{"exec1", "exec2"},
+					},
+				},
+			},
+		},
+	}
+	require.NoError(t, p.Insert())
+
+	a := APIPatch{}
+	err := a.BuildFromService(p, nil)
+	require.NoError(t, err)
+
+	// We expect the tasks from the patch to be only non-execution tasks + display tasks.
+	require.Len(t, a.VariantsTasks, 1)
+	assert.Len(t, a.VariantsTasks[0].Tasks, 3)
+
+	tasks := []string{}
+	for _, task := range a.VariantsTasks[0].Tasks {
+		tasks = append(tasks, utility.FromStringPtr(task))
+	}
+
+	assert.Contains(t, tasks, "variant_task_1")
+	assert.Contains(t, tasks, "variant_task_2")
+	assert.NotContains(t, tasks, "exec1")
+	assert.NotContains(t, tasks, "exec2")
+	assert.Contains(t, tasks, "display_task")
+}