113 lines
2.5 KiB
Go
113 lines
2.5 KiB
Go
package docker
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func newMockLinkContainer(id string, ip string) *Container {
|
|
return &Container{
|
|
Config: &Config{},
|
|
ID: id,
|
|
NetworkSettings: &NetworkSettings{
|
|
IPAddress: ip,
|
|
},
|
|
}
|
|
}
|
|
|
|
func TestLinkNew(t *testing.T) {
|
|
toID := GenerateID()
|
|
fromID := GenerateID()
|
|
|
|
from := newMockLinkContainer(fromID, "172.0.17.2")
|
|
from.Config.Env = []string{}
|
|
from.State = State{Running: true}
|
|
ports := make(map[Port]struct{})
|
|
|
|
ports[Port("6379/tcp")] = struct{}{}
|
|
|
|
from.Config.ExposedPorts = ports
|
|
|
|
to := newMockLinkContainer(toID, "172.0.17.3")
|
|
|
|
link, err := NewLink(to, from, "/db/docker", "172.0.17.1")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if link == nil {
|
|
t.FailNow()
|
|
}
|
|
if link.Name != "/db/docker" {
|
|
t.Fail()
|
|
}
|
|
if link.Alias() != "docker" {
|
|
t.Fail()
|
|
}
|
|
if link.ParentIP != "172.0.17.3" {
|
|
t.Fail()
|
|
}
|
|
if link.ChildIP != "172.0.17.2" {
|
|
t.Fail()
|
|
}
|
|
if link.BridgeInterface != "172.0.17.1" {
|
|
t.Fail()
|
|
}
|
|
for _, p := range link.Ports {
|
|
if p != Port("6379/tcp") {
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestLinkEnv(t *testing.T) {
|
|
toID := GenerateID()
|
|
fromID := GenerateID()
|
|
|
|
from := newMockLinkContainer(fromID, "172.0.17.2")
|
|
from.Config.Env = []string{"PASSWORD=gordon"}
|
|
from.State = State{Running: true}
|
|
ports := make(map[Port]struct{})
|
|
|
|
ports[Port("6379/tcp")] = struct{}{}
|
|
|
|
from.Config.ExposedPorts = ports
|
|
|
|
to := newMockLinkContainer(toID, "172.0.17.3")
|
|
|
|
link, err := NewLink(to, from, "/db/docker", "172.0.17.1")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
rawEnv := link.ToEnv()
|
|
env := make(map[string]string, len(rawEnv))
|
|
for _, e := range rawEnv {
|
|
parts := strings.Split(e, "=")
|
|
if len(parts) != 2 {
|
|
t.FailNow()
|
|
}
|
|
env[parts[0]] = parts[1]
|
|
}
|
|
if env["DOCKER_PORT"] != "tcp://172.0.17.2:6379" {
|
|
t.Fatalf("Expected 172.0.17.2:6379, got %s", env["DOCKER_PORT"])
|
|
}
|
|
if env["DOCKER_PORT_6379_TCP"] != "tcp://172.0.17.2:6379" {
|
|
t.Fatalf("Expected tcp://172.0.17.2:6379, got %s", env["DOCKER_PORT_6379_TCP"])
|
|
}
|
|
if env["DOCKER_PORT_6379_TCP_PROTO"] != "tcp" {
|
|
t.Fatalf("Expected tcp, got %s", env["DOCKER_PORT_6379_TCP_PROTO"])
|
|
}
|
|
if env["DOCKER_PORT_6379_TCP_ADDR"] != "172.0.17.2" {
|
|
t.Fatalf("Expected 172.0.17.2, got %s", env["DOCKER_PORT_6379_TCP_ADDR"])
|
|
}
|
|
if env["DOCKER_PORT_6379_TCP_PORT"] != "6379" {
|
|
t.Fatalf("Expected 6379, got %s", env["DOCKER_PORT_6379_TCP_PORT"])
|
|
}
|
|
if env["DOCKER_NAME"] != "/db/docker" {
|
|
t.Fatalf("Expected /db/docker, got %s", env["DOCKER_NAME"])
|
|
}
|
|
if env["DOCKER_ENV_PASSWORD"] != "gordon" {
|
|
t.Fatalf("Expected gordon, got %s", env["DOCKER_ENV_PASSWORD"])
|
|
}
|
|
}
|