From 8d25368462fd86dff992db523710ac902e6ffcf7 Mon Sep 17 00:00:00 2001
From: Alex Dadgar <alex.dadgar@gmail.com>
Date: Sun, 20 Dec 2015 17:30:45 -0800
Subject: [PATCH] Add TaskStates to mock.Alloc. Fixes some test race conditions

---
 nomad/mock/mock.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/nomad/mock/mock.go b/nomad/mock/mock.go
index b3a468d64..3c2bb4f5e 100644
--- a/nomad/mock/mock.go
+++ b/nomad/mock/mock.go
@@ -235,6 +235,11 @@ func Alloc() *structs.Allocation {
 				},
 			},
 		},
+		TaskStates: map[string]*structs.TaskState{
+			"web": &structs.TaskState{
+				State: structs.TaskStatePending,
+			},
+		},
 		Job:           Job(),
 		DesiredStatus: structs.AllocDesiredStatusRun,
 		ClientStatus:  structs.AllocClientStatusPending,