66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package dockertest
|
|
|
|
import (
|
|
"camlistore.org/pkg/netutil"
|
|
"fmt"
|
|
"os/exec"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// ContainerID represents a container and offers methods like Kill or IP.
|
|
type ContainerID string
|
|
|
|
// IP retrieves the container's IP address.
|
|
func (c ContainerID) IP() (string, error) {
|
|
return IP(string(c))
|
|
}
|
|
|
|
// Kill runs "docker kill" on the container.
|
|
func (c ContainerID) Kill() error {
|
|
return KillContainer(string(c))
|
|
}
|
|
|
|
// Remove runs "docker rm" on the container
|
|
func (c ContainerID) Remove() error {
|
|
if Debug || c == "nil" {
|
|
return nil
|
|
}
|
|
return runDockerCommand("docker", "rm", "-v", string(c)).Run()
|
|
}
|
|
|
|
// KillRemove calls Kill on the container, and then Remove if there was
|
|
// no error.
|
|
func (c ContainerID) KillRemove() error {
|
|
if err := c.Kill(); err != nil {
|
|
return err
|
|
}
|
|
return c.Remove()
|
|
}
|
|
|
|
// lookup retrieves the ip address of the container, and tries to reach
|
|
// before timeout the tcp address at this ip and given port.
|
|
func (c ContainerID) lookup(ports []int, timeout time.Duration) (ip string, err error) {
|
|
if DockerMachineAvailable {
|
|
var out []byte
|
|
out, err = exec.Command("docker-machine", "ip", DockerMachineName).Output()
|
|
ip = strings.TrimSpace(string(out))
|
|
} else if BindDockerToLocalhost != "" {
|
|
ip = "127.0.0.1"
|
|
} else {
|
|
ip, err = c.IP()
|
|
}
|
|
if err != nil {
|
|
err = fmt.Errorf("error getting IP: %v", err)
|
|
return
|
|
}
|
|
for _, port := range ports {
|
|
addr := fmt.Sprintf("%s:%d", ip, port)
|
|
err = netutil.AwaitReachable(addr, timeout)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|