123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158 |
- package links
- import (
- "github.com/docker/docker/nat"
- "strings"
- "testing"
- )
- func TestLinkNaming(t *testing.T) {
- ports := make(nat.PortSet)
- ports[nat.Port("6379/tcp")] = struct{}{}
- link, err := NewLink("172.0.17.3", "172.0.17.2", "/db/docker-1", nil, ports, nil)
- 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]
- }
- value, ok := env["DOCKER_1_PORT"]
- if !ok {
- t.Fatalf("DOCKER_1_PORT not found in env")
- }
- if value != "tcp://172.0.17.2:6379" {
- t.Fatalf("Expected 172.0.17.2:6379, got %s", env["DOCKER_1_PORT"])
- }
- }
- func TestLinkNew(t *testing.T) {
- ports := make(nat.PortSet)
- ports[nat.Port("6379/tcp")] = struct{}{}
- link, err := NewLink("172.0.17.3", "172.0.17.2", "/db/docker", nil, ports, nil)
- 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()
- }
- for _, p := range link.Ports {
- if p != nat.Port("6379/tcp") {
- t.Fail()
- }
- }
- }
- func TestLinkEnv(t *testing.T) {
- ports := make(nat.PortSet)
- ports[nat.Port("6379/tcp")] = struct{}{}
- link, err := NewLink("172.0.17.3", "172.0.17.2", "/db/docker", []string{"PASSWORD=gordon"}, ports, nil)
- 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"])
- }
- }
- func TestLinkMultipleEnv(t *testing.T) {
- ports := make(nat.PortSet)
- ports[nat.Port("6379/tcp")] = struct{}{}
- ports[nat.Port("6380/tcp")] = struct{}{}
- ports[nat.Port("6381/tcp")] = struct{}{}
- link, err := NewLink("172.0.17.3", "172.0.17.2", "/db/docker", []string{"PASSWORD=gordon"}, ports, nil)
- 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_START"] != "tcp://172.0.17.2:6379" {
- t.Fatalf("Expected tcp://172.0.17.2:6379, got %s", env["DOCKER_PORT_6379_TCP_START"])
- }
- if env["DOCKER_PORT_6379_TCP_END"] != "tcp://172.0.17.2:6381" {
- t.Fatalf("Expected tcp://172.0.17.2:6381, got %s", env["DOCKER_PORT_6379_TCP_END"])
- }
- 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_START"] != "6379" {
- t.Fatalf("Expected 6379, got %s", env["DOCKER_PORT_6379_TCP_PORT_START"])
- }
- if env["DOCKER_PORT_6379_TCP_PORT_END"] != "6381" {
- t.Fatalf("Expected 6381, got %s", env["DOCKER_PORT_6379_TCP_PORT_END"])
- }
- 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"])
- }
- }
|