stateful_set_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712
  1. /*
  2. Copyright 2016 The Kubernetes Authors.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. */
  13. package statefulset
  14. import (
  15. "sort"
  16. "testing"
  17. apps "k8s.io/api/apps/v1"
  18. "k8s.io/api/core/v1"
  19. metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  20. "k8s.io/apimachinery/pkg/runtime"
  21. "k8s.io/apimachinery/pkg/util/sets"
  22. "k8s.io/client-go/informers"
  23. "k8s.io/client-go/kubernetes/fake"
  24. "k8s.io/client-go/tools/cache"
  25. "k8s.io/client-go/tools/record"
  26. "k8s.io/kubernetes/pkg/controller"
  27. "k8s.io/kubernetes/pkg/controller/history"
  28. )
  29. func alwaysReady() bool { return true }
  30. func TestStatefulSetControllerCreates(t *testing.T) {
  31. set := newStatefulSet(3)
  32. ssc, spc := newFakeStatefulSetController(set)
  33. if err := scaleUpStatefulSetController(set, ssc, spc); err != nil {
  34. t.Errorf("Failed to turn up StatefulSet : %s", err)
  35. }
  36. if obj, _, err := spc.setsIndexer.Get(set); err != nil {
  37. t.Error(err)
  38. } else {
  39. set = obj.(*apps.StatefulSet)
  40. }
  41. if set.Status.Replicas != 3 {
  42. t.Errorf("set.Status.Replicas = %v; want 3", set.Status.Replicas)
  43. }
  44. }
  45. func TestStatefulSetControllerDeletes(t *testing.T) {
  46. set := newStatefulSet(3)
  47. ssc, spc := newFakeStatefulSetController(set)
  48. if err := scaleUpStatefulSetController(set, ssc, spc); err != nil {
  49. t.Errorf("Failed to turn up StatefulSet : %s", err)
  50. }
  51. if obj, _, err := spc.setsIndexer.Get(set); err != nil {
  52. t.Error(err)
  53. } else {
  54. set = obj.(*apps.StatefulSet)
  55. }
  56. if set.Status.Replicas != 3 {
  57. t.Errorf("set.Status.Replicas = %v; want 3", set.Status.Replicas)
  58. }
  59. *set.Spec.Replicas = 0
  60. if err := scaleDownStatefulSetController(set, ssc, spc); err != nil {
  61. t.Errorf("Failed to turn down StatefulSet : %s", err)
  62. }
  63. if obj, _, err := spc.setsIndexer.Get(set); err != nil {
  64. t.Error(err)
  65. } else {
  66. set = obj.(*apps.StatefulSet)
  67. }
  68. if set.Status.Replicas != 0 {
  69. t.Errorf("set.Status.Replicas = %v; want 0", set.Status.Replicas)
  70. }
  71. }
  72. func TestStatefulSetControllerRespectsTermination(t *testing.T) {
  73. set := newStatefulSet(3)
  74. ssc, spc := newFakeStatefulSetController(set)
  75. if err := scaleUpStatefulSetController(set, ssc, spc); err != nil {
  76. t.Errorf("Failed to turn up StatefulSet : %s", err)
  77. }
  78. if obj, _, err := spc.setsIndexer.Get(set); err != nil {
  79. t.Error(err)
  80. } else {
  81. set = obj.(*apps.StatefulSet)
  82. }
  83. if set.Status.Replicas != 3 {
  84. t.Errorf("set.Status.Replicas = %v; want 3", set.Status.Replicas)
  85. }
  86. pods, err := spc.addTerminatingPod(set, 3)
  87. if err != nil {
  88. t.Error(err)
  89. }
  90. pods, err = spc.addTerminatingPod(set, 4)
  91. if err != nil {
  92. t.Error(err)
  93. }
  94. ssc.syncStatefulSet(set, pods)
  95. selector, err := metav1.LabelSelectorAsSelector(set.Spec.Selector)
  96. if err != nil {
  97. t.Error(err)
  98. }
  99. pods, err = spc.podsLister.Pods(set.Namespace).List(selector)
  100. if err != nil {
  101. t.Error(err)
  102. }
  103. if len(pods) != 5 {
  104. t.Error("StatefulSet does not respect termination")
  105. }
  106. sort.Sort(ascendingOrdinal(pods))
  107. spc.DeleteStatefulPod(set, pods[3])
  108. spc.DeleteStatefulPod(set, pods[4])
  109. *set.Spec.Replicas = 0
  110. if err := scaleDownStatefulSetController(set, ssc, spc); err != nil {
  111. t.Errorf("Failed to turn down StatefulSet : %s", err)
  112. }
  113. if obj, _, err := spc.setsIndexer.Get(set); err != nil {
  114. t.Error(err)
  115. } else {
  116. set = obj.(*apps.StatefulSet)
  117. }
  118. if set.Status.Replicas != 0 {
  119. t.Errorf("set.Status.Replicas = %v; want 0", set.Status.Replicas)
  120. }
  121. }
  122. func TestStatefulSetControllerBlocksScaling(t *testing.T) {
  123. set := newStatefulSet(3)
  124. ssc, spc := newFakeStatefulSetController(set)
  125. if err := scaleUpStatefulSetController(set, ssc, spc); err != nil {
  126. t.Errorf("Failed to turn up StatefulSet : %s", err)
  127. }
  128. if obj, _, err := spc.setsIndexer.Get(set); err != nil {
  129. t.Error(err)
  130. } else {
  131. set = obj.(*apps.StatefulSet)
  132. }
  133. if set.Status.Replicas != 3 {
  134. t.Errorf("set.Status.Replicas = %v; want 3", set.Status.Replicas)
  135. }
  136. *set.Spec.Replicas = 5
  137. fakeResourceVersion(set)
  138. spc.setsIndexer.Update(set)
  139. pods, err := spc.setPodTerminated(set, 0)
  140. if err != nil {
  141. t.Error("Failed to set pod terminated at ordinal 0")
  142. }
  143. ssc.enqueueStatefulSet(set)
  144. fakeWorker(ssc)
  145. selector, err := metav1.LabelSelectorAsSelector(set.Spec.Selector)
  146. if err != nil {
  147. t.Error(err)
  148. }
  149. pods, err = spc.podsLister.Pods(set.Namespace).List(selector)
  150. if err != nil {
  151. t.Error(err)
  152. }
  153. if len(pods) != 3 {
  154. t.Error("StatefulSet does not block scaling")
  155. }
  156. sort.Sort(ascendingOrdinal(pods))
  157. spc.DeleteStatefulPod(set, pods[0])
  158. ssc.enqueueStatefulSet(set)
  159. fakeWorker(ssc)
  160. pods, err = spc.podsLister.Pods(set.Namespace).List(selector)
  161. if err != nil {
  162. t.Error(err)
  163. }
  164. if len(pods) != 3 {
  165. t.Error("StatefulSet does not resume when terminated Pod is removed")
  166. }
  167. }
  168. func TestStatefulSetControllerDeletionTimestamp(t *testing.T) {
  169. set := newStatefulSet(3)
  170. set.DeletionTimestamp = new(metav1.Time)
  171. ssc, spc := newFakeStatefulSetController(set)
  172. spc.setsIndexer.Add(set)
  173. // Force a sync. It should not try to create any Pods.
  174. ssc.enqueueStatefulSet(set)
  175. fakeWorker(ssc)
  176. selector, err := metav1.LabelSelectorAsSelector(set.Spec.Selector)
  177. if err != nil {
  178. t.Fatal(err)
  179. }
  180. pods, err := spc.podsLister.Pods(set.Namespace).List(selector)
  181. if err != nil {
  182. t.Fatal(err)
  183. }
  184. if got, want := len(pods), 0; got != want {
  185. t.Errorf("len(pods) = %v, want %v", got, want)
  186. }
  187. }
  188. func TestStatefulSetControllerDeletionTimestampRace(t *testing.T) {
  189. set := newStatefulSet(3)
  190. // The bare client says it IS deleted.
  191. set.DeletionTimestamp = new(metav1.Time)
  192. ssc, spc := newFakeStatefulSetController(set)
  193. // The lister (cache) says it's NOT deleted.
  194. set2 := *set
  195. set2.DeletionTimestamp = nil
  196. spc.setsIndexer.Add(&set2)
  197. // The recheck occurs in the presence of a matching orphan.
  198. pod := newStatefulSetPod(set, 1)
  199. pod.OwnerReferences = nil
  200. spc.podsIndexer.Add(pod)
  201. // Force a sync. It should not try to create any Pods.
  202. ssc.enqueueStatefulSet(set)
  203. fakeWorker(ssc)
  204. selector, err := metav1.LabelSelectorAsSelector(set.Spec.Selector)
  205. if err != nil {
  206. t.Fatal(err)
  207. }
  208. pods, err := spc.podsLister.Pods(set.Namespace).List(selector)
  209. if err != nil {
  210. t.Fatal(err)
  211. }
  212. if got, want := len(pods), 1; got != want {
  213. t.Errorf("len(pods) = %v, want %v", got, want)
  214. }
  215. }
  216. func TestStatefulSetControllerAddPod(t *testing.T) {
  217. ssc, spc := newFakeStatefulSetController()
  218. set1 := newStatefulSet(3)
  219. set2 := newStatefulSet(3)
  220. pod1 := newStatefulSetPod(set1, 0)
  221. pod2 := newStatefulSetPod(set2, 0)
  222. spc.setsIndexer.Add(set1)
  223. spc.setsIndexer.Add(set2)
  224. ssc.addPod(pod1)
  225. key, done := ssc.queue.Get()
  226. if key == nil || done {
  227. t.Error("failed to enqueue StatefulSet")
  228. } else if key, ok := key.(string); !ok {
  229. t.Error("key is not a string")
  230. } else if expectedKey, _ := controller.KeyFunc(set1); expectedKey != key {
  231. t.Errorf("expected StatefulSet key %s found %s", expectedKey, key)
  232. }
  233. ssc.queue.Done(key)
  234. ssc.addPod(pod2)
  235. key, done = ssc.queue.Get()
  236. if key == nil || done {
  237. t.Error("failed to enqueue StatefulSet")
  238. } else if key, ok := key.(string); !ok {
  239. t.Error("key is not a string")
  240. } else if expectedKey, _ := controller.KeyFunc(set2); expectedKey != key {
  241. t.Errorf("expected StatefulSet key %s found %s", expectedKey, key)
  242. }
  243. ssc.queue.Done(key)
  244. }
  245. func TestStatefulSetControllerAddPodOrphan(t *testing.T) {
  246. ssc, spc := newFakeStatefulSetController()
  247. set1 := newStatefulSet(3)
  248. set2 := newStatefulSet(3)
  249. set2.Name = "foo2"
  250. set3 := newStatefulSet(3)
  251. set3.Name = "foo3"
  252. set3.Spec.Selector.MatchLabels = map[string]string{"foo3": "bar"}
  253. pod := newStatefulSetPod(set1, 0)
  254. spc.setsIndexer.Add(set1)
  255. spc.setsIndexer.Add(set2)
  256. spc.setsIndexer.Add(set3)
  257. // Make pod an orphan. Expect matching sets to be queued.
  258. pod.OwnerReferences = nil
  259. ssc.addPod(pod)
  260. if got, want := ssc.queue.Len(), 2; got != want {
  261. t.Errorf("queue.Len() = %v, want %v", got, want)
  262. }
  263. }
  264. func TestStatefulSetControllerAddPodNoSet(t *testing.T) {
  265. ssc, _ := newFakeStatefulSetController()
  266. set := newStatefulSet(3)
  267. pod := newStatefulSetPod(set, 0)
  268. ssc.addPod(pod)
  269. ssc.queue.ShutDown()
  270. key, _ := ssc.queue.Get()
  271. if key != nil {
  272. t.Errorf("StatefulSet enqueued key for Pod with no Set %s", key)
  273. }
  274. }
  275. func TestStatefulSetControllerUpdatePod(t *testing.T) {
  276. ssc, spc := newFakeStatefulSetController()
  277. set1 := newStatefulSet(3)
  278. set2 := newStatefulSet(3)
  279. set2.Name = "foo2"
  280. pod1 := newStatefulSetPod(set1, 0)
  281. pod2 := newStatefulSetPod(set2, 0)
  282. spc.setsIndexer.Add(set1)
  283. spc.setsIndexer.Add(set2)
  284. prev := *pod1
  285. fakeResourceVersion(pod1)
  286. ssc.updatePod(&prev, pod1)
  287. key, done := ssc.queue.Get()
  288. if key == nil || done {
  289. t.Error("failed to enqueue StatefulSet")
  290. } else if key, ok := key.(string); !ok {
  291. t.Error("key is not a string")
  292. } else if expectedKey, _ := controller.KeyFunc(set1); expectedKey != key {
  293. t.Errorf("expected StatefulSet key %s found %s", expectedKey, key)
  294. }
  295. prev = *pod2
  296. fakeResourceVersion(pod2)
  297. ssc.updatePod(&prev, pod2)
  298. key, done = ssc.queue.Get()
  299. if key == nil || done {
  300. t.Error("failed to enqueue StatefulSet")
  301. } else if key, ok := key.(string); !ok {
  302. t.Error("key is not a string")
  303. } else if expectedKey, _ := controller.KeyFunc(set2); expectedKey != key {
  304. t.Errorf("expected StatefulSet key %s found %s", expectedKey, key)
  305. }
  306. }
  307. func TestStatefulSetControllerUpdatePodWithNoSet(t *testing.T) {
  308. ssc, _ := newFakeStatefulSetController()
  309. set := newStatefulSet(3)
  310. pod := newStatefulSetPod(set, 0)
  311. prev := *pod
  312. fakeResourceVersion(pod)
  313. ssc.updatePod(&prev, pod)
  314. ssc.queue.ShutDown()
  315. key, _ := ssc.queue.Get()
  316. if key != nil {
  317. t.Errorf("StatefulSet enqueued key for Pod with no Set %s", key)
  318. }
  319. }
  320. func TestStatefulSetControllerUpdatePodWithSameVersion(t *testing.T) {
  321. ssc, spc := newFakeStatefulSetController()
  322. set := newStatefulSet(3)
  323. pod := newStatefulSetPod(set, 0)
  324. spc.setsIndexer.Add(set)
  325. ssc.updatePod(pod, pod)
  326. ssc.queue.ShutDown()
  327. key, _ := ssc.queue.Get()
  328. if key != nil {
  329. t.Errorf("StatefulSet enqueued key for Pod with no Set %s", key)
  330. }
  331. }
  332. func TestStatefulSetControllerUpdatePodOrphanWithNewLabels(t *testing.T) {
  333. ssc, spc := newFakeStatefulSetController()
  334. set := newStatefulSet(3)
  335. pod := newStatefulSetPod(set, 0)
  336. pod.OwnerReferences = nil
  337. set2 := newStatefulSet(3)
  338. set2.Name = "foo2"
  339. spc.setsIndexer.Add(set)
  340. spc.setsIndexer.Add(set2)
  341. clone := *pod
  342. clone.Labels = map[string]string{"foo2": "bar2"}
  343. fakeResourceVersion(&clone)
  344. ssc.updatePod(&clone, pod)
  345. if got, want := ssc.queue.Len(), 2; got != want {
  346. t.Errorf("queue.Len() = %v, want %v", got, want)
  347. }
  348. }
  349. func TestStatefulSetControllerUpdatePodChangeControllerRef(t *testing.T) {
  350. ssc, spc := newFakeStatefulSetController()
  351. set := newStatefulSet(3)
  352. set2 := newStatefulSet(3)
  353. set2.Name = "foo2"
  354. pod := newStatefulSetPod(set, 0)
  355. pod2 := newStatefulSetPod(set2, 0)
  356. spc.setsIndexer.Add(set)
  357. spc.setsIndexer.Add(set2)
  358. clone := *pod
  359. clone.OwnerReferences = pod2.OwnerReferences
  360. fakeResourceVersion(&clone)
  361. ssc.updatePod(&clone, pod)
  362. if got, want := ssc.queue.Len(), 2; got != want {
  363. t.Errorf("queue.Len() = %v, want %v", got, want)
  364. }
  365. }
  366. func TestStatefulSetControllerUpdatePodRelease(t *testing.T) {
  367. ssc, spc := newFakeStatefulSetController()
  368. set := newStatefulSet(3)
  369. set2 := newStatefulSet(3)
  370. set2.Name = "foo2"
  371. pod := newStatefulSetPod(set, 0)
  372. spc.setsIndexer.Add(set)
  373. spc.setsIndexer.Add(set2)
  374. clone := *pod
  375. clone.OwnerReferences = nil
  376. fakeResourceVersion(&clone)
  377. ssc.updatePod(pod, &clone)
  378. if got, want := ssc.queue.Len(), 2; got != want {
  379. t.Errorf("queue.Len() = %v, want %v", got, want)
  380. }
  381. }
  382. func TestStatefulSetControllerDeletePod(t *testing.T) {
  383. ssc, spc := newFakeStatefulSetController()
  384. set1 := newStatefulSet(3)
  385. set2 := newStatefulSet(3)
  386. set2.Name = "foo2"
  387. pod1 := newStatefulSetPod(set1, 0)
  388. pod2 := newStatefulSetPod(set2, 0)
  389. spc.setsIndexer.Add(set1)
  390. spc.setsIndexer.Add(set2)
  391. ssc.deletePod(pod1)
  392. key, done := ssc.queue.Get()
  393. if key == nil || done {
  394. t.Error("failed to enqueue StatefulSet")
  395. } else if key, ok := key.(string); !ok {
  396. t.Error("key is not a string")
  397. } else if expectedKey, _ := controller.KeyFunc(set1); expectedKey != key {
  398. t.Errorf("expected StatefulSet key %s found %s", expectedKey, key)
  399. }
  400. ssc.deletePod(pod2)
  401. key, done = ssc.queue.Get()
  402. if key == nil || done {
  403. t.Error("failed to enqueue StatefulSet")
  404. } else if key, ok := key.(string); !ok {
  405. t.Error("key is not a string")
  406. } else if expectedKey, _ := controller.KeyFunc(set2); expectedKey != key {
  407. t.Errorf("expected StatefulSet key %s found %s", expectedKey, key)
  408. }
  409. }
  410. func TestStatefulSetControllerDeletePodOrphan(t *testing.T) {
  411. ssc, spc := newFakeStatefulSetController()
  412. set1 := newStatefulSet(3)
  413. set2 := newStatefulSet(3)
  414. set2.Name = "foo2"
  415. pod1 := newStatefulSetPod(set1, 0)
  416. spc.setsIndexer.Add(set1)
  417. spc.setsIndexer.Add(set2)
  418. pod1.OwnerReferences = nil
  419. ssc.deletePod(pod1)
  420. if got, want := ssc.queue.Len(), 0; got != want {
  421. t.Errorf("queue.Len() = %v, want %v", got, want)
  422. }
  423. }
  424. func TestStatefulSetControllerDeletePodTombstone(t *testing.T) {
  425. ssc, spc := newFakeStatefulSetController()
  426. set := newStatefulSet(3)
  427. pod := newStatefulSetPod(set, 0)
  428. spc.setsIndexer.Add(set)
  429. tombstoneKey, _ := controller.KeyFunc(pod)
  430. tombstone := cache.DeletedFinalStateUnknown{Key: tombstoneKey, Obj: pod}
  431. ssc.deletePod(tombstone)
  432. key, done := ssc.queue.Get()
  433. if key == nil || done {
  434. t.Error("failed to enqueue StatefulSet")
  435. } else if key, ok := key.(string); !ok {
  436. t.Error("key is not a string")
  437. } else if expectedKey, _ := controller.KeyFunc(set); expectedKey != key {
  438. t.Errorf("expected StatefulSet key %s found %s", expectedKey, key)
  439. }
  440. }
  441. func TestStatefulSetControllerGetStatefulSetsForPod(t *testing.T) {
  442. ssc, spc := newFakeStatefulSetController()
  443. set1 := newStatefulSet(3)
  444. set2 := newStatefulSet(3)
  445. set2.Name = "foo2"
  446. pod := newStatefulSetPod(set1, 0)
  447. spc.setsIndexer.Add(set1)
  448. spc.setsIndexer.Add(set2)
  449. spc.podsIndexer.Add(pod)
  450. sets := ssc.getStatefulSetsForPod(pod)
  451. if got, want := len(sets), 2; got != want {
  452. t.Errorf("len(sets) = %v, want %v", got, want)
  453. }
  454. }
  455. func TestGetPodsForStatefulSetAdopt(t *testing.T) {
  456. set := newStatefulSet(5)
  457. pod1 := newStatefulSetPod(set, 1)
  458. // pod2 is an orphan with matching labels and name.
  459. pod2 := newStatefulSetPod(set, 2)
  460. pod2.OwnerReferences = nil
  461. // pod3 has wrong labels.
  462. pod3 := newStatefulSetPod(set, 3)
  463. pod3.OwnerReferences = nil
  464. pod3.Labels = nil
  465. // pod4 has wrong name.
  466. pod4 := newStatefulSetPod(set, 4)
  467. pod4.OwnerReferences = nil
  468. pod4.Name = "x" + pod4.Name
  469. ssc, spc := newFakeStatefulSetController(set, pod1, pod2, pod3, pod4)
  470. spc.podsIndexer.Add(pod1)
  471. spc.podsIndexer.Add(pod2)
  472. spc.podsIndexer.Add(pod3)
  473. spc.podsIndexer.Add(pod4)
  474. selector, err := metav1.LabelSelectorAsSelector(set.Spec.Selector)
  475. if err != nil {
  476. t.Fatal(err)
  477. }
  478. pods, err := ssc.getPodsForStatefulSet(set, selector)
  479. if err != nil {
  480. t.Fatalf("getPodsForStatefulSet() error: %v", err)
  481. }
  482. got := sets.NewString()
  483. for _, pod := range pods {
  484. got.Insert(pod.Name)
  485. }
  486. // pod2 should be claimed, pod3 and pod4 ignored
  487. want := sets.NewString(pod1.Name, pod2.Name)
  488. if !got.Equal(want) {
  489. t.Errorf("getPodsForStatefulSet() = %v, want %v", got, want)
  490. }
  491. }
  492. func TestGetPodsForStatefulSetRelease(t *testing.T) {
  493. set := newStatefulSet(3)
  494. ssc, spc := newFakeStatefulSetController(set)
  495. pod1 := newStatefulSetPod(set, 1)
  496. // pod2 is owned but has wrong name.
  497. pod2 := newStatefulSetPod(set, 2)
  498. pod2.Name = "x" + pod2.Name
  499. // pod3 is owned but has wrong labels.
  500. pod3 := newStatefulSetPod(set, 3)
  501. pod3.Labels = nil
  502. // pod4 is an orphan that doesn't match.
  503. pod4 := newStatefulSetPod(set, 4)
  504. pod4.OwnerReferences = nil
  505. pod4.Labels = nil
  506. spc.podsIndexer.Add(pod1)
  507. spc.podsIndexer.Add(pod2)
  508. spc.podsIndexer.Add(pod3)
  509. selector, err := metav1.LabelSelectorAsSelector(set.Spec.Selector)
  510. if err != nil {
  511. t.Fatal(err)
  512. }
  513. pods, err := ssc.getPodsForStatefulSet(set, selector)
  514. if err != nil {
  515. t.Fatalf("getPodsForStatefulSet() error: %v", err)
  516. }
  517. got := sets.NewString()
  518. for _, pod := range pods {
  519. got.Insert(pod.Name)
  520. }
  521. // Expect only pod1 (pod2 and pod3 should be released, pod4 ignored).
  522. want := sets.NewString(pod1.Name)
  523. if !got.Equal(want) {
  524. t.Errorf("getPodsForStatefulSet() = %v, want %v", got, want)
  525. }
  526. }
  527. func newFakeStatefulSetController(initialObjects ...runtime.Object) (*StatefulSetController, *fakeStatefulPodControl) {
  528. client := fake.NewSimpleClientset(initialObjects...)
  529. informerFactory := informers.NewSharedInformerFactory(client, controller.NoResyncPeriodFunc())
  530. fpc := newFakeStatefulPodControl(informerFactory.Core().V1().Pods(), informerFactory.Apps().V1().StatefulSets())
  531. ssu := newFakeStatefulSetStatusUpdater(informerFactory.Apps().V1().StatefulSets())
  532. ssc := NewStatefulSetController(
  533. informerFactory.Core().V1().Pods(),
  534. informerFactory.Apps().V1().StatefulSets(),
  535. informerFactory.Core().V1().PersistentVolumeClaims(),
  536. informerFactory.Apps().V1().ControllerRevisions(),
  537. client,
  538. )
  539. ssh := history.NewFakeHistory(informerFactory.Apps().V1().ControllerRevisions())
  540. ssc.podListerSynced = alwaysReady
  541. ssc.setListerSynced = alwaysReady
  542. recorder := record.NewFakeRecorder(10)
  543. ssc.control = NewDefaultStatefulSetControl(fpc, ssu, ssh, recorder)
  544. return ssc, fpc
  545. }
  546. func fakeWorker(ssc *StatefulSetController) {
  547. if obj, done := ssc.queue.Get(); !done {
  548. ssc.sync(obj.(string))
  549. ssc.queue.Done(obj)
  550. }
  551. }
  552. func getPodAtOrdinal(pods []*v1.Pod, ordinal int) *v1.Pod {
  553. if 0 > ordinal || ordinal >= len(pods) {
  554. return nil
  555. }
  556. sort.Sort(ascendingOrdinal(pods))
  557. return pods[ordinal]
  558. }
  559. func scaleUpStatefulSetController(set *apps.StatefulSet, ssc *StatefulSetController, spc *fakeStatefulPodControl) error {
  560. spc.setsIndexer.Add(set)
  561. ssc.enqueueStatefulSet(set)
  562. fakeWorker(ssc)
  563. selector, err := metav1.LabelSelectorAsSelector(set.Spec.Selector)
  564. if err != nil {
  565. return err
  566. }
  567. for set.Status.ReadyReplicas < *set.Spec.Replicas {
  568. pods, err := spc.podsLister.Pods(set.Namespace).List(selector)
  569. if err != nil {
  570. return err
  571. }
  572. ord := len(pods) - 1
  573. pod := getPodAtOrdinal(pods, ord)
  574. if pods, err = spc.setPodPending(set, ord); err != nil {
  575. return err
  576. }
  577. pod = getPodAtOrdinal(pods, ord)
  578. ssc.addPod(pod)
  579. fakeWorker(ssc)
  580. pod = getPodAtOrdinal(pods, ord)
  581. prev := *pod
  582. if pods, err = spc.setPodRunning(set, ord); err != nil {
  583. return err
  584. }
  585. pod = getPodAtOrdinal(pods, ord)
  586. ssc.updatePod(&prev, pod)
  587. fakeWorker(ssc)
  588. pod = getPodAtOrdinal(pods, ord)
  589. prev = *pod
  590. if pods, err = spc.setPodReady(set, ord); err != nil {
  591. return err
  592. }
  593. pod = getPodAtOrdinal(pods, ord)
  594. ssc.updatePod(&prev, pod)
  595. fakeWorker(ssc)
  596. if err := assertMonotonicInvariants(set, spc); err != nil {
  597. return err
  598. }
  599. if obj, _, err := spc.setsIndexer.Get(set); err != nil {
  600. return err
  601. } else {
  602. set = obj.(*apps.StatefulSet)
  603. }
  604. }
  605. return assertMonotonicInvariants(set, spc)
  606. }
  607. func scaleDownStatefulSetController(set *apps.StatefulSet, ssc *StatefulSetController, spc *fakeStatefulPodControl) error {
  608. selector, err := metav1.LabelSelectorAsSelector(set.Spec.Selector)
  609. if err != nil {
  610. return err
  611. }
  612. pods, err := spc.podsLister.Pods(set.Namespace).List(selector)
  613. if err != nil {
  614. return err
  615. }
  616. ord := len(pods) - 1
  617. pod := getPodAtOrdinal(pods, ord)
  618. prev := *pod
  619. fakeResourceVersion(set)
  620. spc.setsIndexer.Add(set)
  621. ssc.enqueueStatefulSet(set)
  622. fakeWorker(ssc)
  623. pods, err = spc.addTerminatingPod(set, ord)
  624. if err != nil {
  625. return err
  626. }
  627. pod = getPodAtOrdinal(pods, ord)
  628. ssc.updatePod(&prev, pod)
  629. fakeWorker(ssc)
  630. spc.DeleteStatefulPod(set, pod)
  631. ssc.deletePod(pod)
  632. fakeWorker(ssc)
  633. for set.Status.Replicas > *set.Spec.Replicas {
  634. pods, err = spc.podsLister.Pods(set.Namespace).List(selector)
  635. if err != nil {
  636. return err
  637. }
  638. ord := len(pods)
  639. pods, err = spc.addTerminatingPod(set, ord)
  640. if err != nil {
  641. return err
  642. }
  643. pod = getPodAtOrdinal(pods, ord)
  644. ssc.updatePod(&prev, pod)
  645. fakeWorker(ssc)
  646. spc.DeleteStatefulPod(set, pod)
  647. ssc.deletePod(pod)
  648. fakeWorker(ssc)
  649. if obj, _, err := spc.setsIndexer.Get(set); err != nil {
  650. return err
  651. } else {
  652. set = obj.(*apps.StatefulSet)
  653. }
  654. }
  655. return assertMonotonicInvariants(set, spc)
  656. }