124 lines
3.5 KiB
Go
124 lines
3.5 KiB
Go
package allocrunner
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/hashicorp/nomad/nomad/structs"
|
|
|
|
"github.com/hashicorp/nomad/helper/testlog"
|
|
"github.com/hashicorp/nomad/nomad/mock"
|
|
)
|
|
|
|
func TestTaskHookCoordinator_OnlyMainApp(t *testing.T) {
|
|
alloc := mock.Alloc()
|
|
tasks := alloc.Job.TaskGroups[0].Tasks
|
|
task := tasks[0]
|
|
logger := testlog.HCLogger(t)
|
|
|
|
coord := newTaskHookCoordinator(logger, tasks)
|
|
|
|
ch := coord.startConditionForTask(task)
|
|
|
|
require.Truef(t, isChannelClosed(ch), "%s channel was open, should be closed", task.Name)
|
|
}
|
|
|
|
func TestTaskHookCoordinator_PrestartRunsBeforeMain(t *testing.T) {
|
|
logger := testlog.HCLogger(t)
|
|
|
|
alloc := mock.Alloc()
|
|
tasks := alloc.Job.TaskGroups[0].Tasks
|
|
tasks = append(tasks, initTask())
|
|
tasks = append(tasks, sidecarTask())
|
|
|
|
mainTask := tasks[0]
|
|
initTask := tasks[1]
|
|
sideTask := tasks[2]
|
|
|
|
coord := newTaskHookCoordinator(logger, tasks)
|
|
mainCh := coord.startConditionForTask(mainTask)
|
|
initCh := coord.startConditionForTask(initTask)
|
|
sideCh := coord.startConditionForTask(sideTask)
|
|
|
|
require.Falsef(t, isChannelClosed(mainCh), "%s channel was closed, should be open", mainTask.Name)
|
|
require.Truef(t, isChannelClosed(initCh), "%s channel was open, should be closed", initTask.Name)
|
|
require.Truef(t, isChannelClosed(sideCh), "%s channel was open, should be closed", sideTask.Name)
|
|
}
|
|
|
|
func TestTaskHookCoordinator_MainRunsAfterPrestart(t *testing.T) {
|
|
logger := testlog.HCLogger(t)
|
|
|
|
alloc := mock.Alloc()
|
|
tasks := alloc.Job.TaskGroups[0].Tasks
|
|
tasks = append(tasks, initTask())
|
|
tasks = append(tasks, sidecarTask())
|
|
|
|
coord := newTaskHookCoordinator(logger, tasks)
|
|
mainCh := coord.startConditionForTask(tasks[0])
|
|
initCh := coord.startConditionForTask(tasks[1])
|
|
sideCh := coord.startConditionForTask(tasks[2])
|
|
|
|
mainTaskName := tasks[0].Name
|
|
initTaskName := tasks[1].Name
|
|
sideTaskName := tasks[2].Name
|
|
|
|
require.Falsef(t, isChannelClosed(mainCh), "%s channel was closed, should be open", mainTaskName)
|
|
require.Truef(t, isChannelClosed(initCh), "%s channel was open, should be closed", initTaskName)
|
|
require.Truef(t, isChannelClosed(sideCh), "%s channel was open, should be closed", sideTaskName)
|
|
|
|
states := map[string]*structs.TaskState{
|
|
mainTaskName: &structs.TaskState{
|
|
State: structs.TaskStatePending,
|
|
Failed: false,
|
|
},
|
|
initTaskName: &structs.TaskState{
|
|
State: structs.TaskStateDead,
|
|
Failed: false,
|
|
StartedAt: time.Now(),
|
|
FinishedAt: time.Now(),
|
|
},
|
|
sideTaskName: &structs.TaskState{
|
|
State: structs.TaskStateRunning,
|
|
Failed: false,
|
|
StartedAt: time.Now(),
|
|
},
|
|
}
|
|
|
|
coord.taskStateUpdated(states)
|
|
|
|
require.Truef(t, isChannelClosed(initCh), "%s channel was open, should be closed", mainTaskName)
|
|
require.Truef(t, isChannelClosed(initCh), "%s channel was open, should be closed", initTaskName)
|
|
require.Truef(t, isChannelClosed(sideCh), "%s channel was open, should be closed", sideTaskName)
|
|
}
|
|
|
|
func isChannelClosed(ch <-chan struct{}) bool {
|
|
select {
|
|
case <-ch:
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func sidecarTask() *structs.Task {
|
|
return &structs.Task{
|
|
Name: "sidecar",
|
|
Lifecycle: &structs.TaskLifecycleConfig{
|
|
Hook: structs.TaskLifecycleHookPrestart,
|
|
BlockUntil: structs.TaskLifecycleBlockUntilRunning,
|
|
},
|
|
}
|
|
}
|
|
|
|
func initTask() *structs.Task {
|
|
return &structs.Task{
|
|
Name: "init",
|
|
Lifecycle: &structs.TaskLifecycleConfig{
|
|
Hook: structs.TaskLifecycleHookPrestart,
|
|
BlockUntil: structs.TaskLifecycleBlockUntilCompleted,
|
|
},
|
|
}
|
|
}
|