allow for arbitrary test interval

Signed-off-by: Evan Hazlett <ejhazlett@gmail.com>
docker · Jan 13, 2015 · a0b3aab · a0b3aab
1 parent 241a003
commit a0b3aab
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/_integration-test/machine_test.go b/_integration-test/machine_test.go
@@ -9,9 +9,7 @@ import (
 )
 
 const (
-	machineName  = "machine-integration-test-%s"
-	waitInterval = 30
-	waitDuration = time.Duration(waitInterval * time.Second)
+	machineName = "machine-integration-test-%s"
 )
 
 func machineCreate(name string, t *testing.T, wg *sync.WaitGroup) {

diff --git a/_integration-test/test_vars.go b/_integration-test/test_vars.go
@@ -4,7 +4,9 @@ import (
 	"fmt"
 	"os"
 	"os/exec"
+	"strconv"
 	"strings"
+	"time"
 )
 
 type (
@@ -16,6 +18,8 @@ type (
 var (
 	machineBinary      = "machine"
 	machineTestDrivers []MachineDriver
+	waitInterval       int
+	waitDuration       time.Duration
 )
 
 func init() {
@@ -39,6 +43,19 @@ func init() {
 		}
 	}
 
+	interval := os.Getenv("MACHINE_TEST_DURATION")
+	if interval == "" {
+		interval = "30"
+	}
+	wait, err := strconv.Atoi(interval)
+	if err != nil {
+		fmt.Printf("invalid interval: %s\n", err)
+		os.Exit(1)
+	}
+
+	waitInterval = wait
+	waitDuration = time.Duration(time.Duration(waitInterval) * time.Second)
+
 	// find machine binary
 	if machineBin := os.Getenv("MACHINE_BINARY"); machineBin != "" {
 		machineBinary = machineBin