tests: fix TestAllocations_GarbageCollect

Mahmood Ali · Mahmood Ali · commit 80830221fe4e · 2020-03-24T17:38:59.000-04:00
diff --git a/client/alloc_endpoint_test.go b/client/alloc_endpoint_test.go
@@ -216,10 +216,12 @@ func TestAllocations_GarbageCollect(t *testing.T) {
 
 	a := mock.Alloc()
 	a.Job.TaskGroups[0].Tasks[0].Driver = "mock_driver"
-	a.Job.TaskGroups[0].RestartPolicy = &nstructs.RestartPolicy{
+	rp := &nstructs.RestartPolicy{
 		Attempts: 0,
 		Mode:     nstructs.RestartPolicyModeFail,
 	}
+	a.Job.TaskGroups[0].RestartPolicy = rp
+	a.Job.TaskGroups[0].Tasks[0].RestartPolicy = rp
 	a.Job.TaskGroups[0].Tasks[0].Config = map[string]interface{}{
 		"run_for": "10ms",
 	}

Original file line number	Diff line number	Diff line change
`@@ -216,10 +216,12 @@ func TestAllocations_GarbageCollect(t *testing.T) {`
`216`	`216`
`217`	`217`	`a := mock.Alloc()`
`218`	`218`	`a.Job.TaskGroups[0].Tasks[0].Driver = "mock_driver"`
`219`		`- a.Job.TaskGroups[0].RestartPolicy = &nstructs.RestartPolicy{`
	`219`	`+ rp := &nstructs.RestartPolicy{`
`220`	`220`	`Attempts: 0,`
`221`	`221`	`Mode: nstructs.RestartPolicyModeFail,`
`222`	`222`	`}`
	`223`	`+ a.Job.TaskGroups[0].RestartPolicy = rp`
	`224`	`+ a.Job.TaskGroups[0].Tasks[0].RestartPolicy = rp`
`223`	`225`	`a.Job.TaskGroups[0].Tasks[0].Config = map[string]interface{}{`
`224`	`226`	`"run_for": "10ms",`
`225`	`227`	`}`