links_test.go 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. package docker
  2. import (
  3. "github.com/dotcloud/docker/nat"
  4. "github.com/dotcloud/docker/runconfig"
  5. "strings"
  6. "testing"
  7. )
  8. func newMockLinkContainer(id string, ip string) *Container {
  9. return &Container{
  10. Config: &runconfig.Config{},
  11. ID: id,
  12. NetworkSettings: &NetworkSettings{
  13. IPAddress: ip,
  14. },
  15. }
  16. }
  17. func TestLinkNew(t *testing.T) {
  18. toID := GenerateID()
  19. fromID := GenerateID()
  20. from := newMockLinkContainer(fromID, "172.0.17.2")
  21. from.Config.Env = []string{}
  22. from.State = State{Running: true}
  23. ports := make(nat.PortSet)
  24. ports[nat.Port("6379/tcp")] = struct{}{}
  25. from.Config.ExposedPorts = ports
  26. to := newMockLinkContainer(toID, "172.0.17.3")
  27. link, err := NewLink(to, from, "/db/docker", nil)
  28. if err != nil {
  29. t.Fatal(err)
  30. }
  31. if link == nil {
  32. t.FailNow()
  33. }
  34. if link.Name != "/db/docker" {
  35. t.Fail()
  36. }
  37. if link.Alias() != "docker" {
  38. t.Fail()
  39. }
  40. if link.ParentIP != "172.0.17.3" {
  41. t.Fail()
  42. }
  43. if link.ChildIP != "172.0.17.2" {
  44. t.Fail()
  45. }
  46. for _, p := range link.Ports {
  47. if p != nat.Port("6379/tcp") {
  48. t.Fail()
  49. }
  50. }
  51. }
  52. func TestLinkEnv(t *testing.T) {
  53. toID := GenerateID()
  54. fromID := GenerateID()
  55. from := newMockLinkContainer(fromID, "172.0.17.2")
  56. from.Config.Env = []string{"PASSWORD=gordon"}
  57. from.State = State{Running: true}
  58. ports := make(nat.PortSet)
  59. ports[nat.Port("6379/tcp")] = struct{}{}
  60. from.Config.ExposedPorts = ports
  61. to := newMockLinkContainer(toID, "172.0.17.3")
  62. link, err := NewLink(to, from, "/db/docker", nil)
  63. if err != nil {
  64. t.Fatal(err)
  65. }
  66. rawEnv := link.ToEnv()
  67. env := make(map[string]string, len(rawEnv))
  68. for _, e := range rawEnv {
  69. parts := strings.Split(e, "=")
  70. if len(parts) != 2 {
  71. t.FailNow()
  72. }
  73. env[parts[0]] = parts[1]
  74. }
  75. if env["DOCKER_PORT"] != "tcp://172.0.17.2:6379" {
  76. t.Fatalf("Expected 172.0.17.2:6379, got %s", env["DOCKER_PORT"])
  77. }
  78. if env["DOCKER_PORT_6379_TCP"] != "tcp://172.0.17.2:6379" {
  79. t.Fatalf("Expected tcp://172.0.17.2:6379, got %s", env["DOCKER_PORT_6379_TCP"])
  80. }
  81. if env["DOCKER_PORT_6379_TCP_PROTO"] != "tcp" {
  82. t.Fatalf("Expected tcp, got %s", env["DOCKER_PORT_6379_TCP_PROTO"])
  83. }
  84. if env["DOCKER_PORT_6379_TCP_ADDR"] != "172.0.17.2" {
  85. t.Fatalf("Expected 172.0.17.2, got %s", env["DOCKER_PORT_6379_TCP_ADDR"])
  86. }
  87. if env["DOCKER_PORT_6379_TCP_PORT"] != "6379" {
  88. t.Fatalf("Expected 6379, got %s", env["DOCKER_PORT_6379_TCP_PORT"])
  89. }
  90. if env["DOCKER_NAME"] != "/db/docker" {
  91. t.Fatalf("Expected /db/docker, got %s", env["DOCKER_NAME"])
  92. }
  93. if env["DOCKER_ENV_PASSWORD"] != "gordon" {
  94. t.Fatalf("Expected gordon, got %s", env["DOCKER_ENV_PASSWORD"])
  95. }
  96. }