luxolus/open-nomad
2
0
Fork 0
Code Issues 1 Pull requests Packages Releases Wiki Activity

Fail validation if system job has affinities

Browse source
This commit is contained in:
Preetha Appan 2018-07-23 11:06:49 -05:00
parent 0bc030c6fb
commit dbbb4a957a
No known key found for this signature in database
GPG key ID: 9F7C19990A50EAFC
2 changed files with 65 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
nomad/structs/structs.go
View file

@ -2172,11 +2172,16 @@ func (j *Job) Validate() error {
mErr.Errors = append(mErr.Errors, outer) mErr.Errors = append(mErr.Errors, outer)
} }
} }
if j.Type == JobTypeSystem {
for idx, affinity := range j.Affinities { if j.Affinities != nil {
if err := affinity.Validate(); err != nil { mErr.Errors = append(mErr.Errors, fmt.Errorf("System jobs should not have an affinity stanza"))
outer := fmt.Errorf("Affinity %d validation failed: %s", idx+1, err) }
mErr.Errors = append(mErr.Errors, outer) } else {
for idx, affinity := range j.Affinities {
if err := affinity.Validate(); err != nil {
outer := fmt.Errorf("Affinity %d validation failed: %s", idx+1, err)
mErr.Errors = append(mErr.Errors, outer)
}
} }
} }
@ -3424,11 +3429,16 @@ func (tg *TaskGroup) Validate(j *Job) error {
mErr.Errors = append(mErr.Errors, outer) mErr.Errors = append(mErr.Errors, outer)
} }
} }
if j.Type == JobTypeSystem {
for idx, affinity := range tg.Affinities { if tg.Affinities != nil {
if err := affinity.Validate(); err != nil { mErr.Errors = append(mErr.Errors, fmt.Errorf("System jobs should not have an affinity stanza"))
outer := fmt.Errorf("Affinity %d validation failed: %s", idx+1, err) }
mErr.Errors = append(mErr.Errors, outer) } else {
for idx, affinity := range tg.Affinities {
if err := affinity.Validate(); err != nil {
outer := fmt.Errorf("Affinity %d validation failed: %s", idx+1, err)
mErr.Errors = append(mErr.Errors, outer)
}
} }
} }
@ -3528,7 +3538,7 @@ func (tg *TaskGroup) Validate(j *Job) error {
// Validate the tasks // Validate the tasks
for _, task := range tg.Tasks { for _, task := range tg.Tasks {
if err := task.Validate(tg.EphemeralDisk); err != nil { if err := task.Validate(tg.EphemeralDisk, j.Type); err != nil {
outer := fmt.Errorf("Task %s validation failed: %v", task.Name, err) outer := fmt.Errorf("Task %s validation failed: %v", task.Name, err)
mErr.Errors = append(mErr.Errors, outer) mErr.Errors = append(mErr.Errors, outer)
} }
@ -4164,7 +4174,7 @@ func (t *Task) GoString() string {
} }
// Validate is used to sanity check a task // Validate is used to sanity check a task
func (t *Task) Validate(ephemeralDisk *EphemeralDisk) error { func (t *Task) Validate(ephemeralDisk *EphemeralDisk, jobType string) error {
var mErr multierror.Error var mErr multierror.Error
if t.Name == "" { if t.Name == "" {
mErr.Errors = append(mErr.Errors, errors.New("Missing task name")) mErr.Errors = append(mErr.Errors, errors.New("Missing task name"))
@ -4218,10 +4228,16 @@ func (t *Task) Validate(ephemeralDisk *EphemeralDisk) error {
} }
} }
for idx, affinity := range t.Affinities { if jobType == JobTypeSystem {
if err := affinity.Validate(); err != nil { if t.Affinities != nil {
outer := fmt.Errorf("Affinity %d validation failed: %s", idx+1, err) mErr.Errors = append(mErr.Errors, fmt.Errorf("System jobs should not have an affinity stanza"))
mErr.Errors = append(mErr.Errors, outer) }
} else {
for idx, affinity := range t.Affinities {
if err := affinity.Validate(); err != nil {
outer := fmt.Errorf("Affinity %d validation failed: %s", idx+1, err)
mErr.Errors = append(mErr.Errors, outer)
}
} }
} }

45
nomad/structs/structs_test.go
View file

@ -384,6 +384,27 @@ func TestJob_SystemJob_Validate(t *testing.T) {
if err := j.Validate(); err != nil { if err := j.Validate(); err != nil {
t.Fatalf("unexpected err: %v", err) t.Fatalf("unexpected err: %v", err)
} }
// Add affinities at job, task group and task level, that should fail validation
j.Affinities = []*Affinity{{
Operand: "=",
LTarget: "${node.datacenter}",
RTarget: "dc1",
}}
j.TaskGroups[0].Affinities = []*Affinity{{
Operand: "=",
LTarget: "${meta.rack}",
RTarget: "r1",
}}
j.TaskGroups[0].Tasks[0].Affinities = []*Affinity{{
Operand: "=",
LTarget: "${meta.rack}",
RTarget: "r1",
}}
err = j.Validate()
require.NotNil(t, err)
require.Contains(t, err.Error(), "System jobs should not have an affinity stanza")
} }
func TestJob_VaultPolicies(t *testing.T) { func TestJob_VaultPolicies(t *testing.T) {
@ -739,7 +760,7 @@ func TestTaskGroup_Validate(t *testing.T) {
func TestTask_Validate(t *testing.T) { func TestTask_Validate(t *testing.T) {
task := &Task{} task := &Task{}
ephemeralDisk := DefaultEphemeralDisk() ephemeralDisk := DefaultEphemeralDisk()
err := task.Validate(ephemeralDisk) err := task.Validate(ephemeralDisk, JobTypeBatch)
mErr := err.(*multierror.Error) mErr := err.(*multierror.Error)
if !strings.Contains(mErr.Errors[0].Error(), "task name") { if !strings.Contains(mErr.Errors[0].Error(), "task name") {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
@ -752,7 +773,7 @@ func TestTask_Validate(t *testing.T) {
} }
task = &Task{Name: "web/foo"} task = &Task{Name: "web/foo"}
err = task.Validate(ephemeralDisk) err = task.Validate(ephemeralDisk, JobTypeBatch)
mErr = err.(*multierror.Error) mErr = err.(*multierror.Error)
if !strings.Contains(mErr.Errors[0].Error(), "slashes") { if !strings.Contains(mErr.Errors[0].Error(), "slashes") {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
@ -769,7 +790,7 @@ func TestTask_Validate(t *testing.T) {
LogConfig: DefaultLogConfig(), LogConfig: DefaultLogConfig(),
} }
ephemeralDisk.SizeMB = 200 ephemeralDisk.SizeMB = 200
err = task.Validate(ephemeralDisk) err = task.Validate(ephemeralDisk, JobTypeBatch)
if err != nil { if err != nil {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
} }
@ -783,7 +804,7 @@ func TestTask_Validate(t *testing.T) {
LTarget: "${meta.rack}", LTarget: "${meta.rack}",
}) })
err = task.Validate(ephemeralDisk) err = task.Validate(ephemeralDisk, JobTypeBatch)
mErr = err.(*multierror.Error) mErr = err.(*multierror.Error)
if !strings.Contains(mErr.Errors[0].Error(), "task level: distinct_hosts") { if !strings.Contains(mErr.Errors[0].Error(), "task level: distinct_hosts") {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
@ -866,7 +887,7 @@ func TestTask_Validate_Services(t *testing.T) {
}, },
} }
err := task.Validate(ephemeralDisk) err := task.Validate(ephemeralDisk, JobTypeService)
if err == nil { if err == nil {
t.Fatal("expected an error") t.Fatal("expected an error")
} }
@ -887,7 +908,7 @@ func TestTask_Validate_Services(t *testing.T) {
t.Fatalf("err: %v", err) t.Fatalf("err: %v", err)
} }
if err = task1.Validate(ephemeralDisk); err != nil { if err = task1.Validate(ephemeralDisk, JobTypeService); err != nil {
t.Fatalf("err : %v", err) t.Fatalf("err : %v", err)
} }
} }
@ -946,7 +967,7 @@ func TestTask_Validate_Service_AddressMode_Ok(t *testing.T) {
for _, service := range cases { for _, service := range cases {
task := getTask(service) task := getTask(service)
t.Run(service.Name, func(t *testing.T) { t.Run(service.Name, func(t *testing.T) {
if err := task.Validate(ephemeralDisk); err != nil { if err := task.Validate(ephemeralDisk, JobTypeService); err != nil {
t.Fatalf("unexpected err: %v", err) t.Fatalf("unexpected err: %v", err)
} }
}) })
@ -999,7 +1020,7 @@ func TestTask_Validate_Service_AddressMode_Bad(t *testing.T) {
for _, service := range cases { for _, service := range cases {
task := getTask(service) task := getTask(service)
t.Run(service.Name, func(t *testing.T) { t.Run(service.Name, func(t *testing.T) {
err := task.Validate(ephemeralDisk) err := task.Validate(ephemeralDisk, JobTypeService)
if err == nil { if err == nil {
t.Fatalf("expected an error") t.Fatalf("expected an error")
} }
@ -1320,7 +1341,7 @@ func TestTask_Validate_LogConfig(t *testing.T) {
SizeMB: 1, SizeMB: 1,
} }
err := task.Validate(ephemeralDisk) err := task.Validate(ephemeralDisk, JobTypeService)
mErr := err.(*multierror.Error) mErr := err.(*multierror.Error)
if !strings.Contains(mErr.Errors[3].Error(), "log storage") { if !strings.Contains(mErr.Errors[3].Error(), "log storage") {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
@ -1337,7 +1358,7 @@ func TestTask_Validate_Template(t *testing.T) {
SizeMB: 1, SizeMB: 1,
} }
err := task.Validate(ephemeralDisk) err := task.Validate(ephemeralDisk, JobTypeService)
if !strings.Contains(err.Error(), "Template 1 validation failed") { if !strings.Contains(err.Error(), "Template 1 validation failed") {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
} }
@ -1350,7 +1371,7 @@ func TestTask_Validate_Template(t *testing.T) {
} }
task.Templates = []*Template{good, good} task.Templates = []*Template{good, good}
err = task.Validate(ephemeralDisk) err = task.Validate(ephemeralDisk, JobTypeService)
if !strings.Contains(err.Error(), "same destination as") { if !strings.Contains(err.Error(), "same destination as") {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
} }
@ -1363,7 +1384,7 @@ func TestTask_Validate_Template(t *testing.T) {
}, },
} }
err = task.Validate(ephemeralDisk) err = task.Validate(ephemeralDisk, JobTypeService)
if err == nil { if err == nil {
t.Fatalf("expected error from Template.Validate") t.Fatalf("expected error from Template.Validate")
} }