fix tests

This commit is contained in:
Nick Ethier
2018-12-20 01:05:17 -05:00
parent 431d7af281
commit 457ce6ca1b

View File

@@ -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
}
}