mirror of https://github.com/docker/docs.git
73 lines
1.4 KiB
Go
73 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type (
|
|
MachineDriver struct {
|
|
name string
|
|
}
|
|
)
|
|
|
|
var (
|
|
machineBinary = "machine"
|
|
machineTestDrivers []MachineDriver
|
|
waitInterval int
|
|
waitDuration time.Duration
|
|
)
|
|
|
|
func init() {
|
|
// 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{
|
|
{
|
|
name: "virtualbox",
|
|
},
|
|
{
|
|
name: "digitalocean",
|
|
},
|
|
}
|
|
}
|
|
|
|
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
|
|
} else {
|
|
whichCmd := exec.Command("which", "machine")
|
|
out, _, err := runCommandWithOutput(whichCmd)
|
|
if err == nil {
|
|
machineBinary = stripTrailingCharacters(out)
|
|
} else {
|
|
fmt.Printf("ERROR: couldn't resolve full path to the Machine binary")
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
}
|