docker_api_swarm_test.go 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902
  1. // +build !windows
  2. package main
  3. import (
  4. "fmt"
  5. "net/http"
  6. "os"
  7. "path/filepath"
  8. "strconv"
  9. "strings"
  10. "sync"
  11. "syscall"
  12. "time"
  13. "github.com/docker/docker/pkg/integration/checker"
  14. "github.com/docker/engine-api/types/swarm"
  15. "github.com/go-check/check"
  16. )
  17. var defaultReconciliationTimeout = 30 * time.Second
  18. func (s *DockerSwarmSuite) TestApiSwarmInit(c *check.C) {
  19. testRequires(c, Network)
  20. // todo: should find a better way to verify that components are running than /info
  21. d1 := s.AddDaemon(c, true, true)
  22. info, err := d1.info()
  23. c.Assert(err, checker.IsNil)
  24. c.Assert(info.ControlAvailable, checker.True)
  25. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  26. d2 := s.AddDaemon(c, true, false)
  27. info, err = d2.info()
  28. c.Assert(err, checker.IsNil)
  29. c.Assert(info.ControlAvailable, checker.False)
  30. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  31. // Leaving cluster
  32. c.Assert(d2.Leave(false), checker.IsNil)
  33. info, err = d2.info()
  34. c.Assert(err, checker.IsNil)
  35. c.Assert(info.ControlAvailable, checker.False)
  36. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  37. c.Assert(d2.Join(swarm.JoinRequest{JoinToken: d1.joinTokens(c).Worker, RemoteAddrs: []string{d1.listenAddr}}), checker.IsNil)
  38. info, err = d2.info()
  39. c.Assert(err, checker.IsNil)
  40. c.Assert(info.ControlAvailable, checker.False)
  41. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  42. // Current state restoring after restarts
  43. err = d1.Stop()
  44. c.Assert(err, checker.IsNil)
  45. err = d2.Stop()
  46. c.Assert(err, checker.IsNil)
  47. err = d1.Start()
  48. c.Assert(err, checker.IsNil)
  49. err = d2.Start()
  50. c.Assert(err, checker.IsNil)
  51. info, err = d1.info()
  52. c.Assert(err, checker.IsNil)
  53. c.Assert(info.ControlAvailable, checker.True)
  54. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  55. info, err = d2.info()
  56. c.Assert(err, checker.IsNil)
  57. c.Assert(info.ControlAvailable, checker.False)
  58. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  59. }
  60. func (s *DockerSwarmSuite) TestApiSwarmJoinToken(c *check.C) {
  61. testRequires(c, Network)
  62. d1 := s.AddDaemon(c, false, false)
  63. c.Assert(d1.Init(swarm.InitRequest{}), checker.IsNil)
  64. d2 := s.AddDaemon(c, false, false)
  65. err := d2.Join(swarm.JoinRequest{RemoteAddrs: []string{d1.listenAddr}})
  66. c.Assert(err, checker.NotNil)
  67. c.Assert(err.Error(), checker.Contains, "join token is necessary")
  68. info, err := d2.info()
  69. c.Assert(err, checker.IsNil)
  70. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  71. err = d2.Join(swarm.JoinRequest{JoinToken: "foobaz", RemoteAddrs: []string{d1.listenAddr}})
  72. c.Assert(err, checker.NotNil)
  73. c.Assert(err.Error(), checker.Contains, "join token is necessary")
  74. info, err = d2.info()
  75. c.Assert(err, checker.IsNil)
  76. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  77. workerToken := d1.joinTokens(c).Worker
  78. c.Assert(d2.Join(swarm.JoinRequest{JoinToken: workerToken, RemoteAddrs: []string{d1.listenAddr}}), checker.IsNil)
  79. info, err = d2.info()
  80. c.Assert(err, checker.IsNil)
  81. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  82. c.Assert(d2.Leave(false), checker.IsNil)
  83. info, err = d2.info()
  84. c.Assert(err, checker.IsNil)
  85. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  86. // change tokens
  87. d1.rotateTokens(c)
  88. err = d2.Join(swarm.JoinRequest{JoinToken: workerToken, RemoteAddrs: []string{d1.listenAddr}})
  89. c.Assert(err, checker.NotNil)
  90. c.Assert(err.Error(), checker.Contains, "join token is necessary")
  91. info, err = d2.info()
  92. c.Assert(err, checker.IsNil)
  93. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  94. workerToken = d1.joinTokens(c).Worker
  95. c.Assert(d2.Join(swarm.JoinRequest{JoinToken: workerToken, RemoteAddrs: []string{d1.listenAddr}}), checker.IsNil)
  96. info, err = d2.info()
  97. c.Assert(err, checker.IsNil)
  98. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  99. c.Assert(d2.Leave(false), checker.IsNil)
  100. info, err = d2.info()
  101. c.Assert(err, checker.IsNil)
  102. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  103. // change spec, don't change tokens
  104. d1.updateSwarm(c, func(s *swarm.Spec) {})
  105. err = d2.Join(swarm.JoinRequest{RemoteAddrs: []string{d1.listenAddr}})
  106. c.Assert(err, checker.NotNil)
  107. c.Assert(err.Error(), checker.Contains, "join token is necessary")
  108. info, err = d2.info()
  109. c.Assert(err, checker.IsNil)
  110. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  111. c.Assert(d2.Join(swarm.JoinRequest{JoinToken: workerToken, RemoteAddrs: []string{d1.listenAddr}}), checker.IsNil)
  112. info, err = d2.info()
  113. c.Assert(err, checker.IsNil)
  114. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  115. c.Assert(d2.Leave(false), checker.IsNil)
  116. info, err = d2.info()
  117. c.Assert(err, checker.IsNil)
  118. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  119. }
  120. func (s *DockerSwarmSuite) TestApiSwarmCAHash(c *check.C) {
  121. testRequires(c, Network)
  122. d1 := s.AddDaemon(c, true, true)
  123. d2 := s.AddDaemon(c, false, false)
  124. splitToken := strings.Split(d1.joinTokens(c).Worker, "-")
  125. splitToken[2] = "1kxftv4ofnc6mt30lmgipg6ngf9luhwqopfk1tz6bdmnkubg0e"
  126. replacementToken := strings.Join(splitToken, "-")
  127. err := d2.Join(swarm.JoinRequest{JoinToken: replacementToken, RemoteAddrs: []string{d1.listenAddr}})
  128. c.Assert(err, checker.NotNil)
  129. c.Assert(err.Error(), checker.Contains, "remote CA does not match fingerprint")
  130. }
  131. func (s *DockerSwarmSuite) TestApiSwarmPromoteDemote(c *check.C) {
  132. testRequires(c, Network)
  133. d1 := s.AddDaemon(c, false, false)
  134. c.Assert(d1.Init(swarm.InitRequest{}), checker.IsNil)
  135. d2 := s.AddDaemon(c, true, false)
  136. info, err := d2.info()
  137. c.Assert(err, checker.IsNil)
  138. c.Assert(info.ControlAvailable, checker.False)
  139. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  140. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  141. n.Spec.Role = swarm.NodeRoleManager
  142. })
  143. waitAndAssert(c, defaultReconciliationTimeout, d2.checkControlAvailable, checker.True)
  144. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  145. n.Spec.Role = swarm.NodeRoleWorker
  146. })
  147. waitAndAssert(c, defaultReconciliationTimeout, d2.checkControlAvailable, checker.False)
  148. // Demoting last node should fail
  149. node := d1.getNode(c, d1.NodeID)
  150. node.Spec.Role = swarm.NodeRoleWorker
  151. url := fmt.Sprintf("/nodes/%s/update?version=%d", node.ID, node.Version.Index)
  152. status, out, err := d1.SockRequest("POST", url, node.Spec)
  153. c.Assert(err, checker.IsNil)
  154. c.Assert(status, checker.Equals, http.StatusInternalServerError, check.Commentf("output: %q", string(out)))
  155. c.Assert(string(out), checker.Contains, "last manager of the swarm")
  156. info, err = d1.info()
  157. c.Assert(err, checker.IsNil)
  158. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  159. c.Assert(info.ControlAvailable, checker.True)
  160. // Promote already demoted node
  161. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  162. n.Spec.Role = swarm.NodeRoleManager
  163. })
  164. waitAndAssert(c, defaultReconciliationTimeout, d2.checkControlAvailable, checker.True)
  165. }
  166. func (s *DockerSwarmSuite) TestApiSwarmServicesEmptyList(c *check.C) {
  167. testRequires(c, Network)
  168. d := s.AddDaemon(c, true, true)
  169. services := d.listServices(c)
  170. c.Assert(services, checker.NotNil)
  171. c.Assert(len(services), checker.Equals, 0, check.Commentf("services: %#v", services))
  172. }
  173. func (s *DockerSwarmSuite) TestApiSwarmServicesCreate(c *check.C) {
  174. testRequires(c, Network)
  175. d := s.AddDaemon(c, true, true)
  176. instances := 2
  177. id := d.createService(c, simpleTestService, setInstances(instances))
  178. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, instances)
  179. service := d.getService(c, id)
  180. instances = 5
  181. d.updateService(c, service, setInstances(instances))
  182. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, instances)
  183. d.removeService(c, service.ID)
  184. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, 0)
  185. }
  186. func (s *DockerSwarmSuite) TestApiSwarmServicesMultipleAgents(c *check.C) {
  187. testRequires(c, Network)
  188. d1 := s.AddDaemon(c, true, true)
  189. d2 := s.AddDaemon(c, true, false)
  190. d3 := s.AddDaemon(c, true, false)
  191. time.Sleep(1 * time.Second) // make sure all daemons are ready to accept tasks
  192. instances := 9
  193. id := d1.createService(c, simpleTestService, setInstances(instances))
  194. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.GreaterThan, 0)
  195. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.GreaterThan, 0)
  196. waitAndAssert(c, defaultReconciliationTimeout, d3.checkActiveContainerCount, checker.GreaterThan, 0)
  197. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  198. // reconciliation on d2 node down
  199. c.Assert(d2.Stop(), checker.IsNil)
  200. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  201. // test downscaling
  202. instances = 5
  203. d1.updateService(c, d1.getService(c, id), setInstances(instances))
  204. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  205. }
  206. func (s *DockerSwarmSuite) TestApiSwarmServicesCreateGlobal(c *check.C) {
  207. testRequires(c, Network)
  208. d1 := s.AddDaemon(c, true, true)
  209. d2 := s.AddDaemon(c, true, false)
  210. d3 := s.AddDaemon(c, true, false)
  211. d1.createService(c, simpleTestService, setGlobalMode)
  212. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.Equals, 1)
  213. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.Equals, 1)
  214. waitAndAssert(c, defaultReconciliationTimeout, d3.checkActiveContainerCount, checker.Equals, 1)
  215. d4 := s.AddDaemon(c, true, false)
  216. d5 := s.AddDaemon(c, true, false)
  217. waitAndAssert(c, defaultReconciliationTimeout, d4.checkActiveContainerCount, checker.Equals, 1)
  218. waitAndAssert(c, defaultReconciliationTimeout, d5.checkActiveContainerCount, checker.Equals, 1)
  219. }
  220. func (s *DockerSwarmSuite) TestApiSwarmServicesUpdate(c *check.C) {
  221. const nodeCount = 3
  222. var daemons [nodeCount]*SwarmDaemon
  223. for i := 0; i < nodeCount; i++ {
  224. daemons[i] = s.AddDaemon(c, true, i == 0)
  225. }
  226. // wait for nodes ready
  227. waitAndAssert(c, 5*time.Second, daemons[0].checkNodeReadyCount, checker.Equals, nodeCount)
  228. // service image at start
  229. image1 := "busybox:latest"
  230. // target image in update
  231. image2 := "busybox:test"
  232. // create a different tag
  233. for _, d := range daemons {
  234. out, err := d.Cmd("tag", image1, image2)
  235. c.Assert(err, checker.IsNil, check.Commentf(out))
  236. }
  237. // create service
  238. instances := 5
  239. parallelism := 2
  240. id := daemons[0].createService(c, serviceForUpdate, setInstances(instances))
  241. // wait for tasks ready
  242. waitAndAssert(c, defaultReconciliationTimeout, daemons[0].checkRunningTaskImages, checker.DeepEquals,
  243. map[string]int{image1: instances})
  244. // issue service update
  245. service := daemons[0].getService(c, id)
  246. daemons[0].updateService(c, service, setImage(image2))
  247. // first batch
  248. waitAndAssert(c, defaultReconciliationTimeout, daemons[0].checkRunningTaskImages, checker.DeepEquals,
  249. map[string]int{image1: instances - parallelism, image2: parallelism})
  250. // 2nd batch
  251. waitAndAssert(c, defaultReconciliationTimeout, daemons[0].checkRunningTaskImages, checker.DeepEquals,
  252. map[string]int{image1: instances - 2*parallelism, image2: 2 * parallelism})
  253. // 3nd batch
  254. waitAndAssert(c, defaultReconciliationTimeout, daemons[0].checkRunningTaskImages, checker.DeepEquals,
  255. map[string]int{image2: instances})
  256. }
  257. func (s *DockerSwarmSuite) TestApiSwarmServicesStateReporting(c *check.C) {
  258. testRequires(c, Network)
  259. testRequires(c, SameHostDaemon)
  260. testRequires(c, DaemonIsLinux)
  261. d1 := s.AddDaemon(c, true, true)
  262. d2 := s.AddDaemon(c, true, true)
  263. d3 := s.AddDaemon(c, true, false)
  264. time.Sleep(1 * time.Second) // make sure all daemons are ready to accept
  265. instances := 9
  266. d1.createService(c, simpleTestService, setInstances(instances))
  267. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  268. getContainers := func() map[string]*SwarmDaemon {
  269. m := make(map[string]*SwarmDaemon)
  270. for _, d := range []*SwarmDaemon{d1, d2, d3} {
  271. for _, id := range d.activeContainers() {
  272. m[id] = d
  273. }
  274. }
  275. return m
  276. }
  277. containers := getContainers()
  278. c.Assert(containers, checker.HasLen, instances)
  279. var toRemove string
  280. for i := range containers {
  281. toRemove = i
  282. }
  283. _, err := containers[toRemove].Cmd("stop", toRemove)
  284. c.Assert(err, checker.IsNil)
  285. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  286. containers2 := getContainers()
  287. c.Assert(containers2, checker.HasLen, instances)
  288. for i := range containers {
  289. if i == toRemove {
  290. c.Assert(containers2[i], checker.IsNil)
  291. } else {
  292. c.Assert(containers2[i], checker.NotNil)
  293. }
  294. }
  295. containers = containers2
  296. for i := range containers {
  297. toRemove = i
  298. }
  299. // try with killing process outside of docker
  300. pidStr, err := containers[toRemove].Cmd("inspect", "-f", "{{.State.Pid}}", toRemove)
  301. c.Assert(err, checker.IsNil)
  302. pid, err := strconv.Atoi(strings.TrimSpace(pidStr))
  303. c.Assert(err, checker.IsNil)
  304. c.Assert(syscall.Kill(pid, syscall.SIGKILL), checker.IsNil)
  305. time.Sleep(time.Second) // give some time to handle the signal
  306. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  307. containers2 = getContainers()
  308. c.Assert(containers2, checker.HasLen, instances)
  309. for i := range containers {
  310. if i == toRemove {
  311. c.Assert(containers2[i], checker.IsNil)
  312. } else {
  313. c.Assert(containers2[i], checker.NotNil)
  314. }
  315. }
  316. }
  317. func (s *DockerSwarmSuite) TestApiSwarmLeaderElection(c *check.C) {
  318. // Create 3 nodes
  319. d1 := s.AddDaemon(c, true, true)
  320. d2 := s.AddDaemon(c, true, true)
  321. d3 := s.AddDaemon(c, true, true)
  322. // assert that the first node we made is the leader, and the other two are followers
  323. c.Assert(d1.getNode(c, d1.NodeID).ManagerStatus.Leader, checker.True)
  324. c.Assert(d1.getNode(c, d2.NodeID).ManagerStatus.Leader, checker.False)
  325. c.Assert(d1.getNode(c, d3.NodeID).ManagerStatus.Leader, checker.False)
  326. leader := d1
  327. // stop the leader
  328. leader.Stop()
  329. // wait for an election to occur
  330. var newleader *SwarmDaemon
  331. for _, d := range []*SwarmDaemon{d2, d3} {
  332. if d.getNode(c, d.NodeID).ManagerStatus.Leader {
  333. newleader = d
  334. break
  335. }
  336. }
  337. // assert that we have a new leader
  338. c.Assert(newleader, checker.NotNil)
  339. // add the old leader back
  340. leader.Start()
  341. // clear leader and reinit the followers list
  342. followers := make([]*SwarmDaemon, 0, 3)
  343. // pick out the leader and the followers again
  344. for _, d := range []*SwarmDaemon{d1, d2, d3} {
  345. if d1.getNode(c, d.NodeID).ManagerStatus.Leader {
  346. leader = d
  347. } else {
  348. followers = append(followers, d)
  349. }
  350. }
  351. // verify that we still only have 1 leader and 2 followers
  352. c.Assert(leader, checker.NotNil)
  353. c.Assert(followers, checker.HasLen, 2)
  354. // and that after we added d1 back, the leader hasn't changed
  355. c.Assert(leader.NodeID, checker.Equals, newleader.NodeID)
  356. }
  357. func (s *DockerSwarmSuite) TestApiSwarmRaftQuorum(c *check.C) {
  358. testRequires(c, Network)
  359. d1 := s.AddDaemon(c, true, true)
  360. d2 := s.AddDaemon(c, true, true)
  361. d3 := s.AddDaemon(c, true, true)
  362. d1.createService(c, simpleTestService)
  363. c.Assert(d2.Stop(), checker.IsNil)
  364. d1.createService(c, simpleTestService, func(s *swarm.Service) {
  365. s.Spec.Name = "top1"
  366. })
  367. c.Assert(d3.Stop(), checker.IsNil)
  368. var service swarm.Service
  369. simpleTestService(&service)
  370. service.Spec.Name = "top2"
  371. status, out, err := d1.SockRequest("POST", "/services/create", service.Spec)
  372. c.Assert(err, checker.IsNil)
  373. c.Assert(status, checker.Equals, http.StatusInternalServerError, check.Commentf("deadline exceeded", string(out)))
  374. c.Assert(d2.Start(), checker.IsNil)
  375. d1.createService(c, simpleTestService, func(s *swarm.Service) {
  376. s.Spec.Name = "top3"
  377. })
  378. }
  379. func (s *DockerSwarmSuite) TestApiSwarmListNodes(c *check.C) {
  380. testRequires(c, Network)
  381. d1 := s.AddDaemon(c, true, true)
  382. d2 := s.AddDaemon(c, true, false)
  383. d3 := s.AddDaemon(c, true, false)
  384. nodes := d1.listNodes(c)
  385. c.Assert(len(nodes), checker.Equals, 3, check.Commentf("nodes: %#v", nodes))
  386. loop0:
  387. for _, n := range nodes {
  388. for _, d := range []*SwarmDaemon{d1, d2, d3} {
  389. if n.ID == d.NodeID {
  390. continue loop0
  391. }
  392. }
  393. c.Errorf("unknown nodeID %v", n.ID)
  394. }
  395. }
  396. func (s *DockerSwarmSuite) TestApiSwarmNodeUpdate(c *check.C) {
  397. testRequires(c, Network)
  398. d := s.AddDaemon(c, true, true)
  399. nodes := d.listNodes(c)
  400. d.updateNode(c, nodes[0].ID, func(n *swarm.Node) {
  401. n.Spec.Availability = swarm.NodeAvailabilityPause
  402. })
  403. n := d.getNode(c, nodes[0].ID)
  404. c.Assert(n.Spec.Availability, checker.Equals, swarm.NodeAvailabilityPause)
  405. }
  406. func (s *DockerSwarmSuite) TestApiSwarmNodeDrainPause(c *check.C) {
  407. testRequires(c, Network)
  408. d1 := s.AddDaemon(c, true, true)
  409. d2 := s.AddDaemon(c, true, false)
  410. time.Sleep(1 * time.Second) // make sure all daemons are ready to accept tasks
  411. // start a service, expect balanced distribution
  412. instances := 8
  413. id := d1.createService(c, simpleTestService, setInstances(instances))
  414. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.GreaterThan, 0)
  415. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.GreaterThan, 0)
  416. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount), checker.Equals, instances)
  417. // drain d2, all containers should move to d1
  418. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  419. n.Spec.Availability = swarm.NodeAvailabilityDrain
  420. })
  421. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.Equals, instances)
  422. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.Equals, 0)
  423. // set d2 back to active
  424. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  425. n.Spec.Availability = swarm.NodeAvailabilityActive
  426. })
  427. instances = 1
  428. d1.updateService(c, d1.getService(c, id), setInstances(instances))
  429. waitAndAssert(c, defaultReconciliationTimeout*2, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount), checker.Equals, instances)
  430. instances = 8
  431. d1.updateService(c, d1.getService(c, id), setInstances(instances))
  432. // drained node first so we don't get any old containers
  433. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.GreaterThan, 0)
  434. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.GreaterThan, 0)
  435. waitAndAssert(c, defaultReconciliationTimeout*2, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount), checker.Equals, instances)
  436. d2ContainerCount := len(d2.activeContainers())
  437. // set d2 to paused, scale service up, only d1 gets new tasks
  438. d1.updateNode(c, d2.NodeID, func(n *swarm.Node) {
  439. n.Spec.Availability = swarm.NodeAvailabilityPause
  440. })
  441. instances = 14
  442. d1.updateService(c, d1.getService(c, id), setInstances(instances))
  443. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.Equals, instances-d2ContainerCount)
  444. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.Equals, d2ContainerCount)
  445. }
  446. func (s *DockerSwarmSuite) TestApiSwarmLeaveRemovesContainer(c *check.C) {
  447. testRequires(c, Network)
  448. d := s.AddDaemon(c, true, true)
  449. instances := 2
  450. d.createService(c, simpleTestService, setInstances(instances))
  451. id, err := d.Cmd("run", "-d", "busybox", "top")
  452. c.Assert(err, checker.IsNil)
  453. id = strings.TrimSpace(id)
  454. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, instances+1)
  455. c.Assert(d.Leave(false), checker.NotNil)
  456. c.Assert(d.Leave(true), checker.IsNil)
  457. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, 1)
  458. id2, err := d.Cmd("ps", "-q")
  459. c.Assert(err, checker.IsNil)
  460. c.Assert(id, checker.HasPrefix, strings.TrimSpace(id2))
  461. }
  462. // #23629
  463. func (s *DockerSwarmSuite) TestApiSwarmLeaveOnPendingJoin(c *check.C) {
  464. s.AddDaemon(c, true, true)
  465. d2 := s.AddDaemon(c, false, false)
  466. id, err := d2.Cmd("run", "-d", "busybox", "top")
  467. c.Assert(err, checker.IsNil)
  468. id = strings.TrimSpace(id)
  469. go d2.Join(swarm.JoinRequest{
  470. RemoteAddrs: []string{"nosuchhost:1234"},
  471. })
  472. waitAndAssert(c, defaultReconciliationTimeout, d2.checkLocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  473. waitAndAssert(c, defaultReconciliationTimeout, d2.checkActiveContainerCount, checker.Equals, 1)
  474. id2, err := d2.Cmd("ps", "-q")
  475. c.Assert(err, checker.IsNil)
  476. c.Assert(id, checker.HasPrefix, strings.TrimSpace(id2))
  477. }
  478. // #23705
  479. func (s *DockerSwarmSuite) TestApiSwarmRestoreOnPendingJoin(c *check.C) {
  480. d := s.AddDaemon(c, false, false)
  481. go d.Join(swarm.JoinRequest{
  482. RemoteAddrs: []string{"nosuchhost:1234"},
  483. })
  484. waitAndAssert(c, defaultReconciliationTimeout, d.checkLocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  485. c.Assert(d.Stop(), checker.IsNil)
  486. c.Assert(d.Start(), checker.IsNil)
  487. info, err := d.info()
  488. c.Assert(err, checker.IsNil)
  489. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateInactive)
  490. }
  491. func (s *DockerSwarmSuite) TestApiSwarmManagerRestore(c *check.C) {
  492. testRequires(c, Network)
  493. d1 := s.AddDaemon(c, true, true)
  494. instances := 2
  495. id := d1.createService(c, simpleTestService, setInstances(instances))
  496. d1.getService(c, id)
  497. d1.Stop()
  498. d1.Start()
  499. d1.getService(c, id)
  500. d2 := s.AddDaemon(c, true, true)
  501. d2.getService(c, id)
  502. d2.Stop()
  503. d2.Start()
  504. d2.getService(c, id)
  505. d3 := s.AddDaemon(c, true, true)
  506. d3.getService(c, id)
  507. d3.Stop()
  508. d3.Start()
  509. d3.getService(c, id)
  510. d3.Kill()
  511. time.Sleep(1 * time.Second) // time to handle signal
  512. d3.Start()
  513. d3.getService(c, id)
  514. }
  515. func (s *DockerSwarmSuite) TestApiSwarmScaleNoRollingUpdate(c *check.C) {
  516. testRequires(c, Network)
  517. d := s.AddDaemon(c, true, true)
  518. instances := 2
  519. id := d.createService(c, simpleTestService, setInstances(instances))
  520. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, instances)
  521. containers := d.activeContainers()
  522. instances = 4
  523. d.updateService(c, d.getService(c, id), setInstances(instances))
  524. waitAndAssert(c, defaultReconciliationTimeout, d.checkActiveContainerCount, checker.Equals, instances)
  525. containers2 := d.activeContainers()
  526. loop0:
  527. for _, c1 := range containers {
  528. for _, c2 := range containers2 {
  529. if c1 == c2 {
  530. continue loop0
  531. }
  532. }
  533. c.Errorf("container %v not found in new set %#v", c1, containers2)
  534. }
  535. }
  536. func (s *DockerSwarmSuite) TestApiSwarmInvalidAddress(c *check.C) {
  537. d := s.AddDaemon(c, false, false)
  538. req := swarm.InitRequest{
  539. ListenAddr: "",
  540. }
  541. status, _, err := d.SockRequest("POST", "/swarm/init", req)
  542. c.Assert(err, checker.IsNil)
  543. c.Assert(status, checker.Equals, http.StatusInternalServerError)
  544. req2 := swarm.JoinRequest{
  545. ListenAddr: "0.0.0.0:2377",
  546. RemoteAddrs: []string{""},
  547. }
  548. status, _, err = d.SockRequest("POST", "/swarm/join", req2)
  549. c.Assert(err, checker.IsNil)
  550. c.Assert(status, checker.Equals, http.StatusInternalServerError)
  551. }
  552. func (s *DockerSwarmSuite) TestApiSwarmForceNewCluster(c *check.C) {
  553. d1 := s.AddDaemon(c, true, true)
  554. d2 := s.AddDaemon(c, true, true)
  555. instances := 2
  556. id := d1.createService(c, simpleTestService, setInstances(instances))
  557. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d2.checkActiveContainerCount), checker.Equals, instances)
  558. c.Assert(d2.Stop(), checker.IsNil)
  559. time.Sleep(5 * time.Second)
  560. c.Assert(d1.Init(swarm.InitRequest{
  561. ForceNewCluster: true,
  562. Spec: swarm.Spec{},
  563. }), checker.IsNil)
  564. waitAndAssert(c, defaultReconciliationTimeout, d1.checkActiveContainerCount, checker.Equals, instances)
  565. d3 := s.AddDaemon(c, true, true)
  566. info, err := d3.info()
  567. c.Assert(err, checker.IsNil)
  568. c.Assert(info.ControlAvailable, checker.True)
  569. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  570. instances = 4
  571. d3.updateService(c, d3.getService(c, id), setInstances(instances))
  572. waitAndAssert(c, defaultReconciliationTimeout, reducedCheck(sumAsIntegers, d1.checkActiveContainerCount, d3.checkActiveContainerCount), checker.Equals, instances)
  573. }
  574. func simpleTestService(s *swarm.Service) {
  575. var ureplicas uint64
  576. ureplicas = 1
  577. s.Spec = swarm.ServiceSpec{
  578. TaskTemplate: swarm.TaskSpec{
  579. ContainerSpec: swarm.ContainerSpec{
  580. Image: "busybox:latest",
  581. Command: []string{"/bin/top"},
  582. },
  583. },
  584. Mode: swarm.ServiceMode{
  585. Replicated: &swarm.ReplicatedService{
  586. Replicas: &ureplicas,
  587. },
  588. },
  589. }
  590. s.Spec.Name = "top"
  591. }
  592. func serviceForUpdate(s *swarm.Service) {
  593. var ureplicas uint64
  594. ureplicas = 1
  595. s.Spec = swarm.ServiceSpec{
  596. TaskTemplate: swarm.TaskSpec{
  597. ContainerSpec: swarm.ContainerSpec{
  598. Image: "busybox:latest",
  599. Command: []string{"/bin/top"},
  600. },
  601. },
  602. Mode: swarm.ServiceMode{
  603. Replicated: &swarm.ReplicatedService{
  604. Replicas: &ureplicas,
  605. },
  606. },
  607. UpdateConfig: &swarm.UpdateConfig{
  608. Parallelism: 2,
  609. Delay: 8 * time.Second,
  610. FailureAction: swarm.UpdateFailureActionContinue,
  611. },
  612. }
  613. s.Spec.Name = "updatetest"
  614. }
  615. func setInstances(replicas int) serviceConstructor {
  616. ureplicas := uint64(replicas)
  617. return func(s *swarm.Service) {
  618. s.Spec.Mode = swarm.ServiceMode{
  619. Replicated: &swarm.ReplicatedService{
  620. Replicas: &ureplicas,
  621. },
  622. }
  623. }
  624. }
  625. func setImage(image string) serviceConstructor {
  626. return func(s *swarm.Service) {
  627. s.Spec.TaskTemplate.ContainerSpec.Image = image
  628. }
  629. }
  630. func setGlobalMode(s *swarm.Service) {
  631. s.Spec.Mode = swarm.ServiceMode{
  632. Global: &swarm.GlobalService{},
  633. }
  634. }
  635. func checkClusterHealth(c *check.C, cl []*SwarmDaemon, managerCount, workerCount int) {
  636. var totalMCount, totalWCount int
  637. for _, d := range cl {
  638. info, err := d.info()
  639. c.Assert(err, check.IsNil)
  640. if !info.ControlAvailable {
  641. totalWCount++
  642. continue
  643. }
  644. var leaderFound bool
  645. totalMCount++
  646. var mCount, wCount int
  647. for _, n := range d.listNodes(c) {
  648. c.Assert(n.Status.State, checker.Equals, swarm.NodeStateReady, check.Commentf("state of node %s, reported by %s", n.ID, d.Info.NodeID))
  649. c.Assert(n.Spec.Availability, checker.Equals, swarm.NodeAvailabilityActive, check.Commentf("availability of node %s, reported by %s", n.ID, d.Info.NodeID))
  650. if n.Spec.Role == swarm.NodeRoleManager {
  651. c.Assert(n.ManagerStatus, checker.NotNil, check.Commentf("manager status of node %s (manager), reported by %s", n.ID, d.Info.NodeID))
  652. if n.ManagerStatus.Leader {
  653. leaderFound = true
  654. }
  655. mCount++
  656. } else {
  657. c.Assert(n.ManagerStatus, checker.IsNil, check.Commentf("manager status of node %s (worker), reported by %s", n.ID, d.Info.NodeID))
  658. wCount++
  659. }
  660. }
  661. c.Assert(leaderFound, checker.True, check.Commentf("lack of leader reported by node %s", info.NodeID))
  662. c.Assert(mCount, checker.Equals, managerCount, check.Commentf("managers count reported by node %s", info.NodeID))
  663. c.Assert(wCount, checker.Equals, workerCount, check.Commentf("workers count reported by node %s", info.NodeID))
  664. }
  665. c.Assert(totalMCount, checker.Equals, managerCount)
  666. c.Assert(totalWCount, checker.Equals, workerCount)
  667. }
  668. func (s *DockerSwarmSuite) TestApiSwarmRestartCluster(c *check.C) {
  669. mCount, wCount := 5, 1
  670. var nodes []*SwarmDaemon
  671. for i := 0; i < mCount; i++ {
  672. manager := s.AddDaemon(c, true, true)
  673. info, err := manager.info()
  674. c.Assert(err, checker.IsNil)
  675. c.Assert(info.ControlAvailable, checker.True)
  676. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  677. nodes = append(nodes, manager)
  678. }
  679. for i := 0; i < wCount; i++ {
  680. worker := s.AddDaemon(c, true, false)
  681. info, err := worker.info()
  682. c.Assert(err, checker.IsNil)
  683. c.Assert(info.ControlAvailable, checker.False)
  684. c.Assert(info.LocalNodeState, checker.Equals, swarm.LocalNodeStateActive)
  685. nodes = append(nodes, worker)
  686. }
  687. // stop whole cluster
  688. {
  689. var wg sync.WaitGroup
  690. wg.Add(len(nodes))
  691. errs := make(chan error, len(nodes))
  692. for _, d := range nodes {
  693. go func(daemon *SwarmDaemon) {
  694. defer wg.Done()
  695. if err := daemon.Stop(); err != nil {
  696. errs <- err
  697. }
  698. if root := os.Getenv("DOCKER_REMAP_ROOT"); root != "" {
  699. daemon.root = filepath.Dir(daemon.root)
  700. }
  701. }(d)
  702. }
  703. wg.Wait()
  704. close(errs)
  705. for err := range errs {
  706. c.Assert(err, check.IsNil)
  707. }
  708. }
  709. // start whole cluster
  710. {
  711. var wg sync.WaitGroup
  712. wg.Add(len(nodes))
  713. errs := make(chan error, len(nodes))
  714. for _, d := range nodes {
  715. go func(daemon *SwarmDaemon) {
  716. defer wg.Done()
  717. if err := daemon.Start("--iptables=false"); err != nil {
  718. errs <- err
  719. }
  720. }(d)
  721. }
  722. wg.Wait()
  723. close(errs)
  724. for err := range errs {
  725. c.Assert(err, check.IsNil)
  726. }
  727. }
  728. checkClusterHealth(c, nodes, mCount, wCount)
  729. }