91 lines
3.4 KiB
Go
91 lines
3.4 KiB
Go
package dockertest
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
)
|
|
|
|
// SetupNSQdContainer sets up a real NSQ instance for testing purposes
|
|
// using a Docker container and executing `/nsqd`. It returns the container ID and its IP address,
|
|
// or makes the test fail on error.
|
|
func SetupNSQdContainer() (c ContainerID, ip string, tcpPort int, httpPort int, err error) {
|
|
// --name nsqd -p 4150:4150 -p 4151:4151 nsqio/nsq /nsqd --broadcast-address=192.168.99.100 --lookupd-tcp-address=192.168.99.100:4160
|
|
tcpPort = RandomPort()
|
|
httpPort = RandomPort()
|
|
tcpForward := fmt.Sprintf("%d:%d", tcpPort, 4150)
|
|
if BindDockerToLocalhost != "" {
|
|
tcpForward = "127.0.0.1:" + tcpForward
|
|
}
|
|
|
|
httpForward := fmt.Sprintf("%d:%d", httpPort, 4151)
|
|
if BindDockerToLocalhost != "" {
|
|
httpForward = "127.0.0.1:" + httpForward
|
|
}
|
|
|
|
c, ip, err = SetupContainer(NSQImageName, tcpPort, 15*time.Second, func() (string, error) {
|
|
return run("--name", GenerateContainerID(), "-d", "-P", "-p", tcpForward, "-p", httpForward, NSQImageName, "/nsqd", fmt.Sprintf("--broadcast-address=%s", ip), fmt.Sprintf("--lookupd-tcp-address=%s:4160", ip))
|
|
})
|
|
return
|
|
}
|
|
|
|
// SetupNSQLookupdContainer sets up a real NSQ instance for testing purposes
|
|
// using a Docker container and executing `/nsqlookupd`. It returns the container ID and its IP address,
|
|
// or makes the test fail on error.
|
|
func SetupNSQLookupdContainer() (c ContainerID, ip string, tcpPort int, httpPort int, err error) {
|
|
// docker run --name lookupd -p 4160:4160 -p 4161:4161 nsqio/nsq /nsqlookupd
|
|
tcpPort = RandomPort()
|
|
httpPort = RandomPort()
|
|
tcpForward := fmt.Sprintf("%d:%d", tcpPort, 4160)
|
|
if BindDockerToLocalhost != "" {
|
|
tcpForward = "127.0.0.1:" + tcpForward
|
|
}
|
|
|
|
httpForward := fmt.Sprintf("%d:%d", httpPort, 4161)
|
|
if BindDockerToLocalhost != "" {
|
|
httpForward = "127.0.0.1:" + httpForward
|
|
}
|
|
|
|
c, ip, err = SetupContainer(NSQImageName, tcpPort, 15*time.Second, func() (string, error) {
|
|
return run("--name", GenerateContainerID(), "-d", "-P", "-p", tcpForward, "-p", httpForward, NSQImageName, "/nsqlookupd")
|
|
})
|
|
return
|
|
}
|
|
|
|
// ConnectToNSQLookupd starts a NSQ image with `/nsqlookupd` running and passes the IP, HTTP port, and TCP port to the connector callback function.
|
|
// The url will match the ip pattern (e.g. 123.123.123.123).
|
|
func ConnectToNSQLookupd(tries int, delay time.Duration, connector func(ip string, httpPort int, tcpPort int) bool) (c ContainerID, err error) {
|
|
c, ip, tcpPort, httpPort, err := SetupNSQLookupdContainer()
|
|
if err != nil {
|
|
return c, fmt.Errorf("Could not set up NSQLookupd container: %v", err)
|
|
}
|
|
|
|
for try := 0; try <= tries; try++ {
|
|
time.Sleep(delay)
|
|
if connector(ip, httpPort, tcpPort) {
|
|
return c, nil
|
|
}
|
|
log.Printf("Try %d failed. Retrying.", try)
|
|
}
|
|
return c, errors.New("Could not set up NSQLookupd container.")
|
|
}
|
|
|
|
// ConnectToNSQd starts a NSQ image with `/nsqd` running and passes the IP, HTTP port, and TCP port to the connector callback function.
|
|
// The url will match the ip pattern (e.g. 123.123.123.123).
|
|
func ConnectToNSQd(tries int, delay time.Duration, connector func(ip string, httpPort int, tcpPort int) bool) (c ContainerID, err error) {
|
|
c, ip, tcpPort, httpPort, err := SetupNSQdContainer()
|
|
if err != nil {
|
|
return c, fmt.Errorf("Could not set up NSQd container: %v", err)
|
|
}
|
|
|
|
for try := 0; try <= tries; try++ {
|
|
time.Sleep(delay)
|
|
if connector(ip, httpPort, tcpPort) {
|
|
return c, nil
|
|
}
|
|
log.Printf("Try %d failed. Retrying.", try)
|
|
}
|
|
return c, errors.New("Could not set up NSQd container.")
|
|
}
|