Merge pull request #83 from pmenglund/docker-machine

core: add support to connect to docker-machine

dockertest.go

@@ -2,12 +2,13 @@ package dockertest
 
 import (
 	"fmt"
+	"os"
+	"runtime"
+	"time"
+
 	"github.com/cenk/backoff"
 	dc "github.com/fsouza/go-dockerclient"
 	"github.com/pkg/errors"
-	"time"
-	"runtime"
-	"os"
 )
 
 // Pool represents a connection to the docker API and is used to create and remove docker images.
@@ -57,11 +58,19 @@ func NewTLSPool(endpoint, certpath string) (*Pool, error) {
 }
 
 // NewPool creates a new pool. You can pass an empty string to use the default, which is taken from the environment
-// variable DOCKER_URL or if that is not defined a sensible default for the operating system you are on.
+// variable DOCKER_URL, or from docker-machine if the environment variable DOCKER_MACHINE_NAME is set,
+// or if neither is defined a sensible default for the operating system you are on.
 func NewPool(endpoint string) (*Pool, error) {
 	if endpoint == "" {
 		if os.Getenv("DOCKER_URL") != "" {
 			endpoint = os.Getenv("DOCKER_URL")
+		} else if os.Getenv("DOCKER_MACHINE_NAME") != "" {
+			client, err := dc.NewClientFromEnv()
+			if err != nil {
+				return nil, errors.Wrap(err, "")
+			}
+
+			return &Pool{Client: client}, nil
 		} else if runtime.GOOS == "windows" {
 			endpoint = "http://localhost:2375"
 		} else {
@@ -124,7 +133,6 @@ func (d *Pool) Run(repository, tag string, env []string) (*Resource, error) {
 	}, nil
 }
 
-
 // Purge removes a container and linked volumes from docker.
 func (d *Pool) Purge(r *Resource) error {
 	if err := d.Client.KillContainer(dc.KillContainerOptions{ID: r.Container.ID}); err != nil {