From 93716e9d2ba07f48d88f15ebf62a09c8583674c8 Mon Sep 17 00:00:00 2001
From: Evan Hazlett <ejhazlett@gmail.com>
Date: Tue, 16 Dec 2014 13:14:49 -0500
Subject: [PATCH] allow filtering driver integration tests

Signed-off-by: Evan Hazlett <ejhazlett@gmail.com>
---
 integration-test/test_vars.go | 28 +++++++++++++++++++++-------
 1 file changed, 21 insertions(+), 7 deletions(-)

diff --git a/integration-test/test_vars.go b/integration-test/test_vars.go
index eb38405345..40f0ce9c52 100644
--- a/integration-test/test_vars.go
+++ b/integration-test/test_vars.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"os"
 	"os/exec"
+	"strings"
 )
 
 type (
@@ -18,14 +19,27 @@ var (
 )
 
 func init() {
-	machineTestDrivers = []MachineDriver{
-		MachineDriver{
-			name: "virtualbox",
-		},
-		MachineDriver{
-			name: "digitalocean",
-		},
+	// allow filtering driver tests
+	if machineTests := os.Getenv("MACHINE_TESTS"); machineTests != "" {
+		tests := strings.Split(machineTests, " ")
+		for _, test := range tests {
+			mcn := MachineDriver{
+				name: test,
+			}
+			machineTestDrivers = append(machineTestDrivers, mcn)
+		}
+	} else {
+		machineTestDrivers = []MachineDriver{
+			MachineDriver{
+				name: "virtualbox",
+			},
+			MachineDriver{
+				name: "digitalocean",
+			},
+		}
 	}
+
+	// find machine binary
 	if machineBin := os.Getenv("MACHINE_BINARY"); machineBin != "" {
 		machineBinary = machineBin
 	} else {