e3d78c343c
UID/GID 0 is usually reserved for the root user/group. While Nomad clients are expected to run as root it may not always be the case. Setting these values as -1 if not defined will fallback to the pervious behaviour of not attempting to set file ownership and use whatever UID/GID the Nomad agent is running as. It will also keep backwards compatibility, which is specially important for platforms where this feature is not supported, like Windows.
2456 lines
67 KiB
Go
2456 lines
67 KiB
Go
package api
|
|
|
|
import (
|
|
"reflect"
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/nomad/api/internal/testutil"
|
|
"github.com/kr/pretty"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestJobs_Register(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Listing jobs before registering returns nothing
|
|
resp, _, err := jobs.List(nil)
|
|
require.Nil(err)
|
|
require.Emptyf(resp, "expected 0 jobs, got: %d", len(resp))
|
|
|
|
// Create a job and attempt to register it
|
|
job := testJob()
|
|
resp2, wm, err := jobs.Register(job, nil)
|
|
require.Nil(err)
|
|
require.NotNil(resp2)
|
|
require.NotEmpty(resp2.EvalID)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the jobs back out again
|
|
resp, qm, err := jobs.List(nil)
|
|
assertQueryMeta(t, qm)
|
|
require.Nil(err)
|
|
|
|
// Check that we got the expected response
|
|
if len(resp) != 1 || resp[0].ID != *job.ID {
|
|
t.Fatalf("bad: %#v", resp[0])
|
|
}
|
|
}
|
|
|
|
func TestJobs_Register_PreserveCounts(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Listing jobs before registering returns nothing
|
|
resp, _, err := jobs.List(nil)
|
|
require.Nil(err)
|
|
require.Emptyf(resp, "expected 0 jobs, got: %d", len(resp))
|
|
|
|
// Create a job
|
|
task := NewTask("task", "exec").
|
|
SetConfig("command", "/bin/sleep").
|
|
Require(&Resources{
|
|
CPU: intToPtr(100),
|
|
MemoryMB: intToPtr(256),
|
|
}).
|
|
SetLogConfig(&LogConfig{
|
|
MaxFiles: intToPtr(1),
|
|
MaxFileSizeMB: intToPtr(2),
|
|
})
|
|
|
|
group1 := NewTaskGroup("group1", 1).
|
|
AddTask(task).
|
|
RequireDisk(&EphemeralDisk{
|
|
SizeMB: intToPtr(25),
|
|
})
|
|
group2 := NewTaskGroup("group2", 2).
|
|
AddTask(task).
|
|
RequireDisk(&EphemeralDisk{
|
|
SizeMB: intToPtr(25),
|
|
})
|
|
|
|
job := NewBatchJob("job", "redis", "global", 1).
|
|
AddDatacenter("dc1").
|
|
AddTaskGroup(group1).
|
|
AddTaskGroup(group2)
|
|
|
|
// Create a job and register it
|
|
resp2, wm, err := jobs.Register(job, nil)
|
|
require.Nil(err)
|
|
require.NotNil(resp2)
|
|
require.NotEmpty(resp2.EvalID)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Update the job, new groups to test PreserveCounts
|
|
group1.Count = nil
|
|
group2.Count = intToPtr(0)
|
|
group3 := NewTaskGroup("group3", 3).
|
|
AddTask(task).
|
|
RequireDisk(&EphemeralDisk{
|
|
SizeMB: intToPtr(25),
|
|
})
|
|
job.AddTaskGroup(group3)
|
|
|
|
// Update the job, with PreserveCounts = true
|
|
_, _, err = jobs.RegisterOpts(job, &RegisterOptions{
|
|
PreserveCounts: true,
|
|
}, nil)
|
|
require.NoError(err)
|
|
|
|
// Query the job scale status
|
|
status, _, err := jobs.ScaleStatus(*job.ID, nil)
|
|
require.NoError(err)
|
|
require.Equal(1, status.TaskGroups["group1"].Desired) // present and nil => preserved
|
|
require.Equal(2, status.TaskGroups["group2"].Desired) // present and specified => preserved
|
|
require.Equal(3, status.TaskGroups["group3"].Desired) // new => as specific in job spec
|
|
}
|
|
|
|
func TestJobs_Register_NoPreserveCounts(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Listing jobs before registering returns nothing
|
|
resp, _, err := jobs.List(nil)
|
|
require.Nil(err)
|
|
require.Emptyf(resp, "expected 0 jobs, got: %d", len(resp))
|
|
|
|
// Create a job
|
|
task := NewTask("task", "exec").
|
|
SetConfig("command", "/bin/sleep").
|
|
Require(&Resources{
|
|
CPU: intToPtr(100),
|
|
MemoryMB: intToPtr(256),
|
|
}).
|
|
SetLogConfig(&LogConfig{
|
|
MaxFiles: intToPtr(1),
|
|
MaxFileSizeMB: intToPtr(2),
|
|
})
|
|
|
|
group1 := NewTaskGroup("group1", 1).
|
|
AddTask(task).
|
|
RequireDisk(&EphemeralDisk{
|
|
SizeMB: intToPtr(25),
|
|
})
|
|
group2 := NewTaskGroup("group2", 2).
|
|
AddTask(task).
|
|
RequireDisk(&EphemeralDisk{
|
|
SizeMB: intToPtr(25),
|
|
})
|
|
|
|
job := NewBatchJob("job", "redis", "global", 1).
|
|
AddDatacenter("dc1").
|
|
AddTaskGroup(group1).
|
|
AddTaskGroup(group2)
|
|
|
|
// Create a job and register it
|
|
resp2, wm, err := jobs.Register(job, nil)
|
|
require.Nil(err)
|
|
require.NotNil(resp2)
|
|
require.NotEmpty(resp2.EvalID)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Update the job, new groups to test PreserveCounts
|
|
group1.Count = intToPtr(0)
|
|
group2.Count = nil
|
|
group3 := NewTaskGroup("group3", 3).
|
|
AddTask(task).
|
|
RequireDisk(&EphemeralDisk{
|
|
SizeMB: intToPtr(25),
|
|
})
|
|
job.AddTaskGroup(group3)
|
|
|
|
// Update the job, with PreserveCounts = default [false]
|
|
_, _, err = jobs.Register(job, nil)
|
|
require.NoError(err)
|
|
|
|
// Query the job scale status
|
|
status, _, err := jobs.ScaleStatus(*job.ID, nil)
|
|
require.NoError(err)
|
|
require.Equal("default", status.Namespace)
|
|
require.Equal(0, status.TaskGroups["group1"].Desired) // present => as specified
|
|
require.Equal(1, status.TaskGroups["group2"].Desired) // nil => default (1)
|
|
require.Equal(3, status.TaskGroups["group3"].Desired) // new => as specified
|
|
}
|
|
|
|
func TestJobs_Register_EvalPriority(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
requireAssert := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
|
|
// Listing jobs before registering returns nothing
|
|
listResp, _, err := c.Jobs().List(nil)
|
|
requireAssert.Nil(err)
|
|
requireAssert.Len(listResp, 0)
|
|
|
|
// Create a job and register it with an eval priority.
|
|
job := testJob()
|
|
registerResp, wm, err := c.Jobs().RegisterOpts(job, &RegisterOptions{EvalPriority: 99}, nil)
|
|
requireAssert.Nil(err)
|
|
requireAssert.NotNil(registerResp)
|
|
requireAssert.NotEmpty(registerResp.EvalID)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Check the created job evaluation has a priority that matches our desired
|
|
// value.
|
|
evalInfo, _, err := c.Evaluations().Info(registerResp.EvalID, nil)
|
|
requireAssert.NoError(err)
|
|
requireAssert.Equal(99, evalInfo.Priority)
|
|
}
|
|
|
|
func TestJobs_Register_NoEvalPriority(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
requireAssert := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
|
|
// Listing jobs before registering returns nothing
|
|
listResp, _, err := c.Jobs().List(nil)
|
|
requireAssert.Nil(err)
|
|
requireAssert.Len(listResp, 0)
|
|
|
|
// Create a job and register it with an eval priority.
|
|
job := testJob()
|
|
registerResp, wm, err := c.Jobs().RegisterOpts(job, nil, nil)
|
|
requireAssert.Nil(err)
|
|
requireAssert.NotNil(registerResp)
|
|
requireAssert.NotEmpty(registerResp.EvalID)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Check the created job evaluation has a priority that matches the job
|
|
// priority.
|
|
evalInfo, _, err := c.Evaluations().Info(registerResp.EvalID, nil)
|
|
requireAssert.NoError(err)
|
|
requireAssert.Equal(*job.Priority, evalInfo.Priority)
|
|
}
|
|
|
|
func TestJobs_Validate(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Create a job and attempt to register it
|
|
job := testJob()
|
|
resp, _, err := jobs.Validate(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
if len(resp.ValidationErrors) != 0 {
|
|
t.Fatalf("bad %v", resp)
|
|
}
|
|
|
|
job.ID = nil
|
|
resp1, _, err := jobs.Validate(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
if len(resp1.ValidationErrors) == 0 {
|
|
t.Fatalf("bad %v", resp1)
|
|
}
|
|
}
|
|
|
|
func TestJobs_Canonicalize(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
testCases := []struct {
|
|
name string
|
|
expected *Job
|
|
input *Job
|
|
}{
|
|
{
|
|
name: "empty",
|
|
input: &Job{
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Tasks: []*Task{
|
|
{},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: &Job{
|
|
ID: stringToPtr(""),
|
|
Name: stringToPtr(""),
|
|
Region: stringToPtr("global"),
|
|
Namespace: stringToPtr(DefaultNamespace),
|
|
Type: stringToPtr("service"),
|
|
ParentID: stringToPtr(""),
|
|
Priority: intToPtr(50),
|
|
AllAtOnce: boolToPtr(false),
|
|
ConsulToken: stringToPtr(""),
|
|
ConsulNamespace: stringToPtr(""),
|
|
VaultToken: stringToPtr(""),
|
|
VaultNamespace: stringToPtr(""),
|
|
NomadTokenID: stringToPtr(""),
|
|
Status: stringToPtr(""),
|
|
StatusDescription: stringToPtr(""),
|
|
Stop: boolToPtr(false),
|
|
Stable: boolToPtr(false),
|
|
Version: uint64ToPtr(0),
|
|
CreateIndex: uint64ToPtr(0),
|
|
ModifyIndex: uint64ToPtr(0),
|
|
JobModifyIndex: uint64ToPtr(0),
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Name: stringToPtr(""),
|
|
Count: intToPtr(1),
|
|
EphemeralDisk: &EphemeralDisk{
|
|
Sticky: boolToPtr(false),
|
|
Migrate: boolToPtr(false),
|
|
SizeMB: intToPtr(300),
|
|
},
|
|
RestartPolicy: &RestartPolicy{
|
|
Delay: timeToPtr(15 * time.Second),
|
|
Attempts: intToPtr(2),
|
|
Interval: timeToPtr(30 * time.Minute),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
ReschedulePolicy: &ReschedulePolicy{
|
|
Attempts: intToPtr(0),
|
|
Interval: timeToPtr(0),
|
|
DelayFunction: stringToPtr("exponential"),
|
|
Delay: timeToPtr(30 * time.Second),
|
|
MaxDelay: timeToPtr(1 * time.Hour),
|
|
Unlimited: boolToPtr(true),
|
|
},
|
|
Consul: &Consul{
|
|
Namespace: "",
|
|
},
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
Migrate: DefaultMigrateStrategy(),
|
|
Tasks: []*Task{
|
|
{
|
|
KillTimeout: timeToPtr(5 * time.Second),
|
|
LogConfig: DefaultLogConfig(),
|
|
Resources: DefaultResources(),
|
|
RestartPolicy: defaultServiceJobRestartPolicy(),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "batch",
|
|
input: &Job{
|
|
Type: stringToPtr("batch"),
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Tasks: []*Task{
|
|
{},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: &Job{
|
|
ID: stringToPtr(""),
|
|
Name: stringToPtr(""),
|
|
Region: stringToPtr("global"),
|
|
Namespace: stringToPtr(DefaultNamespace),
|
|
Type: stringToPtr("batch"),
|
|
ParentID: stringToPtr(""),
|
|
Priority: intToPtr(50),
|
|
AllAtOnce: boolToPtr(false),
|
|
ConsulToken: stringToPtr(""),
|
|
ConsulNamespace: stringToPtr(""),
|
|
VaultToken: stringToPtr(""),
|
|
VaultNamespace: stringToPtr(""),
|
|
NomadTokenID: stringToPtr(""),
|
|
Status: stringToPtr(""),
|
|
StatusDescription: stringToPtr(""),
|
|
Stop: boolToPtr(false),
|
|
Stable: boolToPtr(false),
|
|
Version: uint64ToPtr(0),
|
|
CreateIndex: uint64ToPtr(0),
|
|
ModifyIndex: uint64ToPtr(0),
|
|
JobModifyIndex: uint64ToPtr(0),
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Name: stringToPtr(""),
|
|
Count: intToPtr(1),
|
|
EphemeralDisk: &EphemeralDisk{
|
|
Sticky: boolToPtr(false),
|
|
Migrate: boolToPtr(false),
|
|
SizeMB: intToPtr(300),
|
|
},
|
|
RestartPolicy: &RestartPolicy{
|
|
Delay: timeToPtr(15 * time.Second),
|
|
Attempts: intToPtr(3),
|
|
Interval: timeToPtr(24 * time.Hour),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
ReschedulePolicy: &ReschedulePolicy{
|
|
Attempts: intToPtr(1),
|
|
Interval: timeToPtr(24 * time.Hour),
|
|
DelayFunction: stringToPtr("constant"),
|
|
Delay: timeToPtr(5 * time.Second),
|
|
MaxDelay: timeToPtr(0),
|
|
Unlimited: boolToPtr(false),
|
|
},
|
|
Consul: &Consul{
|
|
Namespace: "",
|
|
},
|
|
Tasks: []*Task{
|
|
{
|
|
KillTimeout: timeToPtr(5 * time.Second),
|
|
LogConfig: DefaultLogConfig(),
|
|
Resources: DefaultResources(),
|
|
RestartPolicy: defaultBatchJobRestartPolicy(),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "partial",
|
|
input: &Job{
|
|
Name: stringToPtr("foo"),
|
|
Namespace: stringToPtr("bar"),
|
|
ID: stringToPtr("bar"),
|
|
ParentID: stringToPtr("lol"),
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Name: stringToPtr("bar"),
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "task1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: &Job{
|
|
Namespace: stringToPtr("bar"),
|
|
ID: stringToPtr("bar"),
|
|
Name: stringToPtr("foo"),
|
|
Region: stringToPtr("global"),
|
|
Type: stringToPtr("service"),
|
|
ParentID: stringToPtr("lol"),
|
|
Priority: intToPtr(50),
|
|
AllAtOnce: boolToPtr(false),
|
|
ConsulToken: stringToPtr(""),
|
|
ConsulNamespace: stringToPtr(""),
|
|
VaultToken: stringToPtr(""),
|
|
VaultNamespace: stringToPtr(""),
|
|
NomadTokenID: stringToPtr(""),
|
|
Stop: boolToPtr(false),
|
|
Stable: boolToPtr(false),
|
|
Version: uint64ToPtr(0),
|
|
Status: stringToPtr(""),
|
|
StatusDescription: stringToPtr(""),
|
|
CreateIndex: uint64ToPtr(0),
|
|
ModifyIndex: uint64ToPtr(0),
|
|
JobModifyIndex: uint64ToPtr(0),
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Name: stringToPtr("bar"),
|
|
Count: intToPtr(1),
|
|
EphemeralDisk: &EphemeralDisk{
|
|
Sticky: boolToPtr(false),
|
|
Migrate: boolToPtr(false),
|
|
SizeMB: intToPtr(300),
|
|
},
|
|
RestartPolicy: &RestartPolicy{
|
|
Delay: timeToPtr(15 * time.Second),
|
|
Attempts: intToPtr(2),
|
|
Interval: timeToPtr(30 * time.Minute),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
ReschedulePolicy: &ReschedulePolicy{
|
|
Attempts: intToPtr(0),
|
|
Interval: timeToPtr(0),
|
|
DelayFunction: stringToPtr("exponential"),
|
|
Delay: timeToPtr(30 * time.Second),
|
|
MaxDelay: timeToPtr(1 * time.Hour),
|
|
Unlimited: boolToPtr(true),
|
|
},
|
|
Consul: &Consul{
|
|
Namespace: "",
|
|
},
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
Migrate: DefaultMigrateStrategy(),
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "task1",
|
|
LogConfig: DefaultLogConfig(),
|
|
Resources: DefaultResources(),
|
|
KillTimeout: timeToPtr(5 * time.Second),
|
|
RestartPolicy: defaultServiceJobRestartPolicy(),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "example_template",
|
|
input: &Job{
|
|
ID: stringToPtr("example_template"),
|
|
Name: stringToPtr("example_template"),
|
|
Datacenters: []string{"dc1"},
|
|
Type: stringToPtr("service"),
|
|
Update: &UpdateStrategy{
|
|
MaxParallel: intToPtr(1),
|
|
AutoPromote: boolToPtr(true),
|
|
},
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Name: stringToPtr("cache"),
|
|
Count: intToPtr(1),
|
|
RestartPolicy: &RestartPolicy{
|
|
Interval: timeToPtr(5 * time.Minute),
|
|
Attempts: intToPtr(10),
|
|
Delay: timeToPtr(25 * time.Second),
|
|
Mode: stringToPtr("delay"),
|
|
},
|
|
Update: &UpdateStrategy{
|
|
AutoRevert: boolToPtr(true),
|
|
},
|
|
EphemeralDisk: &EphemeralDisk{
|
|
SizeMB: intToPtr(300),
|
|
},
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "redis",
|
|
Driver: "docker",
|
|
Config: map[string]interface{}{
|
|
"image": "redis:7",
|
|
"port_map": []map[string]int{{
|
|
"db": 6379,
|
|
}},
|
|
},
|
|
RestartPolicy: &RestartPolicy{
|
|
// inherit other values from TG
|
|
Attempts: intToPtr(20),
|
|
},
|
|
Resources: &Resources{
|
|
CPU: intToPtr(500),
|
|
MemoryMB: intToPtr(256),
|
|
Networks: []*NetworkResource{
|
|
{
|
|
MBits: intToPtr(10),
|
|
DynamicPorts: []Port{
|
|
{
|
|
Label: "db",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
Services: []*Service{
|
|
{
|
|
Name: "redis-cache",
|
|
Tags: []string{"global", "cache"},
|
|
CanaryTags: []string{"canary", "global", "cache"},
|
|
PortLabel: "db",
|
|
Checks: []ServiceCheck{
|
|
{
|
|
Name: "alive",
|
|
Type: "tcp",
|
|
Interval: 10 * time.Second,
|
|
Timeout: 2 * time.Second,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
Templates: []*Template{
|
|
{
|
|
EmbeddedTmpl: stringToPtr("---"),
|
|
DestPath: stringToPtr("local/file.yml"),
|
|
},
|
|
{
|
|
EmbeddedTmpl: stringToPtr("FOO=bar\n"),
|
|
DestPath: stringToPtr("local/file.env"),
|
|
Envvars: boolToPtr(true),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: &Job{
|
|
Namespace: stringToPtr(DefaultNamespace),
|
|
ID: stringToPtr("example_template"),
|
|
Name: stringToPtr("example_template"),
|
|
ParentID: stringToPtr(""),
|
|
Priority: intToPtr(50),
|
|
Region: stringToPtr("global"),
|
|
Type: stringToPtr("service"),
|
|
AllAtOnce: boolToPtr(false),
|
|
ConsulToken: stringToPtr(""),
|
|
ConsulNamespace: stringToPtr(""),
|
|
VaultToken: stringToPtr(""),
|
|
VaultNamespace: stringToPtr(""),
|
|
NomadTokenID: stringToPtr(""),
|
|
Stop: boolToPtr(false),
|
|
Stable: boolToPtr(false),
|
|
Version: uint64ToPtr(0),
|
|
Status: stringToPtr(""),
|
|
StatusDescription: stringToPtr(""),
|
|
CreateIndex: uint64ToPtr(0),
|
|
ModifyIndex: uint64ToPtr(0),
|
|
JobModifyIndex: uint64ToPtr(0),
|
|
Datacenters: []string{"dc1"},
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(true),
|
|
},
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Name: stringToPtr("cache"),
|
|
Count: intToPtr(1),
|
|
RestartPolicy: &RestartPolicy{
|
|
Interval: timeToPtr(5 * time.Minute),
|
|
Attempts: intToPtr(10),
|
|
Delay: timeToPtr(25 * time.Second),
|
|
Mode: stringToPtr("delay"),
|
|
},
|
|
ReschedulePolicy: &ReschedulePolicy{
|
|
Attempts: intToPtr(0),
|
|
Interval: timeToPtr(0),
|
|
DelayFunction: stringToPtr("exponential"),
|
|
Delay: timeToPtr(30 * time.Second),
|
|
MaxDelay: timeToPtr(1 * time.Hour),
|
|
Unlimited: boolToPtr(true),
|
|
},
|
|
EphemeralDisk: &EphemeralDisk{
|
|
Sticky: boolToPtr(false),
|
|
Migrate: boolToPtr(false),
|
|
SizeMB: intToPtr(300),
|
|
},
|
|
Consul: &Consul{
|
|
Namespace: "",
|
|
},
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(true),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(true),
|
|
},
|
|
Migrate: DefaultMigrateStrategy(),
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "redis",
|
|
Driver: "docker",
|
|
Config: map[string]interface{}{
|
|
"image": "redis:7",
|
|
"port_map": []map[string]int{{
|
|
"db": 6379,
|
|
}},
|
|
},
|
|
RestartPolicy: &RestartPolicy{
|
|
Interval: timeToPtr(5 * time.Minute),
|
|
Attempts: intToPtr(20),
|
|
Delay: timeToPtr(25 * time.Second),
|
|
Mode: stringToPtr("delay"),
|
|
},
|
|
Resources: &Resources{
|
|
CPU: intToPtr(500),
|
|
Cores: intToPtr(0),
|
|
MemoryMB: intToPtr(256),
|
|
Networks: []*NetworkResource{
|
|
{
|
|
MBits: intToPtr(10),
|
|
DynamicPorts: []Port{
|
|
{
|
|
Label: "db",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
Services: []*Service{
|
|
{
|
|
Name: "redis-cache",
|
|
Tags: []string{"global", "cache"},
|
|
CanaryTags: []string{"canary", "global", "cache"},
|
|
PortLabel: "db",
|
|
AddressMode: "auto",
|
|
OnUpdate: "require_healthy",
|
|
Provider: "consul",
|
|
Checks: []ServiceCheck{
|
|
{
|
|
Name: "alive",
|
|
Type: "tcp",
|
|
Interval: 10 * time.Second,
|
|
Timeout: 2 * time.Second,
|
|
OnUpdate: "require_healthy",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
KillTimeout: timeToPtr(5 * time.Second),
|
|
LogConfig: DefaultLogConfig(),
|
|
Templates: []*Template{
|
|
{
|
|
SourcePath: stringToPtr(""),
|
|
DestPath: stringToPtr("local/file.yml"),
|
|
EmbeddedTmpl: stringToPtr("---"),
|
|
ChangeMode: stringToPtr("restart"),
|
|
ChangeSignal: stringToPtr(""),
|
|
Splay: timeToPtr(5 * time.Second),
|
|
Perms: stringToPtr("0644"),
|
|
Uid: intToPtr(-1),
|
|
Gid: intToPtr(-1),
|
|
LeftDelim: stringToPtr("{{"),
|
|
RightDelim: stringToPtr("}}"),
|
|
Envvars: boolToPtr(false),
|
|
VaultGrace: timeToPtr(0),
|
|
},
|
|
{
|
|
SourcePath: stringToPtr(""),
|
|
DestPath: stringToPtr("local/file.env"),
|
|
EmbeddedTmpl: stringToPtr("FOO=bar\n"),
|
|
ChangeMode: stringToPtr("restart"),
|
|
ChangeSignal: stringToPtr(""),
|
|
Splay: timeToPtr(5 * time.Second),
|
|
Perms: stringToPtr("0644"),
|
|
Uid: intToPtr(-1),
|
|
Gid: intToPtr(-1),
|
|
LeftDelim: stringToPtr("{{"),
|
|
RightDelim: stringToPtr("}}"),
|
|
Envvars: boolToPtr(true),
|
|
VaultGrace: timeToPtr(0),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "periodic",
|
|
input: &Job{
|
|
ID: stringToPtr("bar"),
|
|
Periodic: &PeriodicConfig{},
|
|
},
|
|
expected: &Job{
|
|
Namespace: stringToPtr(DefaultNamespace),
|
|
ID: stringToPtr("bar"),
|
|
ParentID: stringToPtr(""),
|
|
Name: stringToPtr("bar"),
|
|
Region: stringToPtr("global"),
|
|
Type: stringToPtr("service"),
|
|
Priority: intToPtr(50),
|
|
AllAtOnce: boolToPtr(false),
|
|
ConsulToken: stringToPtr(""),
|
|
ConsulNamespace: stringToPtr(""),
|
|
VaultToken: stringToPtr(""),
|
|
VaultNamespace: stringToPtr(""),
|
|
NomadTokenID: stringToPtr(""),
|
|
Stop: boolToPtr(false),
|
|
Stable: boolToPtr(false),
|
|
Version: uint64ToPtr(0),
|
|
Status: stringToPtr(""),
|
|
StatusDescription: stringToPtr(""),
|
|
CreateIndex: uint64ToPtr(0),
|
|
ModifyIndex: uint64ToPtr(0),
|
|
JobModifyIndex: uint64ToPtr(0),
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
Periodic: &PeriodicConfig{
|
|
Enabled: boolToPtr(true),
|
|
Spec: stringToPtr(""),
|
|
SpecType: stringToPtr(PeriodicSpecCron),
|
|
ProhibitOverlap: boolToPtr(false),
|
|
TimeZone: stringToPtr("UTC"),
|
|
},
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "update_merge",
|
|
input: &Job{
|
|
Name: stringToPtr("foo"),
|
|
ID: stringToPtr("bar"),
|
|
ParentID: stringToPtr("lol"),
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(1 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(6 * time.Minute),
|
|
ProgressDeadline: timeToPtr(7 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Name: stringToPtr("bar"),
|
|
Consul: &Consul{
|
|
Namespace: "",
|
|
},
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(2 * time.Second),
|
|
MaxParallel: intToPtr(2),
|
|
HealthCheck: stringToPtr("manual"),
|
|
MinHealthyTime: timeToPtr(1 * time.Second),
|
|
AutoRevert: boolToPtr(true),
|
|
Canary: intToPtr(1),
|
|
AutoPromote: boolToPtr(true),
|
|
},
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "task1",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: stringToPtr("baz"),
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "task1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: &Job{
|
|
Namespace: stringToPtr(DefaultNamespace),
|
|
ID: stringToPtr("bar"),
|
|
Name: stringToPtr("foo"),
|
|
Region: stringToPtr("global"),
|
|
Type: stringToPtr("service"),
|
|
ParentID: stringToPtr("lol"),
|
|
Priority: intToPtr(50),
|
|
AllAtOnce: boolToPtr(false),
|
|
ConsulToken: stringToPtr(""),
|
|
ConsulNamespace: stringToPtr(""),
|
|
VaultToken: stringToPtr(""),
|
|
VaultNamespace: stringToPtr(""),
|
|
NomadTokenID: stringToPtr(""),
|
|
Stop: boolToPtr(false),
|
|
Stable: boolToPtr(false),
|
|
Version: uint64ToPtr(0),
|
|
Status: stringToPtr(""),
|
|
StatusDescription: stringToPtr(""),
|
|
CreateIndex: uint64ToPtr(0),
|
|
ModifyIndex: uint64ToPtr(0),
|
|
JobModifyIndex: uint64ToPtr(0),
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(1 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(6 * time.Minute),
|
|
ProgressDeadline: timeToPtr(7 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Name: stringToPtr("bar"),
|
|
Count: intToPtr(1),
|
|
EphemeralDisk: &EphemeralDisk{
|
|
Sticky: boolToPtr(false),
|
|
Migrate: boolToPtr(false),
|
|
SizeMB: intToPtr(300),
|
|
},
|
|
RestartPolicy: &RestartPolicy{
|
|
Delay: timeToPtr(15 * time.Second),
|
|
Attempts: intToPtr(2),
|
|
Interval: timeToPtr(30 * time.Minute),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
ReschedulePolicy: &ReschedulePolicy{
|
|
Attempts: intToPtr(0),
|
|
Interval: timeToPtr(0),
|
|
DelayFunction: stringToPtr("exponential"),
|
|
Delay: timeToPtr(30 * time.Second),
|
|
MaxDelay: timeToPtr(1 * time.Hour),
|
|
Unlimited: boolToPtr(true),
|
|
},
|
|
Consul: &Consul{
|
|
Namespace: "",
|
|
},
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(2 * time.Second),
|
|
MaxParallel: intToPtr(2),
|
|
HealthCheck: stringToPtr("manual"),
|
|
MinHealthyTime: timeToPtr(1 * time.Second),
|
|
HealthyDeadline: timeToPtr(6 * time.Minute),
|
|
ProgressDeadline: timeToPtr(7 * time.Minute),
|
|
AutoRevert: boolToPtr(true),
|
|
Canary: intToPtr(1),
|
|
AutoPromote: boolToPtr(true),
|
|
},
|
|
Migrate: DefaultMigrateStrategy(),
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "task1",
|
|
LogConfig: DefaultLogConfig(),
|
|
Resources: DefaultResources(),
|
|
KillTimeout: timeToPtr(5 * time.Second),
|
|
RestartPolicy: defaultServiceJobRestartPolicy(),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: stringToPtr("baz"),
|
|
Count: intToPtr(1),
|
|
EphemeralDisk: &EphemeralDisk{
|
|
Sticky: boolToPtr(false),
|
|
Migrate: boolToPtr(false),
|
|
SizeMB: intToPtr(300),
|
|
},
|
|
RestartPolicy: &RestartPolicy{
|
|
Delay: timeToPtr(15 * time.Second),
|
|
Attempts: intToPtr(2),
|
|
Interval: timeToPtr(30 * time.Minute),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
ReschedulePolicy: &ReschedulePolicy{
|
|
Attempts: intToPtr(0),
|
|
Interval: timeToPtr(0),
|
|
DelayFunction: stringToPtr("exponential"),
|
|
Delay: timeToPtr(30 * time.Second),
|
|
MaxDelay: timeToPtr(1 * time.Hour),
|
|
Unlimited: boolToPtr(true),
|
|
},
|
|
Consul: &Consul{
|
|
Namespace: "",
|
|
},
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(1 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(6 * time.Minute),
|
|
ProgressDeadline: timeToPtr(7 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
Migrate: DefaultMigrateStrategy(),
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "task1",
|
|
LogConfig: DefaultLogConfig(),
|
|
Resources: DefaultResources(),
|
|
KillTimeout: timeToPtr(5 * time.Second),
|
|
RestartPolicy: defaultServiceJobRestartPolicy(),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "restart_merge",
|
|
input: &Job{
|
|
Name: stringToPtr("foo"),
|
|
ID: stringToPtr("bar"),
|
|
ParentID: stringToPtr("lol"),
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Name: stringToPtr("bar"),
|
|
RestartPolicy: &RestartPolicy{
|
|
Delay: timeToPtr(15 * time.Second),
|
|
Attempts: intToPtr(2),
|
|
Interval: timeToPtr(30 * time.Minute),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "task1",
|
|
RestartPolicy: &RestartPolicy{
|
|
Attempts: intToPtr(5),
|
|
Delay: timeToPtr(1 * time.Second),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: stringToPtr("baz"),
|
|
RestartPolicy: &RestartPolicy{
|
|
Delay: timeToPtr(20 * time.Second),
|
|
Attempts: intToPtr(2),
|
|
Interval: timeToPtr(30 * time.Minute),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
Consul: &Consul{
|
|
Namespace: "",
|
|
},
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "task1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: &Job{
|
|
Namespace: stringToPtr(DefaultNamespace),
|
|
ID: stringToPtr("bar"),
|
|
Name: stringToPtr("foo"),
|
|
Region: stringToPtr("global"),
|
|
Type: stringToPtr("service"),
|
|
ParentID: stringToPtr("lol"),
|
|
Priority: intToPtr(50),
|
|
AllAtOnce: boolToPtr(false),
|
|
ConsulToken: stringToPtr(""),
|
|
ConsulNamespace: stringToPtr(""),
|
|
VaultToken: stringToPtr(""),
|
|
VaultNamespace: stringToPtr(""),
|
|
NomadTokenID: stringToPtr(""),
|
|
Stop: boolToPtr(false),
|
|
Stable: boolToPtr(false),
|
|
Version: uint64ToPtr(0),
|
|
Status: stringToPtr(""),
|
|
StatusDescription: stringToPtr(""),
|
|
CreateIndex: uint64ToPtr(0),
|
|
ModifyIndex: uint64ToPtr(0),
|
|
JobModifyIndex: uint64ToPtr(0),
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
TaskGroups: []*TaskGroup{
|
|
{
|
|
Name: stringToPtr("bar"),
|
|
Count: intToPtr(1),
|
|
EphemeralDisk: &EphemeralDisk{
|
|
Sticky: boolToPtr(false),
|
|
Migrate: boolToPtr(false),
|
|
SizeMB: intToPtr(300),
|
|
},
|
|
RestartPolicy: &RestartPolicy{
|
|
Delay: timeToPtr(15 * time.Second),
|
|
Attempts: intToPtr(2),
|
|
Interval: timeToPtr(30 * time.Minute),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
ReschedulePolicy: &ReschedulePolicy{
|
|
Attempts: intToPtr(0),
|
|
Interval: timeToPtr(0),
|
|
DelayFunction: stringToPtr("exponential"),
|
|
Delay: timeToPtr(30 * time.Second),
|
|
MaxDelay: timeToPtr(1 * time.Hour),
|
|
Unlimited: boolToPtr(true),
|
|
},
|
|
Consul: &Consul{
|
|
Namespace: "",
|
|
},
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
Migrate: DefaultMigrateStrategy(),
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "task1",
|
|
LogConfig: DefaultLogConfig(),
|
|
Resources: DefaultResources(),
|
|
KillTimeout: timeToPtr(5 * time.Second),
|
|
RestartPolicy: &RestartPolicy{
|
|
Attempts: intToPtr(5),
|
|
Delay: timeToPtr(1 * time.Second),
|
|
Interval: timeToPtr(30 * time.Minute),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: stringToPtr("baz"),
|
|
Count: intToPtr(1),
|
|
EphemeralDisk: &EphemeralDisk{
|
|
Sticky: boolToPtr(false),
|
|
Migrate: boolToPtr(false),
|
|
SizeMB: intToPtr(300),
|
|
},
|
|
RestartPolicy: &RestartPolicy{
|
|
Delay: timeToPtr(20 * time.Second),
|
|
Attempts: intToPtr(2),
|
|
Interval: timeToPtr(30 * time.Minute),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
ReschedulePolicy: &ReschedulePolicy{
|
|
Attempts: intToPtr(0),
|
|
Interval: timeToPtr(0),
|
|
DelayFunction: stringToPtr("exponential"),
|
|
Delay: timeToPtr(30 * time.Second),
|
|
MaxDelay: timeToPtr(1 * time.Hour),
|
|
Unlimited: boolToPtr(true),
|
|
},
|
|
Consul: &Consul{
|
|
Namespace: "",
|
|
},
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
Migrate: DefaultMigrateStrategy(),
|
|
Tasks: []*Task{
|
|
{
|
|
Name: "task1",
|
|
LogConfig: DefaultLogConfig(),
|
|
Resources: DefaultResources(),
|
|
KillTimeout: timeToPtr(5 * time.Second),
|
|
RestartPolicy: &RestartPolicy{
|
|
Delay: timeToPtr(20 * time.Second),
|
|
Attempts: intToPtr(2),
|
|
Interval: timeToPtr(30 * time.Minute),
|
|
Mode: stringToPtr("fail"),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "multiregion",
|
|
input: &Job{
|
|
Name: stringToPtr("foo"),
|
|
ID: stringToPtr("bar"),
|
|
ParentID: stringToPtr("lol"),
|
|
Multiregion: &Multiregion{
|
|
Regions: []*MultiregionRegion{
|
|
{
|
|
Name: "west",
|
|
Count: intToPtr(1),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: &Job{
|
|
Multiregion: &Multiregion{
|
|
Strategy: &MultiregionStrategy{
|
|
MaxParallel: intToPtr(0),
|
|
OnFailure: stringToPtr(""),
|
|
},
|
|
Regions: []*MultiregionRegion{
|
|
{
|
|
Name: "west",
|
|
Count: intToPtr(1),
|
|
Datacenters: []string{},
|
|
Meta: map[string]string{},
|
|
},
|
|
},
|
|
},
|
|
Namespace: stringToPtr(DefaultNamespace),
|
|
ID: stringToPtr("bar"),
|
|
Name: stringToPtr("foo"),
|
|
Region: stringToPtr("global"),
|
|
Type: stringToPtr("service"),
|
|
ParentID: stringToPtr("lol"),
|
|
Priority: intToPtr(50),
|
|
AllAtOnce: boolToPtr(false),
|
|
ConsulToken: stringToPtr(""),
|
|
ConsulNamespace: stringToPtr(""),
|
|
VaultToken: stringToPtr(""),
|
|
VaultNamespace: stringToPtr(""),
|
|
NomadTokenID: stringToPtr(""),
|
|
Stop: boolToPtr(false),
|
|
Stable: boolToPtr(false),
|
|
Version: uint64ToPtr(0),
|
|
Status: stringToPtr(""),
|
|
StatusDescription: stringToPtr(""),
|
|
CreateIndex: uint64ToPtr(0),
|
|
ModifyIndex: uint64ToPtr(0),
|
|
JobModifyIndex: uint64ToPtr(0),
|
|
Update: &UpdateStrategy{
|
|
Stagger: timeToPtr(30 * time.Second),
|
|
MaxParallel: intToPtr(1),
|
|
HealthCheck: stringToPtr("checks"),
|
|
MinHealthyTime: timeToPtr(10 * time.Second),
|
|
HealthyDeadline: timeToPtr(5 * time.Minute),
|
|
ProgressDeadline: timeToPtr(10 * time.Minute),
|
|
AutoRevert: boolToPtr(false),
|
|
Canary: intToPtr(0),
|
|
AutoPromote: boolToPtr(false),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
tc.input.Canonicalize()
|
|
if !reflect.DeepEqual(tc.input, tc.expected) {
|
|
t.Fatalf("Name: %v, Diffs:\n%v", tc.name, pretty.Diff(tc.expected, tc.input))
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestJobs_EnforceRegister(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
require := require.New(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Listing jobs before registering returns nothing
|
|
resp, _, err := jobs.List(nil)
|
|
require.Nil(err)
|
|
require.Empty(resp)
|
|
|
|
// Create a job and attempt to register it with an incorrect index.
|
|
job := testJob()
|
|
resp2, _, err := jobs.EnforceRegister(job, 10, nil)
|
|
require.NotNil(err)
|
|
require.Contains(err.Error(), RegisterEnforceIndexErrPrefix)
|
|
|
|
// Register
|
|
resp2, wm, err := jobs.EnforceRegister(job, 0, nil)
|
|
require.Nil(err)
|
|
require.NotNil(resp2)
|
|
require.NotZero(resp2.EvalID)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the jobs back out again
|
|
resp, qm, err := jobs.List(nil)
|
|
require.Nil(err)
|
|
require.Len(resp, 1)
|
|
require.Equal(*job.ID, resp[0].ID)
|
|
assertQueryMeta(t, qm)
|
|
|
|
// Fail at incorrect index
|
|
curIndex := resp[0].JobModifyIndex
|
|
resp2, _, err = jobs.EnforceRegister(job, 123456, nil)
|
|
require.NotNil(err)
|
|
require.Contains(err.Error(), RegisterEnforceIndexErrPrefix)
|
|
|
|
// Works at correct index
|
|
resp3, wm, err := jobs.EnforceRegister(job, curIndex, nil)
|
|
require.Nil(err)
|
|
require.NotNil(resp3)
|
|
require.NotZero(resp3.EvalID)
|
|
assertWriteMeta(t, wm)
|
|
}
|
|
|
|
func TestJobs_Revert(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Register twice
|
|
job := testJob()
|
|
resp, wm, err := jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
if resp == nil || resp.EvalID == "" {
|
|
t.Fatalf("missing eval id")
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
job.Meta = map[string]string{"foo": "new"}
|
|
resp, wm, err = jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
if resp == nil || resp.EvalID == "" {
|
|
t.Fatalf("missing eval id")
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Fail revert at incorrect enforce
|
|
_, _, err = jobs.Revert(*job.ID, 0, uint64ToPtr(10), nil, "", "")
|
|
if err == nil || !strings.Contains(err.Error(), "enforcing version") {
|
|
t.Fatalf("expected enforcement error: %v", err)
|
|
}
|
|
|
|
// Works at correct index
|
|
revertResp, wm, err := jobs.Revert(*job.ID, 0, uint64ToPtr(1), nil, "", "")
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
if revertResp.EvalID == "" {
|
|
t.Fatalf("missing eval id")
|
|
}
|
|
if revertResp.EvalCreateIndex == 0 {
|
|
t.Fatalf("bad eval create index")
|
|
}
|
|
if revertResp.JobModifyIndex == 0 {
|
|
t.Fatalf("bad job modify index")
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
}
|
|
|
|
func TestJobs_Info(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Trying to retrieve a job by ID before it exists
|
|
// returns an error
|
|
id := "job-id/with\\troublesome:characters\n?&字"
|
|
_, _, err := jobs.Info(id, nil)
|
|
if err == nil || !strings.Contains(err.Error(), "not found") {
|
|
t.Fatalf("expected not found error, got: %#v", err)
|
|
}
|
|
|
|
// Register the job
|
|
job := testJob()
|
|
job.ID = &id
|
|
_, wm, err := jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the job again and ensure it exists
|
|
result, qm, err := jobs.Info(id, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertQueryMeta(t, qm)
|
|
|
|
// Check that the result is what we expect
|
|
if result == nil || *result.ID != *job.ID {
|
|
t.Fatalf("expect: %#v, got: %#v", job, result)
|
|
}
|
|
}
|
|
|
|
func TestJobs_ScaleInvalidAction(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Check if invalid inputs fail
|
|
tests := []struct {
|
|
jobID string
|
|
group string
|
|
value int
|
|
want string
|
|
}{
|
|
{"", "", 1, "404"},
|
|
{"i-dont-exist", "", 1, "400"},
|
|
{"", "i-dont-exist", 1, "404"},
|
|
{"i-dont-exist", "me-neither", 1, "404"},
|
|
}
|
|
for _, test := range tests {
|
|
_, _, err := jobs.Scale(test.jobID, test.group, &test.value, "reason", false, nil, nil)
|
|
require.Errorf(err, "expected jobs.Scale(%s, %s) to fail", test.jobID, test.group)
|
|
require.Containsf(err.Error(), test.want, "jobs.Scale(%s, %s) error doesn't contain %s, got: %s", test.jobID, test.group, test.want, err)
|
|
}
|
|
|
|
// Register test job
|
|
job := testJob()
|
|
job.ID = stringToPtr("TestJobs_Scale")
|
|
_, wm, err := jobs.Register(job, nil)
|
|
require.NoError(err)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Perform a scaling action with bad group name, verify error
|
|
_, _, err = jobs.Scale(*job.ID, "incorrect-group-name", intToPtr(2),
|
|
"because", false, nil, nil)
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "does not exist")
|
|
}
|
|
|
|
func TestJobs_Versions(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Trying to retrieve a job by ID before it exists returns an error
|
|
_, _, _, err := jobs.Versions("job1", false, nil)
|
|
if err == nil || !strings.Contains(err.Error(), "not found") {
|
|
t.Fatalf("expected not found error, got: %#v", err)
|
|
}
|
|
|
|
// Register the job
|
|
job := testJob()
|
|
_, wm, err := jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the job again and ensure it exists
|
|
result, _, qm, err := jobs.Versions("job1", false, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertQueryMeta(t, qm)
|
|
|
|
// Check that the result is what we expect
|
|
if len(result) == 0 || *result[0].ID != *job.ID {
|
|
t.Fatalf("expect: %#v, got: %#v", job, result)
|
|
}
|
|
}
|
|
|
|
func TestJobs_PrefixList(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Listing when nothing exists returns empty
|
|
results, _, err := jobs.PrefixList("dummy")
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
if n := len(results); n != 0 {
|
|
t.Fatalf("expected 0 jobs, got: %d", n)
|
|
}
|
|
|
|
// Register the job
|
|
job := testJob()
|
|
_, wm, err := jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the job again and ensure it exists
|
|
// Listing when nothing exists returns empty
|
|
results, _, err = jobs.PrefixList((*job.ID)[:1])
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
// Check if we have the right list
|
|
if len(results) != 1 || results[0].ID != *job.ID {
|
|
t.Fatalf("bad: %#v", results)
|
|
}
|
|
}
|
|
|
|
func TestJobs_List(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Listing when nothing exists returns empty
|
|
results, _, err := jobs.List(nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
if n := len(results); n != 0 {
|
|
t.Fatalf("expected 0 jobs, got: %d", n)
|
|
}
|
|
|
|
// Register the job
|
|
job := testJob()
|
|
_, wm, err := jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the job again and ensure it exists
|
|
// Listing when nothing exists returns empty
|
|
results, _, err = jobs.List(nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
// Check if we have the right list
|
|
if len(results) != 1 || results[0].ID != *job.ID {
|
|
t.Fatalf("bad: %#v", results)
|
|
}
|
|
}
|
|
|
|
func TestJobs_Allocations(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Looking up by a nonexistent job returns nothing
|
|
allocs, qm, err := jobs.Allocations("job1", true, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
if qm.LastIndex != 0 {
|
|
t.Fatalf("bad index: %d", qm.LastIndex)
|
|
}
|
|
if n := len(allocs); n != 0 {
|
|
t.Fatalf("expected 0 allocs, got: %d", n)
|
|
}
|
|
|
|
// TODO: do something here to create some allocations for
|
|
// an existing job, lookup again.
|
|
}
|
|
|
|
func TestJobs_Evaluations(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Looking up by a nonexistent job ID returns nothing
|
|
evals, qm, err := jobs.Evaluations("job1", nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
if qm.LastIndex != 0 {
|
|
t.Fatalf("bad index: %d", qm.LastIndex)
|
|
}
|
|
if n := len(evals); n != 0 {
|
|
t.Fatalf("expected 0 evals, got: %d", n)
|
|
}
|
|
|
|
// Insert a job. This also creates an evaluation so we should
|
|
// be able to query that out after.
|
|
job := testJob()
|
|
resp, wm, err := jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Look up the evaluations again.
|
|
evals, qm, err = jobs.Evaluations("job1", nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertQueryMeta(t, qm)
|
|
|
|
// Check that we got the evals back, evals are in order most recent to least recent
|
|
// so the last eval is the original registered eval
|
|
idx := len(evals) - 1
|
|
if n := len(evals); n == 0 || evals[idx].ID != resp.EvalID {
|
|
t.Fatalf("expected >= 1 eval (%s), got: %#v", resp.EvalID, evals[idx])
|
|
}
|
|
}
|
|
|
|
func TestJobs_Deregister(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Register a new job
|
|
job := testJob()
|
|
_, wm, err := jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Attempting delete on non-existing job returns an error
|
|
if _, _, err = jobs.Deregister("nope", false, nil); err != nil {
|
|
t.Fatalf("unexpected error deregistering job: %v", err)
|
|
}
|
|
|
|
// Do a soft deregister of an existing job
|
|
evalID, wm3, err := jobs.Deregister("job1", false, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm3)
|
|
if evalID == "" {
|
|
t.Fatalf("missing eval ID")
|
|
}
|
|
|
|
// Check that the job is still queryable
|
|
out, qm1, err := jobs.Info("job1", nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertQueryMeta(t, qm1)
|
|
if out == nil {
|
|
t.Fatalf("missing job")
|
|
}
|
|
|
|
// Do a purge deregister of an existing job
|
|
evalID, wm4, err := jobs.Deregister("job1", true, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm4)
|
|
if evalID == "" {
|
|
t.Fatalf("missing eval ID")
|
|
}
|
|
|
|
// Check that the job is really gone
|
|
result, qm, err := jobs.List(nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertQueryMeta(t, qm)
|
|
if n := len(result); n != 0 {
|
|
t.Fatalf("expected 0 jobs, got: %d", n)
|
|
}
|
|
}
|
|
|
|
func TestJobs_Deregister_EvalPriority(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
requireAssert := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
|
|
// Listing jobs before registering returns nothing
|
|
listResp, _, err := c.Jobs().List(nil)
|
|
requireAssert.Nil(err)
|
|
requireAssert.Len(listResp, 0)
|
|
|
|
// Create a job and register it.
|
|
job := testJob()
|
|
registerResp, wm, err := c.Jobs().Register(job, nil)
|
|
requireAssert.Nil(err)
|
|
requireAssert.NotNil(registerResp)
|
|
requireAssert.NotEmpty(registerResp.EvalID)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Deregister the job with an eval priority.
|
|
evalID, _, err := c.Jobs().DeregisterOpts(*job.ID, &DeregisterOptions{EvalPriority: 97}, nil)
|
|
requireAssert.NoError(err)
|
|
requireAssert.NotEmpty(t, evalID)
|
|
|
|
// Lookup the eval and check the priority on it.
|
|
evalInfo, _, err := c.Evaluations().Info(evalID, nil)
|
|
requireAssert.NoError(err)
|
|
requireAssert.Equal(97, evalInfo.Priority)
|
|
}
|
|
|
|
func TestJobs_Deregister_NoEvalPriority(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
requireAssert := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
|
|
// Listing jobs before registering returns nothing
|
|
listResp, _, err := c.Jobs().List(nil)
|
|
requireAssert.Nil(err)
|
|
requireAssert.Len(listResp, 0)
|
|
|
|
// Create a job and register it.
|
|
job := testJob()
|
|
registerResp, wm, err := c.Jobs().Register(job, nil)
|
|
requireAssert.Nil(err)
|
|
requireAssert.NotNil(registerResp)
|
|
requireAssert.NotEmpty(registerResp.EvalID)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Deregister the job with an eval priority.
|
|
evalID, _, err := c.Jobs().DeregisterOpts(*job.ID, &DeregisterOptions{}, nil)
|
|
requireAssert.NoError(err)
|
|
requireAssert.NotEmpty(t, evalID)
|
|
|
|
// Lookup the eval and check the priority on it.
|
|
evalInfo, _, err := c.Evaluations().Info(evalID, nil)
|
|
requireAssert.NoError(err)
|
|
requireAssert.Equal(*job.Priority, evalInfo.Priority)
|
|
}
|
|
|
|
func TestJobs_ForceEvaluate(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Force-eval on a non-existent job fails
|
|
_, _, err := jobs.ForceEvaluate("job1", nil)
|
|
if err == nil || !strings.Contains(err.Error(), "not found") {
|
|
t.Fatalf("expected not found error, got: %#v", err)
|
|
}
|
|
|
|
// Create a new job
|
|
_, wm, err := jobs.Register(testJob(), nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Try force-eval again
|
|
evalID, wm, err := jobs.ForceEvaluate("job1", nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Retrieve the evals and see if we get a matching one
|
|
evals, qm, err := jobs.Evaluations("job1", nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertQueryMeta(t, qm)
|
|
for _, eval := range evals {
|
|
if eval.ID == evalID {
|
|
return
|
|
}
|
|
}
|
|
t.Fatalf("evaluation %q missing", evalID)
|
|
}
|
|
|
|
func TestJobs_PeriodicForce(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Force-eval on a nonexistent job fails
|
|
_, _, err := jobs.PeriodicForce("job1", nil)
|
|
if err == nil || !strings.Contains(err.Error(), "not found") {
|
|
t.Fatalf("expected not found error, got: %#v", err)
|
|
}
|
|
|
|
// Create a new job
|
|
job := testPeriodicJob()
|
|
_, _, err = jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
testutil.WaitForResult(func() (bool, error) {
|
|
out, _, err := jobs.Info(*job.ID, nil)
|
|
if err != nil || out == nil || *out.ID != *job.ID {
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}, func(err error) {
|
|
t.Fatalf("err: %s", err)
|
|
})
|
|
|
|
// Try force again
|
|
evalID, wm, err := jobs.PeriodicForce(*job.ID, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
if evalID == "" {
|
|
t.Fatalf("empty evalID")
|
|
}
|
|
|
|
// Retrieve the eval
|
|
evals := c.Evaluations()
|
|
eval, qm, err := evals.Info(evalID, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertQueryMeta(t, qm)
|
|
if eval.ID == evalID {
|
|
return
|
|
}
|
|
t.Fatalf("evaluation %q missing", evalID)
|
|
}
|
|
|
|
func TestJobs_Plan(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Create a job and attempt to register it
|
|
job := testJob()
|
|
resp, wm, err := jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
if resp == nil || resp.EvalID == "" {
|
|
t.Fatalf("missing eval id")
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Check that passing a nil job fails
|
|
if _, _, err := jobs.Plan(nil, true, nil); err == nil {
|
|
t.Fatalf("expect an error when job isn't provided")
|
|
}
|
|
|
|
// Make a plan request
|
|
planResp, wm, err := jobs.Plan(job, true, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
if planResp == nil {
|
|
t.Fatalf("nil response")
|
|
}
|
|
|
|
if planResp.JobModifyIndex == 0 {
|
|
t.Fatalf("bad JobModifyIndex value: %#v", planResp)
|
|
}
|
|
if planResp.Diff == nil {
|
|
t.Fatalf("got nil diff: %#v", planResp)
|
|
}
|
|
if planResp.Annotations == nil {
|
|
t.Fatalf("got nil annotations: %#v", planResp)
|
|
}
|
|
// Can make this assertion because there are no clients.
|
|
if len(planResp.CreatedEvals) == 0 {
|
|
t.Fatalf("got no CreatedEvals: %#v", planResp)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Make a plan request w/o the diff
|
|
planResp, wm, err = jobs.Plan(job, false, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
if planResp == nil {
|
|
t.Fatalf("nil response")
|
|
}
|
|
|
|
if planResp.JobModifyIndex == 0 {
|
|
t.Fatalf("bad JobModifyIndex value: %d", planResp.JobModifyIndex)
|
|
}
|
|
if planResp.Diff != nil {
|
|
t.Fatalf("got non-nil diff: %#v", planResp)
|
|
}
|
|
if planResp.Annotations == nil {
|
|
t.Fatalf("got nil annotations: %#v", planResp)
|
|
}
|
|
// Can make this assertion because there are no clients.
|
|
if len(planResp.CreatedEvals) == 0 {
|
|
t.Fatalf("got no CreatedEvals: %#v", planResp)
|
|
}
|
|
}
|
|
|
|
func TestJobs_JobSummary(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Trying to retrieve a job summary before the job exists
|
|
// returns an error
|
|
_, _, err := jobs.Summary("job1", nil)
|
|
if err == nil || !strings.Contains(err.Error(), "not found") {
|
|
t.Fatalf("expected not found error, got: %#v", err)
|
|
}
|
|
|
|
// Register the job
|
|
job := testJob()
|
|
taskName := job.TaskGroups[0].Name
|
|
_, wm, err := jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the job summary again and ensure it exists
|
|
result, qm, err := jobs.Summary("job1", nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertQueryMeta(t, qm)
|
|
|
|
// Check that the result is what we expect
|
|
if *job.ID != result.JobID {
|
|
t.Fatalf("err: expected job id of %s saw %s", *job.ID, result.JobID)
|
|
}
|
|
if _, ok := result.Summary[*taskName]; !ok {
|
|
t.Fatalf("err: unable to find %s key in job summary", *taskName)
|
|
}
|
|
}
|
|
|
|
func TestJobs_NewBatchJob(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
job := NewBatchJob("job1", "myjob", "global", 5)
|
|
expect := &Job{
|
|
Region: stringToPtr("global"),
|
|
ID: stringToPtr("job1"),
|
|
Name: stringToPtr("myjob"),
|
|
Type: stringToPtr(JobTypeBatch),
|
|
Priority: intToPtr(5),
|
|
}
|
|
if !reflect.DeepEqual(job, expect) {
|
|
t.Fatalf("expect: %#v, got: %#v", expect, job)
|
|
}
|
|
}
|
|
|
|
func TestJobs_NewServiceJob(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
job := NewServiceJob("job1", "myjob", "global", 5)
|
|
expect := &Job{
|
|
Region: stringToPtr("global"),
|
|
ID: stringToPtr("job1"),
|
|
Name: stringToPtr("myjob"),
|
|
Type: stringToPtr(JobTypeService),
|
|
Priority: intToPtr(5),
|
|
}
|
|
if !reflect.DeepEqual(job, expect) {
|
|
t.Fatalf("expect: %#v, got: %#v", expect, job)
|
|
}
|
|
}
|
|
|
|
func TestJobs_NewSystemJob(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
job := NewSystemJob("job1", "myjob", "global", 5)
|
|
expect := &Job{
|
|
Region: stringToPtr("global"),
|
|
ID: stringToPtr("job1"),
|
|
Name: stringToPtr("myjob"),
|
|
Type: stringToPtr(JobTypeSystem),
|
|
Priority: intToPtr(5),
|
|
}
|
|
if !reflect.DeepEqual(job, expect) {
|
|
t.Fatalf("expect: %#v, got: %#v", expect, job)
|
|
}
|
|
}
|
|
|
|
func TestJobs_NewSysbatchJob(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
job := NewSysbatchJob("job1", "myjob", "global", 5)
|
|
expect := &Job{
|
|
Region: stringToPtr("global"),
|
|
ID: stringToPtr("job1"),
|
|
Name: stringToPtr("myjob"),
|
|
Type: stringToPtr(JobTypeSysbatch),
|
|
Priority: intToPtr(5),
|
|
}
|
|
require.Equal(t, expect, job)
|
|
}
|
|
|
|
func TestJobs_SetMeta(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
job := &Job{Meta: nil}
|
|
|
|
// Initializes a nil map
|
|
out := job.SetMeta("foo", "bar")
|
|
if job.Meta == nil {
|
|
t.Fatalf("should initialize metadata")
|
|
}
|
|
|
|
// Check that the job was returned
|
|
if job != out {
|
|
t.Fatalf("expect: %#v, got: %#v", job, out)
|
|
}
|
|
|
|
// Setting another pair is additive
|
|
job.SetMeta("baz", "zip")
|
|
expect := map[string]string{"foo": "bar", "baz": "zip"}
|
|
if !reflect.DeepEqual(job.Meta, expect) {
|
|
t.Fatalf("expect: %#v, got: %#v", expect, job.Meta)
|
|
}
|
|
}
|
|
|
|
func TestJobs_Constrain(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
job := &Job{Constraints: nil}
|
|
|
|
// Create and add a constraint
|
|
out := job.Constrain(NewConstraint("kernel.name", "=", "darwin"))
|
|
if n := len(job.Constraints); n != 1 {
|
|
t.Fatalf("expected 1 constraint, got: %d", n)
|
|
}
|
|
|
|
// Check that the job was returned
|
|
if job != out {
|
|
t.Fatalf("expect: %#v, got: %#v", job, out)
|
|
}
|
|
|
|
// Adding another constraint preserves the original
|
|
job.Constrain(NewConstraint("memory.totalbytes", ">=", "128000000"))
|
|
expect := []*Constraint{
|
|
{
|
|
LTarget: "kernel.name",
|
|
RTarget: "darwin",
|
|
Operand: "=",
|
|
},
|
|
{
|
|
LTarget: "memory.totalbytes",
|
|
RTarget: "128000000",
|
|
Operand: ">=",
|
|
},
|
|
}
|
|
if !reflect.DeepEqual(job.Constraints, expect) {
|
|
t.Fatalf("expect: %#v, got: %#v", expect, job.Constraints)
|
|
}
|
|
}
|
|
|
|
func TestJobs_AddAffinity(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
job := &Job{Affinities: nil}
|
|
|
|
// Create and add an affinity
|
|
out := job.AddAffinity(NewAffinity("kernel.version", "=", "4.6", 100))
|
|
if n := len(job.Affinities); n != 1 {
|
|
t.Fatalf("expected 1 affinity, got: %d", n)
|
|
}
|
|
|
|
// Check that the job was returned
|
|
if job != out {
|
|
t.Fatalf("expect: %#v, got: %#v", job, out)
|
|
}
|
|
|
|
// Adding another affinity preserves the original
|
|
job.AddAffinity(NewAffinity("${node.datacenter}", "=", "dc2", 50))
|
|
expect := []*Affinity{
|
|
{
|
|
LTarget: "kernel.version",
|
|
RTarget: "4.6",
|
|
Operand: "=",
|
|
Weight: int8ToPtr(100),
|
|
},
|
|
{
|
|
LTarget: "${node.datacenter}",
|
|
RTarget: "dc2",
|
|
Operand: "=",
|
|
Weight: int8ToPtr(50),
|
|
},
|
|
}
|
|
if !reflect.DeepEqual(job.Affinities, expect) {
|
|
t.Fatalf("expect: %#v, got: %#v", expect, job.Affinities)
|
|
}
|
|
}
|
|
|
|
func TestJobs_Sort(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
jobs := []*JobListStub{
|
|
{ID: "job2"},
|
|
{ID: "job0"},
|
|
{ID: "job1"},
|
|
}
|
|
sort.Sort(JobIDSort(jobs))
|
|
|
|
expect := []*JobListStub{
|
|
{ID: "job0"},
|
|
{ID: "job1"},
|
|
{ID: "job2"},
|
|
}
|
|
if !reflect.DeepEqual(jobs, expect) {
|
|
t.Fatalf("\n\n%#v\n\n%#v", jobs, expect)
|
|
}
|
|
}
|
|
|
|
func TestJobs_AddSpread(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
job := &Job{Spreads: nil}
|
|
|
|
// Create and add a Spread
|
|
spreadTarget := NewSpreadTarget("r1", 50)
|
|
|
|
spread := NewSpread("${meta.rack}", 100, []*SpreadTarget{spreadTarget})
|
|
out := job.AddSpread(spread)
|
|
if n := len(job.Spreads); n != 1 {
|
|
t.Fatalf("expected 1 spread, got: %d", n)
|
|
}
|
|
|
|
// Check that the job was returned
|
|
if job != out {
|
|
t.Fatalf("expect: %#v, got: %#v", job, out)
|
|
}
|
|
|
|
// Adding another spread preserves the original
|
|
spreadTarget2 := NewSpreadTarget("dc1", 100)
|
|
|
|
spread2 := NewSpread("${node.datacenter}", 100, []*SpreadTarget{spreadTarget2})
|
|
job.AddSpread(spread2)
|
|
|
|
expect := []*Spread{
|
|
{
|
|
Attribute: "${meta.rack}",
|
|
Weight: int8ToPtr(100),
|
|
SpreadTarget: []*SpreadTarget{
|
|
{
|
|
Value: "r1",
|
|
Percent: 50,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Attribute: "${node.datacenter}",
|
|
Weight: int8ToPtr(100),
|
|
SpreadTarget: []*SpreadTarget{
|
|
{
|
|
Value: "dc1",
|
|
Percent: 100,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
if !reflect.DeepEqual(job.Spreads, expect) {
|
|
t.Fatalf("expect: %#v, got: %#v", expect, job.Spreads)
|
|
}
|
|
}
|
|
|
|
// TestJobs_ScaleAction tests the scale target for task group count
|
|
func TestJobs_ScaleAction(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
id := "job-id/with\\troublesome:characters\n?&字"
|
|
job := testJobWithScalingPolicy()
|
|
job.ID = &id
|
|
groupName := *job.TaskGroups[0].Name
|
|
origCount := *job.TaskGroups[0].Count
|
|
newCount := origCount + 1
|
|
|
|
// Trying to scale against a target before it exists returns an error
|
|
_, _, err := jobs.Scale(id, "missing", intToPtr(newCount), "this won't work",
|
|
false, nil, nil)
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "not found")
|
|
|
|
// Register the job
|
|
regResp, wm, err := jobs.Register(job, nil)
|
|
require.NoError(err)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Perform scaling action
|
|
scalingResp, wm, err := jobs.Scale(id, groupName,
|
|
intToPtr(newCount), "need more instances", false,
|
|
map[string]interface{}{
|
|
"meta": "data",
|
|
}, nil)
|
|
|
|
require.NoError(err)
|
|
require.NotNil(scalingResp)
|
|
require.NotEmpty(scalingResp.EvalID)
|
|
require.NotEmpty(scalingResp.EvalCreateIndex)
|
|
require.Greater(scalingResp.JobModifyIndex, regResp.JobModifyIndex)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the job again
|
|
resp, _, err := jobs.Info(*job.ID, nil)
|
|
require.NoError(err)
|
|
require.Equal(*resp.TaskGroups[0].Count, newCount)
|
|
|
|
// Check for the scaling event
|
|
status, _, err := jobs.ScaleStatus(*job.ID, nil)
|
|
require.NoError(err)
|
|
require.Len(status.TaskGroups[groupName].Events, 1)
|
|
scalingEvent := status.TaskGroups[groupName].Events[0]
|
|
require.False(scalingEvent.Error)
|
|
require.Equal("need more instances", scalingEvent.Message)
|
|
require.Equal(map[string]interface{}{
|
|
"meta": "data",
|
|
}, scalingEvent.Meta)
|
|
require.Greater(scalingEvent.Time, uint64(0))
|
|
require.NotNil(scalingEvent.EvalID)
|
|
require.Equal(scalingResp.EvalID, *scalingEvent.EvalID)
|
|
require.Equal(int64(origCount), scalingEvent.PreviousCount)
|
|
}
|
|
|
|
func TestJobs_ScaleAction_Error(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
id := "job-id/with\\troublesome:characters\n?&字"
|
|
job := testJobWithScalingPolicy()
|
|
job.ID = &id
|
|
groupName := *job.TaskGroups[0].Name
|
|
prevCount := *job.TaskGroups[0].Count
|
|
|
|
// Register the job
|
|
regResp, wm, err := jobs.Register(job, nil)
|
|
require.NoError(err)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Perform scaling action
|
|
scaleResp, wm, err := jobs.Scale(id, groupName, nil, "something bad happened", true,
|
|
map[string]interface{}{
|
|
"meta": "data",
|
|
}, nil)
|
|
|
|
require.NoError(err)
|
|
require.NotNil(scaleResp)
|
|
require.Empty(scaleResp.EvalID)
|
|
require.Empty(scaleResp.EvalCreateIndex)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the job again
|
|
resp, _, err := jobs.Info(*job.ID, nil)
|
|
require.NoError(err)
|
|
require.Equal(*resp.TaskGroups[0].Count, prevCount)
|
|
require.Equal(regResp.JobModifyIndex, scaleResp.JobModifyIndex)
|
|
require.Empty(scaleResp.EvalCreateIndex)
|
|
require.Empty(scaleResp.EvalID)
|
|
|
|
status, _, err := jobs.ScaleStatus(*job.ID, nil)
|
|
require.NoError(err)
|
|
require.Len(status.TaskGroups[groupName].Events, 1)
|
|
errEvent := status.TaskGroups[groupName].Events[0]
|
|
require.True(errEvent.Error)
|
|
require.Equal("something bad happened", errEvent.Message)
|
|
require.Equal(map[string]interface{}{
|
|
"meta": "data",
|
|
}, errEvent.Meta)
|
|
require.Greater(errEvent.Time, uint64(0))
|
|
require.Nil(errEvent.EvalID)
|
|
}
|
|
|
|
func TestJobs_ScaleAction_Noop(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
id := "job-id/with\\troublesome:characters\n?&字"
|
|
job := testJobWithScalingPolicy()
|
|
job.ID = &id
|
|
groupName := *job.TaskGroups[0].Name
|
|
prevCount := *job.TaskGroups[0].Count
|
|
|
|
// Register the job
|
|
regResp, wm, err := jobs.Register(job, nil)
|
|
require.NoError(err)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Perform scaling action
|
|
scaleResp, wm, err := jobs.Scale(id, groupName, nil, "no count, just informative",
|
|
false, map[string]interface{}{
|
|
"meta": "data",
|
|
}, nil)
|
|
|
|
require.NoError(err)
|
|
require.NotNil(scaleResp)
|
|
require.Empty(scaleResp.EvalID)
|
|
require.Empty(scaleResp.EvalCreateIndex)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the job again
|
|
resp, _, err := jobs.Info(*job.ID, nil)
|
|
require.NoError(err)
|
|
require.Equal(*resp.TaskGroups[0].Count, prevCount)
|
|
require.Equal(regResp.JobModifyIndex, scaleResp.JobModifyIndex)
|
|
require.Empty(scaleResp.EvalCreateIndex)
|
|
require.Empty(scaleResp.EvalID)
|
|
|
|
status, _, err := jobs.ScaleStatus(*job.ID, nil)
|
|
require.NoError(err)
|
|
require.Len(status.TaskGroups[groupName].Events, 1)
|
|
noopEvent := status.TaskGroups[groupName].Events[0]
|
|
require.False(noopEvent.Error)
|
|
require.Equal("no count, just informative", noopEvent.Message)
|
|
require.Equal(map[string]interface{}{
|
|
"meta": "data",
|
|
}, noopEvent.Meta)
|
|
require.Greater(noopEvent.Time, uint64(0))
|
|
require.Nil(noopEvent.EvalID)
|
|
}
|
|
|
|
// TestJobs_ScaleStatus tests the /scale status endpoint for task group count
|
|
func TestJobs_ScaleStatus(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
|
|
require := require.New(t)
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
jobs := c.Jobs()
|
|
|
|
// Trying to retrieve a status before it exists returns an error
|
|
id := "job-id/with\\troublesome:characters\n?&字"
|
|
_, _, err := jobs.ScaleStatus(id, nil)
|
|
require.Error(err)
|
|
require.Contains(err.Error(), "not found")
|
|
|
|
// Register the job
|
|
job := testJob()
|
|
job.ID = &id
|
|
groupName := *job.TaskGroups[0].Name
|
|
groupCount := *job.TaskGroups[0].Count
|
|
_, wm, err := jobs.Register(job, nil)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the scaling endpoint and verify success
|
|
result, qm, err := jobs.ScaleStatus(id, nil)
|
|
require.NoError(err)
|
|
assertQueryMeta(t, qm)
|
|
|
|
// Check that the result is what we expect
|
|
require.Equal(groupCount, result.TaskGroups[groupName].Desired)
|
|
}
|
|
|
|
func TestJobs_Services(t *testing.T) {
|
|
// TODO(jrasell) add tests once registration process is in place.
|
|
}
|
|
|
|
// TestJobs_Parse asserts ParseHCL and ParseHCLOpts use the API to parse HCL.
|
|
func TestJobs_Parse(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
|
|
jobspec := `job "example" {}`
|
|
|
|
// Assert ParseHCL returns an error if Nomad is not running to ensure
|
|
// that parsing is done server-side and not via the jobspec package.
|
|
{
|
|
c, err := NewClient(DefaultConfig())
|
|
require.NoError(t, err)
|
|
|
|
_, err = c.Jobs().ParseHCL(jobspec, false)
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), "Put")
|
|
}
|
|
|
|
c, s := makeClient(t, nil, nil)
|
|
defer s.Stop()
|
|
|
|
// Test ParseHCL
|
|
job1, err := c.Jobs().ParseHCL(jobspec, false)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "example", *job1.Name)
|
|
require.Nil(t, job1.Namespace)
|
|
|
|
job1Canonicalized, err := c.Jobs().ParseHCL(jobspec, true)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "example", *job1Canonicalized.Name)
|
|
require.Equal(t, "default", *job1Canonicalized.Namespace)
|
|
require.NotEqual(t, job1, job1Canonicalized)
|
|
|
|
// Test ParseHCLOpts
|
|
req := &JobsParseRequest{
|
|
JobHCL: jobspec,
|
|
HCLv1: false,
|
|
Canonicalize: false,
|
|
}
|
|
|
|
job2, err := c.Jobs().ParseHCLOpts(req)
|
|
require.NoError(t, err)
|
|
require.Equal(t, job1, job2)
|
|
|
|
// Test ParseHCLOpts with Canonicalize=true
|
|
req = &JobsParseRequest{
|
|
JobHCL: jobspec,
|
|
HCLv1: false,
|
|
Canonicalize: true,
|
|
}
|
|
job2Canonicalized, err := c.Jobs().ParseHCLOpts(req)
|
|
require.NoError(t, err)
|
|
require.Equal(t, job1Canonicalized, job2Canonicalized)
|
|
|
|
// Test ParseHCLOpts with HCLv1=true
|
|
req = &JobsParseRequest{
|
|
JobHCL: jobspec,
|
|
HCLv1: true,
|
|
Canonicalize: false,
|
|
}
|
|
|
|
job3, err := c.Jobs().ParseHCLOpts(req)
|
|
require.NoError(t, err)
|
|
require.Equal(t, job1, job3)
|
|
|
|
// Test ParseHCLOpts with HCLv1=true and Canonicalize=true
|
|
req = &JobsParseRequest{
|
|
JobHCL: jobspec,
|
|
HCLv1: true,
|
|
Canonicalize: true,
|
|
}
|
|
job3Canonicalized, err := c.Jobs().ParseHCLOpts(req)
|
|
require.NoError(t, err)
|
|
require.Equal(t, job1Canonicalized, job3Canonicalized)
|
|
}
|