/pkg/kubectl/rollout_status_test.go
Go | 449 lines | 409 code | 25 blank | 15 comment | 29 complexity | f33879e53e4fed2e9f7191056105e764 MD5 | raw file
- /*
- Copyright 2016 The Kubernetes Authors.
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
- http://www.apache.org/licenses/LICENSE-2.0
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- */
- package kubectl
- import (
- "fmt"
- "testing"
- apps "k8s.io/api/apps/v1beta1"
- api "k8s.io/api/core/v1"
- extensions "k8s.io/api/extensions/v1beta1"
- metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
- "k8s.io/client-go/kubernetes/fake"
- )
- func TestDeploymentStatusViewerStatus(t *testing.T) {
- tests := []struct {
- generation int64
- specReplicas int32
- status extensions.DeploymentStatus
- msg string
- done bool
- }{
- {
- generation: 0,
- specReplicas: 1,
- status: extensions.DeploymentStatus{
- ObservedGeneration: 1,
- Replicas: 1,
- UpdatedReplicas: 0,
- AvailableReplicas: 1,
- UnavailableReplicas: 0,
- },
- msg: "Waiting for rollout to finish: 0 out of 1 new replicas have been updated...\n",
- done: false,
- },
- {
- generation: 1,
- specReplicas: 1,
- status: extensions.DeploymentStatus{
- ObservedGeneration: 1,
- Replicas: 2,
- UpdatedReplicas: 1,
- AvailableReplicas: 2,
- UnavailableReplicas: 0,
- },
- msg: "Waiting for rollout to finish: 1 old replicas are pending termination...\n",
- done: false,
- },
- {
- generation: 1,
- specReplicas: 2,
- status: extensions.DeploymentStatus{
- ObservedGeneration: 1,
- Replicas: 2,
- UpdatedReplicas: 2,
- AvailableReplicas: 1,
- UnavailableReplicas: 1,
- },
- msg: "Waiting for rollout to finish: 1 of 2 updated replicas are available...\n",
- done: false,
- },
- {
- generation: 1,
- specReplicas: 2,
- status: extensions.DeploymentStatus{
- ObservedGeneration: 1,
- Replicas: 2,
- UpdatedReplicas: 2,
- AvailableReplicas: 2,
- UnavailableReplicas: 0,
- },
- msg: "deployment \"foo\" successfully rolled out\n",
- done: true,
- },
- {
- generation: 2,
- specReplicas: 2,
- status: extensions.DeploymentStatus{
- ObservedGeneration: 1,
- Replicas: 2,
- UpdatedReplicas: 2,
- AvailableReplicas: 2,
- UnavailableReplicas: 0,
- },
- msg: "Waiting for deployment spec update to be observed...\n",
- done: false,
- },
- }
- for _, test := range tests {
- d := &extensions.Deployment{
- ObjectMeta: metav1.ObjectMeta{
- Namespace: "bar",
- Name: "foo",
- UID: "8764ae47-9092-11e4-8393-42010af018ff",
- Generation: test.generation,
- },
- Spec: extensions.DeploymentSpec{
- Replicas: &test.specReplicas,
- },
- Status: test.status,
- }
- client := fake.NewSimpleClientset(d).Extensions()
- dsv := &DeploymentStatusViewer{c: client}
- msg, done, err := dsv.Status("bar", "foo", 0)
- if err != nil {
- t.Fatalf("DeploymentStatusViewer.Status(): %v", err)
- }
- if done != test.done || msg != test.msg {
- t.Errorf("DeploymentStatusViewer.Status() for deployment with generation %d, %d replicas specified, and status %+v returned %q, %t, want %q, %t",
- test.generation,
- test.specReplicas,
- test.status,
- msg,
- done,
- test.msg,
- test.done,
- )
- }
- }
- }
- func TestDaemonSetStatusViewerStatus(t *testing.T) {
- tests := []struct {
- generation int64
- status extensions.DaemonSetStatus
- msg string
- done bool
- }{
- {
- generation: 0,
- status: extensions.DaemonSetStatus{
- ObservedGeneration: 1,
- UpdatedNumberScheduled: 0,
- DesiredNumberScheduled: 1,
- NumberAvailable: 0,
- },
- msg: "Waiting for rollout to finish: 0 out of 1 new pods have been updated...\n",
- done: false,
- },
- {
- generation: 1,
- status: extensions.DaemonSetStatus{
- ObservedGeneration: 1,
- UpdatedNumberScheduled: 2,
- DesiredNumberScheduled: 2,
- NumberAvailable: 1,
- },
- msg: "Waiting for rollout to finish: 1 of 2 updated pods are available...\n",
- done: false,
- },
- {
- generation: 1,
- status: extensions.DaemonSetStatus{
- ObservedGeneration: 1,
- UpdatedNumberScheduled: 2,
- DesiredNumberScheduled: 2,
- NumberAvailable: 2,
- },
- msg: "daemon set \"foo\" successfully rolled out\n",
- done: true,
- },
- {
- generation: 2,
- status: extensions.DaemonSetStatus{
- ObservedGeneration: 1,
- UpdatedNumberScheduled: 2,
- DesiredNumberScheduled: 2,
- NumberAvailable: 2,
- },
- msg: "Waiting for daemon set spec update to be observed...\n",
- done: false,
- },
- }
- for i := range tests {
- test := tests[i]
- t.Logf("testing scenario %d", i)
- d := &extensions.DaemonSet{
- ObjectMeta: metav1.ObjectMeta{
- Namespace: "bar",
- Name: "foo",
- UID: "8764ae47-9092-11e4-8393-42010af018ff",
- Generation: test.generation,
- },
- Spec: extensions.DaemonSetSpec{
- UpdateStrategy: extensions.DaemonSetUpdateStrategy{
- Type: extensions.RollingUpdateDaemonSetStrategyType,
- },
- },
- Status: test.status,
- }
- client := fake.NewSimpleClientset(d).Extensions()
- dsv := &DaemonSetStatusViewer{c: client}
- msg, done, err := dsv.Status("bar", "foo", 0)
- if err != nil {
- t.Fatalf("unexpected error: %v", err)
- }
- if done != test.done || msg != test.msg {
- t.Errorf("daemon set with generation %d, %d pods specified, and status:\n%+v\nreturned:\n%q, %t\nwant:\n%q, %t",
- test.generation,
- d.Status.DesiredNumberScheduled,
- test.status,
- msg,
- done,
- test.msg,
- test.done,
- )
- }
- }
- }
- func TestStatefulSetStatusViewerStatus(t *testing.T) {
- tests := []struct {
- name string
- generation int64
- strategy apps.StatefulSetUpdateStrategy
- status apps.StatefulSetStatus
- msg string
- done bool
- err bool
- }{
- {
- name: "on delete returns an error",
- generation: 1,
- strategy: apps.StatefulSetUpdateStrategy{Type: apps.OnDeleteStatefulSetStrategyType},
- status: apps.StatefulSetStatus{
- ObservedGeneration: func() *int64 {
- generation := int64(1)
- return &generation
- }(),
- Replicas: 0,
- ReadyReplicas: 1,
- CurrentReplicas: 0,
- UpdatedReplicas: 0,
- },
- msg: "",
- done: true,
- err: true,
- },
- {
- name: "unobserved update is not complete",
- generation: 2,
- strategy: apps.StatefulSetUpdateStrategy{Type: apps.RollingUpdateStatefulSetStrategyType},
- status: apps.StatefulSetStatus{
- ObservedGeneration: func() *int64 {
- generation := int64(1)
- return &generation
- }(),
- Replicas: 3,
- ReadyReplicas: 3,
- CurrentReplicas: 3,
- UpdatedReplicas: 0,
- },
- msg: "Waiting for statefulset spec update to be observed...\n",
- done: false,
- err: false,
- },
- {
- name: "if all pods are not ready the update is not complete",
- generation: 1,
- strategy: apps.StatefulSetUpdateStrategy{Type: apps.RollingUpdateStatefulSetStrategyType},
- status: apps.StatefulSetStatus{
- ObservedGeneration: func() *int64 {
- generation := int64(2)
- return &generation
- }(),
- Replicas: 3,
- ReadyReplicas: 2,
- CurrentReplicas: 3,
- UpdatedReplicas: 0,
- },
- msg: fmt.Sprintf("Waiting for %d pods to be ready...\n", 1),
- done: false,
- err: false,
- },
- {
- name: "partition update completes when all replicas above the partition are updated",
- generation: 1,
- strategy: apps.StatefulSetUpdateStrategy{Type: apps.RollingUpdateStatefulSetStrategyType,
- RollingUpdate: func() *apps.RollingUpdateStatefulSetStrategy {
- partition := int32(2)
- return &apps.RollingUpdateStatefulSetStrategy{Partition: &partition}
- }()},
- status: apps.StatefulSetStatus{
- ObservedGeneration: func() *int64 {
- generation := int64(2)
- return &generation
- }(),
- Replicas: 3,
- ReadyReplicas: 3,
- CurrentReplicas: 2,
- UpdatedReplicas: 1,
- },
- msg: fmt.Sprintf("partitioned roll out complete: %d new pods have been updated...\n", 1),
- done: true,
- err: false,
- },
- {
- name: "partition update is in progress if all pods above the partition have not been updated",
- generation: 1,
- strategy: apps.StatefulSetUpdateStrategy{Type: apps.RollingUpdateStatefulSetStrategyType,
- RollingUpdate: func() *apps.RollingUpdateStatefulSetStrategy {
- partition := int32(2)
- return &apps.RollingUpdateStatefulSetStrategy{Partition: &partition}
- }()},
- status: apps.StatefulSetStatus{
- ObservedGeneration: func() *int64 {
- generation := int64(2)
- return &generation
- }(),
- Replicas: 3,
- ReadyReplicas: 3,
- CurrentReplicas: 3,
- UpdatedReplicas: 0,
- },
- msg: fmt.Sprintf("Waiting for partitioned roll out to finish: %d out of %d new pods have been updated...\n", 0, 1),
- done: true,
- err: false,
- },
- {
- name: "update completes when all replicas are current",
- generation: 1,
- strategy: apps.StatefulSetUpdateStrategy{Type: apps.RollingUpdateStatefulSetStrategyType},
- status: apps.StatefulSetStatus{
- ObservedGeneration: func() *int64 {
- generation := int64(2)
- return &generation
- }(),
- Replicas: 3,
- ReadyReplicas: 3,
- CurrentReplicas: 3,
- UpdatedReplicas: 3,
- CurrentRevision: "foo",
- UpdateRevision: "foo",
- },
- msg: fmt.Sprintf("statefulset rolling update complete %d pods at revision %s...\n", 3, "foo"),
- done: true,
- err: false,
- },
- }
- for i := range tests {
- test := tests[i]
- s := newStatefulSet(3)
- s.Status = test.status
- s.Spec.UpdateStrategy = test.strategy
- s.Generation = test.generation
- client := fake.NewSimpleClientset(s).AppsV1beta1()
- dsv := &StatefulSetStatusViewer{c: client}
- msg, done, err := dsv.Status(s.Namespace, s.Name, 0)
- if test.err && err == nil {
- t.Fatalf("%s: expected error", test.name)
- }
- if !test.err && err != nil {
- t.Fatalf("%s: %s", test.name, err)
- }
- if done && !test.done {
- t.Errorf("%s: want done %v got %v", test.name, done, test.done)
- }
- if msg != test.msg {
- t.Errorf("%s: want message %s got %s", test.name, test.msg, msg)
- }
- }
- }
- func TestDaemonSetStatusViewerStatusWithWrongUpdateStrategyType(t *testing.T) {
- d := &extensions.DaemonSet{
- ObjectMeta: metav1.ObjectMeta{
- Namespace: "bar",
- Name: "foo",
- UID: "8764ae47-9092-11e4-8393-42010af018ff",
- },
- Spec: extensions.DaemonSetSpec{
- UpdateStrategy: extensions.DaemonSetUpdateStrategy{
- Type: extensions.OnDeleteDaemonSetStrategyType,
- },
- },
- }
- client := fake.NewSimpleClientset(d).Extensions()
- dsv := &DaemonSetStatusViewer{c: client}
- msg, done, err := dsv.Status("bar", "foo", 0)
- errMsg := "Status is available only for RollingUpdate strategy type"
- if err == nil || err.Error() != errMsg {
- t.Errorf("Status for daemon sets with UpdateStrategy type different than RollingUpdate should return error. Instead got: msg: %s\ndone: %t\n err: %v", msg, done, err)
- }
- }
- func newStatefulSet(replicas int32) *apps.StatefulSet {
- return &apps.StatefulSet{
- ObjectMeta: metav1.ObjectMeta{
- Name: "foo",
- Namespace: metav1.NamespaceDefault,
- Labels: map[string]string{"a": "b"},
- },
- Spec: apps.StatefulSetSpec{
- PodManagementPolicy: apps.OrderedReadyPodManagement,
- Selector: &metav1.LabelSelector{MatchLabels: map[string]string{"a": "b"}},
- Template: api.PodTemplateSpec{
- ObjectMeta: metav1.ObjectMeta{
- Labels: map[string]string{"a": "b"},
- },
- Spec: api.PodSpec{
- Containers: []api.Container{
- {
- Name: "test",
- Image: "test_image",
- ImagePullPolicy: api.PullIfNotPresent,
- },
- },
- RestartPolicy: api.RestartPolicyAlways,
- DNSPolicy: api.DNSClusterFirst,
- },
- },
- Replicas: &replicas,
- UpdateStrategy: apps.StatefulSetUpdateStrategy{Type: apps.RollingUpdateStatefulSetStrategyType},
- },
- Status: apps.StatefulSetStatus{},
- }
- }