diff --git a/test/builder/task.go b/test/builder/task.go
index 4c552f56b44..bc13695f76d 100644
--- a/test/builder/task.go
+++ b/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) {