From 457ce6ca1ba0f659dc96e348321ec062f46c326f Mon Sep 17 00:00:00 2001
From: Nick Ethier <nethier@hashicorp.com>
Date: Thu, 20 Dec 2018 01:05:17 -0500
Subject: [PATCH] fix tests

---
 command/alloc_status_test.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/command/alloc_status_test.go b/command/alloc_status_test.go
index e8d8f996d..fbd199424 100644
--- a/command/alloc_status_test.go
+++ b/command/alloc_status_test.go
@@ -96,7 +96,8 @@ func TestAllocStatusCommand_Run(t *testing.T) {
 			return false, err
 		}
 		for _, node := range nodes {
-			if node.Status == structs.NodeStatusReady {
+			if _, ok := node.Drivers["mock_driver"]; ok &&
+				node.Status == structs.NodeStatusReady {
 				return true, nil
 			}
 		}