docker_api_swarm_test.go 22 KB

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