allow filtering driver integration tests

Signed-off-by: Evan Hazlett <ejhazlett@gmail.com>

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{
+	// allow filtering driver tests
-		MachineDriver{
+	if machineTests := os.Getenv("MACHINE_TESTS"); machineTests != "" {
-			name: "virtualbox",
+		tests := strings.Split(machineTests, " ")
-		},
+		for _, test := range tests {
-		MachineDriver{
+			mcn := MachineDriver{
-			name: "digitalocean",
+				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 {