Merge pull request #5244 from hashicorp/b-template-restart-count

Fix double restart counting for templates
This commit is contained in:
Preetha 2019-01-28 10:38:19 -06:00 committed by GitHub
commit 437520a6e6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 6 deletions

View File

@ -441,7 +441,7 @@ func (tm *TaskTemplateManager) handleTemplateRerenders(allRenderedTime time.Time
if restart {
tm.config.Lifecycle.Restart(context.Background(),
structs.NewTaskEvent(structs.TaskRestarting).
structs.NewTaskEvent(structs.TaskRestartSignal).
SetDisplayMessage("Template with change_mode restart re-rendered"), false)
} else if len(signals) != 0 {
var mErr multierror.Error

View File

@ -96,7 +96,7 @@ func (tc *ConsulTemplateTest) TestUpdatesRestartTasks(f *framework.F) {
}
first := allocs[0]
return first.TaskStates["test"].Restarts
}, 5*time.Second, time.Second).Should(Equal(count), "Incorrect restart count")
}, 10*time.Second, time.Second).Should(Equal(count), "Incorrect restart count")
}
// Wrap in retry to wait until placement
@ -132,10 +132,7 @@ func (tc *ConsulTemplateTest) TestUpdatesRestartTasks(f *framework.F) {
require.Nil(err)
// Wrap in retry to wait until restart
// TODO(dani): FIXME: This restart counter should only be 1. This is
// likely an accounting bug in restart tracking from
// template hooks.
waitForRestartCount(2)
waitForRestartCount(1)
}
func (tc *ConsulTemplateTest) AfterEach(f *framework.F) {