From b8695fd7fbbc4e584807d78e069f35d8029e5e51 Mon Sep 17 00:00:00 2001
From: Chelsea Holland Komlo <me@chelseakomlo.com>
Date: Thu, 1 Mar 2018 10:25:16 -0500
Subject: [PATCH] fix up scheduler mocks

---
 nomad/mock/mock.go         | 1 -
 scheduler/feasible_test.go | 3 +++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/nomad/mock/mock.go b/nomad/mock/mock.go
index c0228505fac3..7de4987a25b1 100644
--- 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
index 548307263003..224436efceb3 100644
--- 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,