docker_api_swarm_test.go 25 KB

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