Add Terminated, Running, and Waiting Container States to Test Builder

test/builder/task.go

@@ -400,7 +400,7 @@ func TaskRunPodSecurityContext(context *corev1.PodSecurityContext) TaskRunSpecOp
  }
 }
 
-// StateTerminated set Terminated to the StepState.
+// StateTerminated sets Terminated to the StepState.
 func StateTerminated(exitcode int) StepStateOp {
  return func(s *v1alpha1.StepState) {
  s.ContainerState = corev1.ContainerState{
@@ -409,6 +409,33 @@ func StateTerminated(exitcode int) StepStateOp {
  }
 }
 
+// SetStepStateTerminated sets Terminated state of a step.
+func SetStepStateTerminated(terminated corev1.ContainerStateTerminated) StepStateOp {
+ return func(s *v1alpha1.StepState) {
+ s.ContainerState = corev1.ContainerState{
+ Terminated: &terminated,
+ }
+ }
+}
+
+// SetStepStateRunning sets Running state of a step.
+func SetStepStateRunning(running corev1.ContainerStateRunning) StepStateOp {
+ return func(s *v1alpha1.StepState) {
+ s.ContainerState = corev1.ContainerState{
+ Running: &running,
+ }
+ }
+}
+
+// SetStepStateWaiting sets Waiting state of a step.
+func SetStepStateWaiting(waiting corev1.ContainerStateWaiting) StepStateOp {
+ return func(s *v1alpha1.StepState) {
+ s.ContainerState = corev1.ContainerState{
+ Waiting: &waiting,
+ }
+ }
+}
+
 // TaskRunOwnerReference sets the OwnerReference, with specified kind and name, to the TaskRun.
 func TaskRunOwnerReference(kind, name string, ops ...OwnerReferenceOp) TaskRunOp {
  return func(tr *v1alpha1.TaskRun) {