docker_cli_stack_test.go 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186
  1. package main
  2. import (
  3. "encoding/json"
  4. "io/ioutil"
  5. "os"
  6. "sort"
  7. "strings"
  8. "github.com/docker/docker/api/types/swarm"
  9. "github.com/docker/docker/integration-cli/checker"
  10. "github.com/go-check/check"
  11. )
  12. func (s *DockerSwarmSuite) TestStackRemoveUnknown(c *check.C) {
  13. d := s.AddDaemon(c, true, true)
  14. stackArgs := append([]string{"stack", "remove", "UNKNOWN_STACK"})
  15. out, err := d.Cmd(stackArgs...)
  16. c.Assert(err, checker.IsNil)
  17. c.Assert(out, check.Equals, "Nothing found in stack: UNKNOWN_STACK\n")
  18. }
  19. func (s *DockerSwarmSuite) TestStackPSUnknown(c *check.C) {
  20. d := s.AddDaemon(c, true, true)
  21. stackArgs := append([]string{"stack", "ps", "UNKNOWN_STACK"})
  22. out, err := d.Cmd(stackArgs...)
  23. c.Assert(err, checker.IsNil)
  24. c.Assert(out, check.Equals, "Nothing found in stack: UNKNOWN_STACK\n")
  25. }
  26. func (s *DockerSwarmSuite) TestStackServicesUnknown(c *check.C) {
  27. d := s.AddDaemon(c, true, true)
  28. stackArgs := append([]string{"stack", "services", "UNKNOWN_STACK"})
  29. out, err := d.Cmd(stackArgs...)
  30. c.Assert(err, checker.IsNil)
  31. c.Assert(out, check.Equals, "Nothing found in stack: UNKNOWN_STACK\n")
  32. }
  33. func (s *DockerSwarmSuite) TestStackDeployComposeFile(c *check.C) {
  34. d := s.AddDaemon(c, true, true)
  35. testStackName := "testdeploy"
  36. stackArgs := []string{
  37. "stack", "deploy",
  38. "--compose-file", "fixtures/deploy/default.yaml",
  39. testStackName,
  40. }
  41. out, err := d.Cmd(stackArgs...)
  42. c.Assert(err, checker.IsNil, check.Commentf(out))
  43. out, err = d.Cmd("stack", "ls")
  44. c.Assert(err, checker.IsNil)
  45. c.Assert(out, check.Equals, "NAME SERVICES\n"+"testdeploy 2\n")
  46. out, err = d.Cmd("stack", "rm", testStackName)
  47. c.Assert(err, checker.IsNil)
  48. out, err = d.Cmd("stack", "ls")
  49. c.Assert(err, checker.IsNil)
  50. c.Assert(out, check.Equals, "NAME SERVICES\n")
  51. }
  52. func (s *DockerSwarmSuite) TestStackDeployWithSecretsTwice(c *check.C) {
  53. d := s.AddDaemon(c, true, true)
  54. out, err := d.Cmd("secret", "create", "outside", "fixtures/secrets/default")
  55. c.Assert(err, checker.IsNil, check.Commentf(out))
  56. testStackName := "testdeploy"
  57. stackArgs := []string{
  58. "stack", "deploy",
  59. "--compose-file", "fixtures/deploy/secrets.yaml",
  60. testStackName,
  61. }
  62. out, err = d.Cmd(stackArgs...)
  63. c.Assert(err, checker.IsNil, check.Commentf(out))
  64. out, err = d.Cmd("service", "inspect", "--format", "{{ json .Spec.TaskTemplate.ContainerSpec.Secrets }}", "testdeploy_web")
  65. c.Assert(err, checker.IsNil)
  66. var refs []swarm.SecretReference
  67. c.Assert(json.Unmarshal([]byte(out), &refs), checker.IsNil)
  68. c.Assert(refs, checker.HasLen, 3)
  69. sort.Sort(sortSecrets(refs))
  70. c.Assert(refs[0].SecretName, checker.Equals, "outside")
  71. c.Assert(refs[1].SecretName, checker.Equals, "testdeploy_special")
  72. c.Assert(refs[1].File.Name, checker.Equals, "special")
  73. c.Assert(refs[2].SecretName, checker.Equals, "testdeploy_super")
  74. c.Assert(refs[2].File.Name, checker.Equals, "foo.txt")
  75. c.Assert(refs[2].File.Mode, checker.Equals, os.FileMode(0400))
  76. // Deploy again to ensure there are no errors when secret hasn't changed
  77. out, err = d.Cmd(stackArgs...)
  78. c.Assert(err, checker.IsNil, check.Commentf(out))
  79. }
  80. func (s *DockerSwarmSuite) TestStackRemove(c *check.C) {
  81. d := s.AddDaemon(c, true, true)
  82. stackName := "testdeploy"
  83. stackArgs := []string{
  84. "stack", "deploy",
  85. "--compose-file", "fixtures/deploy/remove.yaml",
  86. stackName,
  87. }
  88. out, err := d.Cmd(stackArgs...)
  89. c.Assert(err, checker.IsNil, check.Commentf(out))
  90. out, err = d.Cmd("stack", "ps", stackName)
  91. c.Assert(err, checker.IsNil)
  92. c.Assert(strings.Split(strings.TrimSpace(out), "\n"), checker.HasLen, 2)
  93. out, err = d.Cmd("stack", "rm", stackName)
  94. c.Assert(err, checker.IsNil, check.Commentf(out))
  95. c.Assert(out, checker.Contains, "Removing service testdeploy_web")
  96. c.Assert(out, checker.Contains, "Removing network testdeploy_default")
  97. c.Assert(out, checker.Contains, "Removing secret testdeploy_special")
  98. }
  99. type sortSecrets []swarm.SecretReference
  100. func (s sortSecrets) Len() int { return len(s) }
  101. func (s sortSecrets) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
  102. func (s sortSecrets) Less(i, j int) bool { return s[i].SecretName < s[j].SecretName }
  103. // testDAB is the DAB JSON used for testing.
  104. // TODO: Use template/text and substitute "Image" with the result of
  105. // `docker inspect --format '{{index .RepoDigests 0}}' busybox:latest`
  106. const testDAB = `{
  107. "Version": "0.1",
  108. "Services": {
  109. "srv1": {
  110. "Image": "busybox@sha256:e4f93f6ed15a0cdd342f5aae387886fba0ab98af0a102da6276eaf24d6e6ade0",
  111. "Command": ["top"]
  112. },
  113. "srv2": {
  114. "Image": "busybox@sha256:e4f93f6ed15a0cdd342f5aae387886fba0ab98af0a102da6276eaf24d6e6ade0",
  115. "Command": ["tail"],
  116. "Args": ["-f", "/dev/null"]
  117. }
  118. }
  119. }`
  120. func (s *DockerSwarmSuite) TestStackDeployWithDAB(c *check.C) {
  121. testRequires(c, ExperimentalDaemon)
  122. // setup
  123. testStackName := "test"
  124. testDABFileName := testStackName + ".dab"
  125. defer os.RemoveAll(testDABFileName)
  126. err := ioutil.WriteFile(testDABFileName, []byte(testDAB), 0444)
  127. c.Assert(err, checker.IsNil)
  128. d := s.AddDaemon(c, true, true)
  129. // deploy
  130. stackArgs := []string{
  131. "stack", "deploy",
  132. "--bundle-file", testDABFileName,
  133. testStackName,
  134. }
  135. out, err := d.Cmd(stackArgs...)
  136. c.Assert(err, checker.IsNil)
  137. c.Assert(out, checker.Contains, "Loading bundle from test.dab\n")
  138. c.Assert(out, checker.Contains, "Creating service test_srv1\n")
  139. c.Assert(out, checker.Contains, "Creating service test_srv2\n")
  140. // ls
  141. stackArgs = []string{"stack", "ls"}
  142. out, err = d.Cmd(stackArgs...)
  143. c.Assert(err, checker.IsNil)
  144. c.Assert(out, check.Equals, "NAME SERVICES\n"+"test 2\n")
  145. // rm
  146. stackArgs = []string{"stack", "rm", testStackName}
  147. out, err = d.Cmd(stackArgs...)
  148. c.Assert(err, checker.IsNil)
  149. c.Assert(out, checker.Contains, "Removing service test_srv1\n")
  150. c.Assert(out, checker.Contains, "Removing service test_srv2\n")
  151. // ls (empty)
  152. stackArgs = []string{"stack", "ls"}
  153. out, err = d.Cmd(stackArgs...)
  154. c.Assert(err, checker.IsNil)
  155. c.Assert(out, check.Equals, "NAME SERVICES\n")
  156. }