dctk_utils_test.go

package dctk

import (
	"net"
	"os/exec"
	"strconv"
	"testing"
	"time"
)

var dockerIP = func() string {
	out, _ := exec.Command("docker", "network", "inspect", "bridge", "--format",
		"{{range .IPAM.Config}}{{.Subnet}}{{end}}").Output()
	subnetStr := string(out[:len(out)-1])

	_, subnet, _ := net.ParseCIDR(subnetStr)

	addrs, _ := net.InterfaceAddrs()
	for _, a := range addrs {
		if ipnet, ok := a.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
			if subnet.Contains(ipnet.IP) {
				return ipnet.IP.String()
			}
		}
	}
	return ""
}()

type externalHubDef struct {
	name  string
	proto string
	port  uint
}

var externalHubDefs = []*externalHubDef{
	{"godcpp-adc", "adc", 1411},
	{"godcpp-nmdc", "nmdc", 1411},
	{"luadch", "adc", 5000},
	{"verlihub", "nmdc", 4111},
}

func foreachExternalHub(t *testing.T, testName string, cb func(t *testing.T, e *externalHub)) {
	for _, def := range externalHubDefs {
		t.Run(def.name, func(t *testing.T) {
			e := newExternalHub(testName, def)
			defer e.close()
			cb(t, e)
		})
	}
}

type externalHub struct {
	su string
}

func newExternalHub(testName string, def *externalHubDef) *externalHub {
	exec.Command("docker", "kill", "dctk-test-hub").Run()
	exec.Command("docker", "wait", "dctk-test-hub").Run()
	exec.Command("docker", "rm", "dctk-test-hub").Run()

	// start hub
	cmd := []string{"docker", "run", "--rm", "-d", "--name=dctk-test-hub"}
	cmd = append(cmd, "dctk-test-hub-"+def.name)
	cmd = append(cmd, testName)
	exec.Command(cmd[0], cmd[1:]...).Run()

	// get hub ip
	byts, _ := exec.Command("docker", "inspect", "-f",
		"{{range .NetworkSettings.Networks}}{{.IPAddress}}{{end}}",
		"dctk-test-hub").Output()
	ip := string(byts[:len(byts)-1])

	address := ip + ":" + strconv.FormatUint(uint64(def.port), 10)

	// wait for hub
	for {
		time.Sleep(1 * time.Second)
		conn, err := net.DialTimeout("tcp", address, 5*time.Second)
		if err != nil {
			continue
		}
		conn.Close()
		break
	}

	return &externalHub{
		su: def.proto + "://" + address,
	}
}

func (e *externalHub) URL() string {
	return e.su
}

func (e *externalHub) close() {
	exec.Command("docker", "kill", "dctk-test-hub").Run()
	exec.Command("docker", "wait", "dctk-test-hub").Run()
}