add unit tests to task package

Signed-off-by: Arash Deshmeh <adeshmeh@ca.ibm.com>
2017-06-29 17:35:26 -04:00
parent 74af31be7f
commit 244e28f41f
9 changed files with 193 additions and 0 deletions
--- a/cli/internal/test/builders/task.go
+++ b/cli/internal/test/builders/task.go
@ -70,6 +70,13 @@ func TaskDesiredState(state swarm.TaskState) func(*swarm.Task) {
 	}
 }

+// TaskSlot sets the task's slot
+func TaskSlot(slot int) func(*swarm.Task) {
+	return func(task *swarm.Task) {
+		task.Slot = slot
+	}
+}
+
 // WithStatus sets the task status
 func WithStatus(statusBuilders ...func(*swarm.TaskStatus)) func(*swarm.Task) {
 	return func(task *swarm.Task) {