@@ -1002,28 +1002,6 @@ func TestDAGExecutionQueue(t *testing.T) {
1002
1002
// TestDAGExecutionQueueSequentialTasks tests the DAGExecutionQueue function for sequential TaskRuns
1003
1003
// in different states for a running or stopping PipelineRun.
1004
1004
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
-
1027
1005
tcs := []struct {
1028
1006
name string
1029
1007
firstTaskRun * v1beta1.TaskRun
@@ -1059,10 +1037,29 @@ func TestDAGExecutionQueueSequentialTasks(t *testing.T) {
1059
1037
}}
1060
1038
for _ , tc := range tcs {
1061
1039
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
+ }
1062
1061
firstTask .TaskRun = tc .firstTaskRun
1063
- defer func () { firstTask .TaskRun = nil }()
1064
1062
secondTask .TaskRun = tc .secondTaskRun
1065
- defer func () { secondTask .TaskRun = nil }()
1066
1063
state := PipelineRunState {& firstTask , & secondTask }
1067
1064
d , err := dagFromState (state )
1068
1065
if err != nil {
@@ -1098,24 +1095,6 @@ func TestDAGExecutionQueueSequentialTasks(t *testing.T) {
1098
1095
// TestDAGExecutionQueueSequentialRuns tests the DAGExecutionQueue function for sequential Runs
1099
1096
// in different states for a running or stopping PipelineRun.
1100
1097
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
-
1119
1098
tcs := []struct {
1120
1099
name string
1121
1100
firstRun * v1beta1.CustomRun
@@ -1151,14 +1130,29 @@ func TestDAGExecutionQueueSequentialRuns(t *testing.T) {
1151
1130
}}
1152
1131
for _ , tc := range tcs {
1153
1132
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
+ }
1154
1150
if tc .firstRun != nil {
1155
1151
firstRun .RunObject = tc .firstRun
1156
1152
}
1157
- defer func () { firstRun .RunObject = nil }()
1158
1153
if tc .secondRun != nil {
1159
1154
secondRun .RunObject = tc .secondRun
1160
1155
}
1161
- defer func () { secondRun .RunObject = nil }()
1162
1156
state := PipelineRunState {& firstRun , & secondRun }
1163
1157
d , err := dagFromState (state )
1164
1158
if err != nil {
0 commit comments