[wip]

Author: tgross

allocrunnersim/allocrunnersim.go

@@ -345,7 +345,7 @@ func (ar *simulatedAllocRunner) stopAll() {
 	// Ensure we have a current copy of the task states, so that we append and
 	// create a correct and full list.
 	ar.allocStateLock.RLock()
-	taskStates := ar.allocState.TaskStates
+	taskStates := ar.allocState.Copy().TaskStates
 	ar.allocStateLock.RUnlock()
 
 	// Perform the task kill, which essentially is the shutdown notification.
@@ -360,8 +360,11 @@ func (ar *simulatedAllocRunner) stopAll() {
 		event.PopulateEventDisplayMessage()
 		taskStates[task.Name].Events = append(taskStates[task.Name].Events, event)
 	}
-	ar.updateAllocAndSendUpdate(taskStates)
+	ar.allocStateLock.Lock()
+	ar.allocState.Copy().TaskStates = taskStates
+	ar.allocStateLock.Unlock()
 
+	ar.updateAllocAndSendUpdate(taskStates)
 	// Mark the tasks as killed and ensure the state is updated to show they
 	// are dead and finished.
 	time.Sleep(200 * time.Millisecond)
@@ -379,6 +382,7 @@ func (ar *simulatedAllocRunner) stopAll() {
 		ar.allocStateLock.Lock()
 		ar.allocState.TaskStates[task.Name].FinishedAt = time.Now()
 		ar.allocState.TaskStates[task.Name].State = structs.TaskStateDead
+
 		ar.allocStateLock.Unlock()
 	}
 	ar.updateAllocAndSendUpdate(taskStates)