docker_api_swarm_test.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769
  1. // +build !windows
  2. package main
  3. import (
  4. "fmt"
  5. "net/http"
  6. "strconv"
  7. "strings"
  8. "syscall"
  9. "time"
  10. "github.com/docker/docker/pkg/integration/checker"
  11. "github.com/docker/engine-api/types/swarm"
  12. "github.com/go-check/check"
  13. )
  14. var defaultReconciliationTimeout = 30 * time.Second
  15. func (s *DockerSwarmSuite) TestApiSwarmInit(c *check.C) {
  16. testRequires(c, Network)
  17. // todo: should find a better way to verify that components are running than /info
  18. d1 := s.AddDaemon(c, true, true)
  19. info, err := d1.info()
  20. c.Assert(err, checker.IsNil)
  21. c.Assert(info.ControlAvailable, checker.True)
  22. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  23. d2 := s.AddDaemon(c, true, false)
  24. info, err = d2.info()
  25. c.Assert(err, checker.IsNil)
  26. c.Assert(info.ControlAvailable, checker.False)
  27. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  28. // Leaving cluster
  29. c.Assert(d2.Leave(false), checker.IsNil)
  30. info, err = d2.info()
  31. c.Assert(err, checker.IsNil)
  32. c.Assert(info.ControlAvailable, checker.False)
  33. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  34. c.Assert(d2.Join(swarm.JoinRequest{RemoteAddrs: []string{d1.listenAddr}}), checker.IsNil)
  35. info, err = d2.info()
  36. c.Assert(err, checker.IsNil)
  37. c.Assert(info.ControlAvailable, checker.False)
  38. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  39. // Current state restoring after restarts
  40. err = d1.Stop()
  41. c.Assert(err, checker.IsNil)
  42. err = d2.Stop()
  43. c.Assert(err, checker.IsNil)
  44. err = d1.Start()
  45. c.Assert(err, checker.IsNil)
  46. err = d2.Start()
  47. c.Assert(err, checker.IsNil)
  48. info, err = d1.info()
  49. c.Assert(err, checker.IsNil)
  50. c.Assert(info.ControlAvailable, checker.True)
  51. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  52. info, err = d2.info()
  53. c.Assert(err, checker.IsNil)
  54. c.Assert(info.ControlAvailable, checker.False)
  55. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  56. }
  57. func (s *DockerSwarmSuite) TestApiSwarmManualAcceptance(c *check.C) {
  58. testRequires(c, Network)
  59. s.testAPISwarmManualAcceptance(c, "")
  60. }
  61. func (s *DockerSwarmSuite) TestApiSwarmManualAcceptanceSecret(c *check.C) {
  62. testRequires(c, Network)
  63. s.testAPISwarmManualAcceptance(c, "foobaz")
  64. }
  65. func (s *DockerSwarmSuite) testAPISwarmManualAcceptance(c *check.C, secret string) {
  66. d1 := s.AddDaemon(c, false, false)
  67. c.Assert(d1.Init(swarm.InitRequest{
  68. Spec: swarm.Spec{
  69. AcceptancePolicy: swarm.AcceptancePolicy{
  70. Policies: []swarm.Policy{
  71. {Role: swarm.NodeRoleWorker, Secret: &secret},
  72. {Role: swarm.NodeRoleManager, Secret: &secret},
  73. },
  74. },
  75. },
  76. }), checker.IsNil)
  77. d2 := s.AddDaemon(c, false, false)
  78. err := d2.Join(swarm.JoinRequest{RemoteAddrs: []string{d1.listenAddr}})
  79. c.Assert(err, checker.NotNil)
  80. if secret == "" {
  81. c.Assert(err.Error(), checker.Contains, "needs to be accepted")
  82. info, err := d2.info()
  83. c.Assert(err, checker.IsNil)
  84. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStatePending)
  85. c.Assert(d2.Leave(false), checker.IsNil)
  86. info, err = d2.info()
  87. c.Assert(err, checker.IsNil)
  88. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  89. } else {
  90. c.Assert(err.Error(), checker.Contains, "valid secret token is necessary")
  91. info, err := d2.info()
  92. c.Assert(err, checker.IsNil)
  93. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  94. }
  95. d3 := s.AddDaemon(c, false, false)
  96. c.Assert(d3.Join(swarm.JoinRequest{Secret: secret, RemoteAddrs: []string{d1.listenAddr}}), checker.NotNil)
  97. info, err := d3.info()
  98. c.Assert(err, checker.IsNil)
  99. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStatePending)
  100. c.Assert(len(info.NodeID), checker.GreaterThan, 5)
  101. d1.updateNode(c, info.NodeID, func(n *swarm.Node) {
  102. n.Spec.Membership = swarm.NodeMembershipAccepted
  103. })
  104. waitAndAssert(c, defaultReconciliationTimeout, d3.checkLocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  105. }
  106. func (s *DockerSwarmSuite) TestApiSwarmSecretAcceptance(c *check.C) {
  107. testRequires(c, Network)
  108. d1 := s.AddDaemon(c, false, false)
  109. secret := "foobar"
  110. c.Assert(d1.Init(swarm.InitRequest{
  111. Spec: swarm.Spec{
  112. AcceptancePolicy: swarm.AcceptancePolicy{
  113. Policies: []swarm.Policy{
  114. {Role: swarm.NodeRoleWorker, Autoaccept: true, Secret: &secret},
  115. {Role: swarm.NodeRoleManager, Secret: &secret},
  116. },
  117. },
  118. },
  119. }), checker.IsNil)
  120. d2 := s.AddDaemon(c, false, false)
  121. err := d2.Join(swarm.JoinRequest{RemoteAddrs: []string{d1.listenAddr}})
  122. c.Assert(err, checker.NotNil)
  123. c.Assert(err.Error(), checker.Contains, "secret token is necessary")
  124. info, err := d2.info()
  125. c.Assert(err, checker.IsNil)
  126. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  127. err = d2.Join(swarm.JoinRequest{Secret: "foobaz", RemoteAddrs: []string{d1.listenAddr}})
  128. c.Assert(err, checker.NotNil)
  129. c.Assert(err.Error(), checker.Contains, "secret token is necessary")
  130. info, err = d2.info()
  131. c.Assert(err, checker.IsNil)
  132. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  133. c.Assert(d2.Join(swarm.JoinRequest{Secret: "foobar", RemoteAddrs: []string{d1.listenAddr}}), checker.IsNil)
  134. info, err = d2.info()
  135. c.Assert(err, checker.IsNil)
  136. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  137. c.Assert(d2.Leave(false), checker.IsNil)
  138. info, err = d2.info()
  139. c.Assert(err, checker.IsNil)
  140. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  141. // change secret
  142. d1.updateSwarm(c, func(s *swarm.Spec) {
  143. for i := range s.AcceptancePolicy.Policies {
  144. p := "foobaz"
  145. s.AcceptancePolicy.Policies[i].Secret = &p
  146. }
  147. })
  148. err = d2.Join(swarm.JoinRequest{Secret: "foobar", RemoteAddrs: []string{d1.listenAddr}})
  149. c.Assert(err, checker.NotNil)
  150. c.Assert(err.Error(), checker.Contains, "secret token is necessary")
  151. info, err = d2.info()
  152. c.Assert(err, checker.IsNil)
  153. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  154. c.Assert(d2.Join(swarm.JoinRequest{Secret: "foobaz", RemoteAddrs: []string{d1.listenAddr}}), checker.IsNil)
  155. info, err = d2.info()
  156. c.Assert(err, checker.IsNil)
  157. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  158. c.Assert(d2.Leave(false), checker.IsNil)
  159. info, err = d2.info()
  160. c.Assert(err, checker.IsNil)
  161. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  162. // change policy, don't change secret
  163. d1.updateSwarm(c, func(s *swarm.Spec) {
  164. for i, p := range s.AcceptancePolicy.Policies {
  165. if p.Role == swarm.NodeRoleManager {
  166. s.AcceptancePolicy.Policies[i].Autoaccept = false
  167. }
  168. s.AcceptancePolicy.Policies[i].Secret = nil
  169. }
  170. })
  171. err = d2.Join(swarm.JoinRequest{RemoteAddrs: []string{d1.listenAddr}})
  172. c.Assert(err, checker.NotNil)
  173. c.Assert(err.Error(), checker.Contains, "secret token is necessary")
  174. info, err = d2.info()
  175. c.Assert(err, checker.IsNil)
  176. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  177. c.Assert(d2.Join(swarm.JoinRequest{Secret: "foobaz", RemoteAddrs: []string{d1.listenAddr}}), checker.IsNil)
  178. info, err = d2.info()
  179. c.Assert(err, checker.IsNil)
  180. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  181. c.Assert(d2.Leave(false), checker.IsNil)
  182. info, err = d2.info()
  183. c.Assert(err, checker.IsNil)
  184. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  185. // clear secret
  186. d1.updateSwarm(c, func(s *swarm.Spec) {
  187. for i := range s.AcceptancePolicy.Policies {
  188. p := ""
  189. s.AcceptancePolicy.Policies[i].Secret = &p
  190. }
  191. })
  192. c.Assert(d2.Join(swarm.JoinRequest{RemoteAddrs: []string{d1.listenAddr}}), checker.IsNil)
  193. info, err = d2.info()
  194. c.Assert(err, checker.IsNil)
  195. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  196. c.Assert(d2.Leave(false), checker.IsNil)
  197. info, err = d2.info()
  198. c.Assert(err, checker.IsNil)
  199. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  200. }
  201. func (s *DockerSwarmSuite) TestApiSwarmCAHash(c *check.C) {
  202. testRequires(c, Network)
  203. d1 := s.AddDaemon(c, true, true)
  204. d2 := s.AddDaemon(c, false, false)
  205. err := d2.Join(swarm.JoinRequest{CACertHash: "foobar", RemoteAddrs: []string{d1.listenAddr}})
  206. c.Assert(err, checker.NotNil)
  207. c.Assert(err.Error(), checker.Contains, "invalid checksum digest format")
  208. c.Assert(len(d1.CACertHash), checker.GreaterThan, 0)
  209. c.Assert(d2.Join(swarm.JoinRequest{CACertHash: d1.CACertHash, RemoteAddrs: []string{d1.listenAddr}}), checker.IsNil)
  210. }
  211. func (s *DockerSwarmSuite) TestApiSwarmPromoteDemote(c *check.C) {
  212. testRequires(c, Network)
  213. d1 := s.AddDaemon(c, false, false)
  214. c.Assert(d1.Init(swarm.InitRequest{
  215. Spec: swarm.Spec{
  216. AcceptancePolicy: swarm.AcceptancePolicy{
  217. Policies: []swarm.Policy{
  218. {Role: swarm.NodeRoleWorker, Autoaccept: true},
  219. {Role: swarm.NodeRoleManager},
  220. },
  221. },
  222. },
  223. }), checker.IsNil)
  224. d2 := s.AddDaemon(c, true, false)
  225. info, err := d2.info()
  226. c.Assert(err, checker.IsNil)
  227. c.Assert(info.ControlAvailable, checker.False)
  228. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  229. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  230. n.Spec.Role = swarm.NodeRoleManager
  231. })
  232. waitAndAssert(c, defaultReconciliationTimeout, d2.checkControlAvailable, checker.True)
  233. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  234. n.Spec.Role = swarm.NodeRoleWorker
  235. })
  236. waitAndAssert(c, defaultReconciliationTimeout, d2.checkControlAvailable, checker.False)
  237. // Demoting last node should fail
  238. node := d1.getNode(c, d1.NodeID)
  239. node.Spec.Role = swarm.NodeRoleWorker
  240. url := fmt.Sprintf("/nodes/%s/update?version=%d", node.ID, node.Version.Index)
  241. status, out, err := d1.SockRequest("POST", url, node.Spec)
  242. c.Assert(err, checker.IsNil)
  243. c.Assert(status, checker.Equals, http.StatusInternalServerError, check.Commentf("output: %q", string(out)))
  244. c.Assert(string(out), checker.Contains, "last manager of the swarm")
  245. info, err = d1.info()
  246. c.Assert(err, checker.IsNil)
  247. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  248. c.Assert(info.ControlAvailable, checker.True)
  249. // Promote already demoted node
  250. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  251. n.Spec.Role = swarm.NodeRoleManager
  252. })
  253. waitAndAssert(c, defaultReconciliationTimeout, d2.checkControlAvailable, checker.True)
  254. }
  255. func (s *DockerSwarmSuite) TestApiSwarmServicesEmptyList(c *check.C) {
  256. testRequires(c, Network)
  257. d := s.AddDaemon(c, true, true)
  258. services := d.listServices(c)
  259. c.Assert(services, checker.NotNil)
  260. c.Assert(len(services), checker.Equals, 0, check.Commentf("services: %#v", services))
  261. }
  262. func (s *DockerSwarmSuite) TestApiSwarmServicesCreate(c *check.C) {
  263. testRequires(c, Network)
  264. d := s.AddDaemon(c, true, true)
  265. instances := 2
  266. id := d.createService(c, simpleTestService, setInstances(instances))
  267. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, instances)
  268. service := d.getService(c, id)
  269. instances = 5
  270. d.updateService(c, service, setInstances(instances))
  271. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, instances)
  272. d.removeService(c, service.ID)
  273. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, 0)
  274. }
  275. func (s *DockerSwarmSuite) TestApiSwarmServicesMultipleAgents(c *check.C) {
  276. testRequires(c, Network)
  277. d1 := s.AddDaemon(c, true, true)
  278. d2 := s.AddDaemon(c, true, false)
  279. d3 := s.AddDaemon(c, true, false)
  280. time.Sleep(1 * time.Second) // make sure all daemons are ready to accept tasks
  281. instances := 9
  282. id := d1.createService(c, simpleTestService, setInstances(instances))
  283. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.GreaterThan, 0)
  284. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.GreaterThan, 0)
  285. waitAndAssert(c, defaultReconciliationTimeout, d3.checkActiveContainerCount, checker.GreaterThan, 0)
  286. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  287. // reconciliation on d2 node down
  288. c.Assert(d2.Stop(), checker.IsNil)
  289. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  290. // test downscaling
  291. instances = 5
  292. d1.updateService(c, d1.getService(c, id), setInstances(instances))
  293. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  294. }
  295. func (s *DockerSwarmSuite) TestApiSwarmServicesCreateGlobal(c *check.C) {
  296. testRequires(c, Network)
  297. d1 := s.AddDaemon(c, true, true)
  298. d2 := s.AddDaemon(c, true, false)
  299. d3 := s.AddDaemon(c, true, false)
  300. d1.createService(c, simpleTestService, setGlobalMode)
  301. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.Equals, 1)
  302. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.Equals, 1)
  303. waitAndAssert(c, defaultReconciliationTimeout, d3.checkActiveContainerCount, checker.Equals, 1)
  304. d4 := s.AddDaemon(c, true, false)
  305. d5 := s.AddDaemon(c, true, false)
  306. waitAndAssert(c, defaultReconciliationTimeout, d4.checkActiveContainerCount, checker.Equals, 1)
  307. waitAndAssert(c, defaultReconciliationTimeout, d5.checkActiveContainerCount, checker.Equals, 1)
  308. }
  309. func (s *DockerSwarmSuite) TestApiSwarmServicesStateReporting(c *check.C) {
  310. testRequires(c, Network)
  311. testRequires(c, SameHostDaemon)
  312. testRequires(c, DaemonIsLinux)
  313. d1 := s.AddDaemon(c, true, true)
  314. d2 := s.AddDaemon(c, true, true)
  315. d3 := s.AddDaemon(c, true, false)
  316. time.Sleep(1 * time.Second) // make sure all daemons are ready to accept
  317. instances := 9
  318. d1.createService(c, simpleTestService, setInstances(instances))
  319. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  320. getContainers := func() map[string]*SwarmDaemon {
  321. m := make(map[string]*SwarmDaemon)
  322. for _, d := range []*SwarmDaemon{d1, d2, d3} {
  323. for _, id := range d.activeContainers() {
  324. m[id] = d
  325. }
  326. }
  327. return m
  328. }
  329. containers := getContainers()
  330. c.Assert(containers, checker.HasLen, instances)
  331. var toRemove string
  332. for i := range containers {
  333. toRemove = i
  334. }
  335. _, err := containers[toRemove].Cmd("stop", toRemove)
  336. c.Assert(err, checker.IsNil)
  337. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  338. containers2 := getContainers()
  339. c.Assert(containers2, checker.HasLen, instances)
  340. for i := range containers {
  341. if i == toRemove {
  342. c.Assert(containers2[i], checker.IsNil)
  343. } else {
  344. c.Assert(containers2[i], checker.NotNil)
  345. }
  346. }
  347. containers = containers2
  348. for i := range containers {
  349. toRemove = i
  350. }
  351. // try with killing process outside of docker
  352. pidStr, err := containers[toRemove].Cmd("inspect", "-f", "{{.State.Pid}}", toRemove)
  353. c.Assert(err, checker.IsNil)
  354. pid, err := strconv.Atoi(strings.TrimSpace(pidStr))
  355. c.Assert(err, checker.IsNil)
  356. c.Assert(syscall.Kill(pid, syscall.SIGKILL), checker.IsNil)
  357. time.Sleep(time.Second) // give some time to handle the signal
  358. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  359. containers2 = getContainers()
  360. c.Assert(containers2, checker.HasLen, instances)
  361. for i := range containers {
  362. if i == toRemove {
  363. c.Assert(containers2[i], checker.IsNil)
  364. } else {
  365. c.Assert(containers2[i], checker.NotNil)
  366. }
  367. }
  368. }
  369. func (s *DockerSwarmSuite) TestApiSwarmRaftQuorum(c *check.C) {
  370. testRequires(c, Network)
  371. d1 := s.AddDaemon(c, true, true)
  372. d2 := s.AddDaemon(c, true, true)
  373. d3 := s.AddDaemon(c, true, true)
  374. d1.createService(c, simpleTestService)
  375. c.Assert(d2.Stop(), checker.IsNil)
  376. d1.createService(c, simpleTestService, func(s *swarm.Service) {
  377. s.Spec.Name = "top1"
  378. })
  379. c.Assert(d3.Stop(), checker.IsNil)
  380. var service swarm.Service
  381. simpleTestService(&service)
  382. service.Spec.Name = "top2"
  383. status, out, err := d1.SockRequest("POST", "/services/create", service.Spec)
  384. c.Assert(err, checker.IsNil)
  385. c.Assert(status, checker.Equals, http.StatusInternalServerError, check.Commentf("deadline exceeded", string(out)))
  386. c.Assert(d2.Start(), checker.IsNil)
  387. d1.createService(c, simpleTestService, func(s *swarm.Service) {
  388. s.Spec.Name = "top3"
  389. })
  390. }
  391. func (s *DockerSwarmSuite) TestApiSwarmListNodes(c *check.C) {
  392. testRequires(c, Network)
  393. d1 := s.AddDaemon(c, true, true)
  394. d2 := s.AddDaemon(c, true, false)
  395. d3 := s.AddDaemon(c, true, false)
  396. nodes := d1.listNodes(c)
  397. c.Assert(len(nodes), checker.Equals, 3, check.Commentf("nodes: %#v", nodes))
  398. loop0:
  399. for _, n := range nodes {
  400. for _, d := range []*SwarmDaemon{d1, d2, d3} {
  401. if n.ID == d.NodeID {
  402. continue loop0
  403. }
  404. }
  405. c.Errorf("unknown nodeID %v", n.ID)
  406. }
  407. }
  408. func (s *DockerSwarmSuite) TestApiSwarmNodeUpdate(c *check.C) {
  409. testRequires(c, Network)
  410. d := s.AddDaemon(c, true, true)
  411. nodes := d.listNodes(c)
  412. d.updateNode(c, nodes[0].ID, func(n *swarm.Node) {
  413. n.Spec.Availability = swarm.NodeAvailabilityPause
  414. })
  415. n := d.getNode(c, nodes[0].ID)
  416. c.Assert(n.Spec.Availability, checker.Equals, swarm.NodeAvailabilityPause)
  417. }
  418. func (s *DockerSwarmSuite) TestApiSwarmNodeDrainPause(c *check.C) {
  419. testRequires(c, Network)
  420. d1 := s.AddDaemon(c, true, true)
  421. d2 := s.AddDaemon(c, true, false)
  422. time.Sleep(1 * time.Second) // make sure all daemons are ready to accept tasks
  423. // start a service, expect balanced distribution
  424. instances := 8
  425. id := d1.createService(c, simpleTestService, setInstances(instances))
  426. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.GreaterThan, 0)
  427. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.GreaterThan, 0)
  428. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount), checker.Equals, instances)
  429. // drain d2, all containers should move to d1
  430. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  431. n.Spec.Availability = swarm.NodeAvailabilityDrain
  432. })
  433. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.Equals, instances)
  434. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.Equals, 0)
  435. // set d2 back to active
  436. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  437. n.Spec.Availability = swarm.NodeAvailabilityActive
  438. })
  439. instances = 1
  440. d1.updateService(c, d1.getService(c, id), setInstances(instances))
  441. waitAndAssert(c, defaultReconciliationTimeout*2, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount), checker.Equals, instances)
  442. instances = 8
  443. d1.updateService(c, d1.getService(c, id), setInstances(instances))
  444. // drained node first so we don't get any old containers
  445. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.GreaterThan, 0)
  446. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.GreaterThan, 0)
  447. waitAndAssert(c, defaultReconciliationTimeout*2, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount), checker.Equals, instances)
  448. d2ContainerCount := len(d2.activeContainers())
  449. // set d2 to paused, scale service up, only d1 gets new tasks
  450. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  451. n.Spec.Availability = swarm.NodeAvailabilityPause
  452. })
  453. instances = 14
  454. d1.updateService(c, d1.getService(c, id), setInstances(instances))
  455. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.Equals, instances-d2ContainerCount)
  456. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.Equals, d2ContainerCount)
  457. }
  458. func (s *DockerSwarmSuite) TestApiSwarmLeaveRemovesContainer(c *check.C) {
  459. testRequires(c, Network)
  460. d := s.AddDaemon(c, true, true)
  461. instances := 2
  462. d.createService(c, simpleTestService, setInstances(instances))
  463. id, err := d.Cmd("run", "-d", "busybox", "top")
  464. c.Assert(err, checker.IsNil)
  465. id = strings.TrimSpace(id)
  466. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, instances+1)
  467. c.Assert(d.Leave(false), checker.NotNil)
  468. c.Assert(d.Leave(true), checker.IsNil)
  469. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, 1)
  470. id2, err := d.Cmd("ps", "-q")
  471. c.Assert(err, checker.IsNil)
  472. c.Assert(id, checker.HasPrefix, strings.TrimSpace(id2))
  473. }
  474. // #23629
  475. func (s *DockerSwarmSuite) TestApiSwarmLeaveOnPendingJoin(c *check.C) {
  476. s.AddDaemon(c, true, true)
  477. d2 := s.AddDaemon(c, false, false)
  478. id, err := d2.Cmd("run", "-d", "busybox", "top")
  479. c.Assert(err, checker.IsNil)
  480. id = strings.TrimSpace(id)
  481. go d2.Join(swarm.JoinRequest{
  482. RemoteAddrs: []string{"nosuchhost:1234"},
  483. }) // will block on pending state
  484. waitAndAssert(c, defaultReconciliationTimeout, d2.checkLocalNodeState, checker.Equals, swarm.LocalNodeStatePending)
  485. c.Assert(d2.Leave(true), checker.IsNil)
  486. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.Equals, 1)
  487. id2, err := d2.Cmd("ps", "-q")
  488. c.Assert(err, checker.IsNil)
  489. c.Assert(id, checker.HasPrefix, strings.TrimSpace(id2))
  490. }
  491. // #23705
  492. func (s *DockerSwarmSuite) TestApiSwarmRestoreOnPendingJoin(c *check.C) {
  493. d := s.AddDaemon(c, false, false)
  494. go d.Join(swarm.JoinRequest{
  495. RemoteAddrs: []string{"nosuchhost:1234"},
  496. }) // will block on pending state
  497. waitAndAssert(c, defaultReconciliationTimeout, d.checkLocalNodeState, checker.Equals, swarm.LocalNodeStatePending)
  498. c.Assert(d.Stop(), checker.IsNil)
  499. c.Assert(d.Start(), checker.IsNil)
  500. info, err := d.info()
  501. c.Assert(err, checker.IsNil)
  502. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  503. }
  504. func (s *DockerSwarmSuite) TestApiSwarmManagerRestore(c *check.C) {
  505. testRequires(c, Network)
  506. d1 := s.AddDaemon(c, true, true)
  507. instances := 2
  508. id := d1.createService(c, simpleTestService, setInstances(instances))
  509. d1.getService(c, id)
  510. d1.Stop()
  511. d1.Start()
  512. d1.getService(c, id)
  513. d2 := s.AddDaemon(c, true, true)
  514. d2.getService(c, id)
  515. d2.Stop()
  516. d2.Start()
  517. d2.getService(c, id)
  518. d3 := s.AddDaemon(c, true, true)
  519. d3.getService(c, id)
  520. d3.Stop()
  521. d3.Start()
  522. d3.getService(c, id)
  523. d3.Kill()
  524. time.Sleep(1 * time.Second) // time to handle signal
  525. d3.Start()
  526. d3.getService(c, id)
  527. }
  528. func (s *DockerSwarmSuite) TestApiSwarmScaleNoRollingUpdate(c *check.C) {
  529. testRequires(c, Network)
  530. d := s.AddDaemon(c, true, true)
  531. instances := 2
  532. id := d.createService(c, simpleTestService, setInstances(instances))
  533. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, instances)
  534. containers := d.activeContainers()
  535. instances = 4
  536. d.updateService(c, d.getService(c, id), setInstances(instances))
  537. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, instances)
  538. containers2 := d.activeContainers()
  539. loop0:
  540. for _, c1 := range containers {
  541. for _, c2 := range containers2 {
  542. if c1 == c2 {
  543. continue loop0
  544. }
  545. }
  546. c.Errorf("container %v not found in new set %#v", c1, containers2)
  547. }
  548. }
  549. func (s *DockerSwarmSuite) TestApiSwarmInvalidAddress(c *check.C) {
  550. d := s.AddDaemon(c, false, false)
  551. req := swarm.InitRequest{
  552. ListenAddr: "",
  553. }
  554. status, _, err := d.SockRequest("POST", "/swarm/init", req)
  555. c.Assert(err, checker.IsNil)
  556. c.Assert(status, checker.Equals, http.StatusInternalServerError)
  557. req2 := swarm.JoinRequest{
  558. ListenAddr: "0.0.0.0:2377",
  559. RemoteAddrs: []string{""},
  560. }
  561. status, _, err = d.SockRequest("POST", "/swarm/join", req2)
  562. c.Assert(err, checker.IsNil)
  563. c.Assert(status, checker.Equals, http.StatusInternalServerError)
  564. }
  565. func (s *DockerSwarmSuite) TestApiSwarmForceNewCluster(c *check.C) {
  566. d1 := s.AddDaemon(c, true, true)
  567. d2 := s.AddDaemon(c, true, true)
  568. instances := 2
  569. id := d1.createService(c, simpleTestService, setInstances(instances))
  570. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount), checker.Equals, instances)
  571. c.Assert(d2.Stop(), checker.IsNil)
  572. time.Sleep(5 * time.Second)
  573. c.Assert(d1.Init(swarm.InitRequest{
  574. ForceNewCluster: true,
  575. Spec: swarm.Spec{
  576. AcceptancePolicy: autoAcceptPolicy,
  577. },
  578. }), checker.IsNil)
  579. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.Equals, instances)
  580. d3 := s.AddDaemon(c, true, true)
  581. info, err := d3.info()
  582. c.Assert(err, checker.IsNil)
  583. c.Assert(info.ControlAvailable, checker.True)
  584. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  585. instances = 4
  586. d3.updateService(c, d3.getService(c, id), setInstances(instances))
  587. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  588. }
  589. func simpleTestService(s *swarm.Service) {
  590. var ureplicas uint64
  591. ureplicas = 1
  592. s.Spec = swarm.ServiceSpec{
  593. TaskTemplate: swarm.TaskSpec{
  594. ContainerSpec: swarm.ContainerSpec{
  595. Image: "busybox:latest",
  596. Command: []string{"/bin/top"},
  597. },
  598. },
  599. Mode: swarm.ServiceMode{
  600. Replicated: &swarm.ReplicatedService{
  601. Replicas: &ureplicas,
  602. },
  603. },
  604. }
  605. s.Spec.Name = "top"
  606. }
  607. func setInstances(replicas int) serviceConstructor {
  608. ureplicas := uint64(replicas)
  609. return func(s *swarm.Service) {
  610. s.Spec.Mode = swarm.ServiceMode{
  611. Replicated: &swarm.ReplicatedService{
  612. Replicas: &ureplicas,
  613. },
  614. }
  615. }
  616. }
  617. func setGlobalMode(s *swarm.Service) {
  618. s.Spec.Mode = swarm.ServiceMode{
  619. Global: &swarm.GlobalService{},
  620. }
  621. }