This commit is contained in:
Alex Dadgar
2016-02-02 15:18:35 -08:00
parent a2132950a5
commit d04fc2647c

View File

@@ -49,7 +49,7 @@ func testTaskRunner(restarts bool) (*MockTaskStateUpdater, *TaskRunner) {
ctx := driver.NewExecContext(allocDir, alloc.ID)
rp := structs.NewRestartPolicy(structs.JobTypeService)
restartTracker := newRestartTracker(rp)
restartTracker := newRestartTracker(rp, alloc.Job.Type)
if !restarts {
restartTracker = noRestartsTracker()
}