fix up scheduler mocks

hashicorp · Mar 1, 2018 · b8695fd · b8695fd
1 parent b417c5e
commit b8695fd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/nomad/mock/mock.go b/nomad/mock/mock.go
@@ -46,7 +46,6 @@ func Node() *structs.Node {
 				},
 			},
 		},
-		Drivers: make(map[string]*structs.DriverInfo),
 		Links: map[string]string{
 			"consul": "foobar.dc1",
 		},

diff --git a/scheduler/feasible_test.go b/scheduler/feasible_test.go
@@ -133,6 +133,9 @@ func TestDriverChecker_HealthChecks(t *testing.T) {
 		mock.Node(),
 		mock.Node(),
 	}
+	for _, e := range nodes {
+		e.Drivers = make(map[string]*structs.DriverInfo)
+	}
 	nodes[0].Attributes["driver.foo"] = "1"
 	nodes[0].Drivers["foo"] = &structs.DriverInfo{
 		Detected:          true,