docker_api_swarm_test.go 20 KB

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