Skip to content

Commit c43a797

Browse files
EmmaMunleytekton-robot
authored andcommitted
Refactor SequentialTasks & SequentialRuns Tests
This commit refactors the tests: TestDAGExecutionQueueSequentialTasks and TestDAGExecutionQueueSequentialRuns to reduce the scope of the variables to individual test cases so that the test cases no longer influence each other.
1 parent a4c80dc commit c43a797

File tree

1 file changed

+38
-44
lines changed

1 file changed

+38
-44
lines changed

pkg/reconciler/pipelinerun/resources/pipelinerunstate_test.go

+38-44
Original file line numberDiff line numberDiff line change
@@ -1002,28 +1002,6 @@ func TestDAGExecutionQueue(t *testing.T) {
10021002
// TestDAGExecutionQueueSequentialTasks tests the DAGExecutionQueue function for sequential TaskRuns
10031003
// in different states for a running or stopping PipelineRun.
10041004
func TestDAGExecutionQueueSequentialTasks(t *testing.T) {
1005-
firstTask := ResolvedPipelineTask{
1006-
PipelineTask: &v1beta1.PipelineTask{
1007-
Name: "task-1",
1008-
TaskRef: &v1beta1.TaskRef{Name: "task"},
1009-
},
1010-
TaskRunName: "task-1",
1011-
ResolvedTask: &resources.ResolvedTask{
1012-
TaskSpec: &task.Spec,
1013-
},
1014-
}
1015-
secondTask := ResolvedPipelineTask{
1016-
PipelineTask: &v1beta1.PipelineTask{
1017-
Name: "task-2",
1018-
TaskRef: &v1beta1.TaskRef{Name: "task"},
1019-
RunAfter: []string{"task-1"},
1020-
},
1021-
TaskRunName: "task-2",
1022-
ResolvedTask: &resources.ResolvedTask{
1023-
TaskSpec: &task.Spec,
1024-
},
1025-
}
1026-
10271005
tcs := []struct {
10281006
name string
10291007
firstTaskRun *v1beta1.TaskRun
@@ -1059,10 +1037,29 @@ func TestDAGExecutionQueueSequentialTasks(t *testing.T) {
10591037
}}
10601038
for _, tc := range tcs {
10611039
t.Run(tc.name, func(t *testing.T) {
1040+
firstTask := ResolvedPipelineTask{
1041+
PipelineTask: &v1beta1.PipelineTask{
1042+
Name: "task-1",
1043+
TaskRef: &v1beta1.TaskRef{Name: "task"},
1044+
},
1045+
TaskRunName: "task-1",
1046+
ResolvedTask: &resources.ResolvedTask{
1047+
TaskSpec: &task.Spec,
1048+
},
1049+
}
1050+
secondTask := ResolvedPipelineTask{
1051+
PipelineTask: &v1beta1.PipelineTask{
1052+
Name: "task-2",
1053+
TaskRef: &v1beta1.TaskRef{Name: "task"},
1054+
RunAfter: []string{"task-1"},
1055+
},
1056+
TaskRunName: "task-2",
1057+
ResolvedTask: &resources.ResolvedTask{
1058+
TaskSpec: &task.Spec,
1059+
},
1060+
}
10621061
firstTask.TaskRun = tc.firstTaskRun
1063-
defer func() { firstTask.TaskRun = nil }()
10641062
secondTask.TaskRun = tc.secondTaskRun
1065-
defer func() { secondTask.TaskRun = nil }()
10661063
state := PipelineRunState{&firstTask, &secondTask}
10671064
d, err := dagFromState(state)
10681065
if err != nil {
@@ -1098,24 +1095,6 @@ func TestDAGExecutionQueueSequentialTasks(t *testing.T) {
10981095
// TestDAGExecutionQueueSequentialRuns tests the DAGExecutionQueue function for sequential Runs
10991096
// in different states for a running or stopping PipelineRun.
11001097
func TestDAGExecutionQueueSequentialRuns(t *testing.T) {
1101-
firstRun := ResolvedPipelineTask{
1102-
PipelineTask: &v1beta1.PipelineTask{
1103-
Name: "task-1",
1104-
TaskRef: &v1beta1.TaskRef{Name: "task"},
1105-
},
1106-
RunObjectName: "task-1",
1107-
CustomTask: true,
1108-
}
1109-
secondRun := ResolvedPipelineTask{
1110-
PipelineTask: &v1beta1.PipelineTask{
1111-
Name: "task-2",
1112-
TaskRef: &v1beta1.TaskRef{Name: "task"},
1113-
RunAfter: []string{"task-1"},
1114-
},
1115-
RunObjectName: "task-2",
1116-
CustomTask: true,
1117-
}
1118-
11191098
tcs := []struct {
11201099
name string
11211100
firstRun *v1beta1.CustomRun
@@ -1151,14 +1130,29 @@ func TestDAGExecutionQueueSequentialRuns(t *testing.T) {
11511130
}}
11521131
for _, tc := range tcs {
11531132
t.Run(tc.name, func(t *testing.T) {
1133+
firstRun := ResolvedPipelineTask{
1134+
PipelineTask: &v1beta1.PipelineTask{
1135+
Name: "task-1",
1136+
TaskRef: &v1beta1.TaskRef{Name: "task"},
1137+
},
1138+
RunObjectName: "task-1",
1139+
CustomTask: true,
1140+
}
1141+
secondRun := ResolvedPipelineTask{
1142+
PipelineTask: &v1beta1.PipelineTask{
1143+
Name: "task-2",
1144+
TaskRef: &v1beta1.TaskRef{Name: "task"},
1145+
RunAfter: []string{"task-1"},
1146+
},
1147+
RunObjectName: "task-2",
1148+
CustomTask: true,
1149+
}
11541150
if tc.firstRun != nil {
11551151
firstRun.RunObject = tc.firstRun
11561152
}
1157-
defer func() { firstRun.RunObject = nil }()
11581153
if tc.secondRun != nil {
11591154
secondRun.RunObject = tc.secondRun
11601155
}
1161-
defer func() { secondRun.RunObject = nil }()
11621156
state := PipelineRunState{&firstRun, &secondRun}
11631157
d, err := dagFromState(state)
11641158
if err != nil {

0 commit comments

Comments
 (0)