| /* |
| Copyright 2014 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 status |
| |
| import ( |
| "reflect" |
| "testing" |
| |
| "github.com/stretchr/testify/assert" |
| "k8s.io/api/core/v1" |
| ) |
| |
| func TestGenerateContainersReadyCondition(t *testing.T) { |
| tests := []struct { |
| spec *v1.PodSpec |
| containerStatuses []v1.ContainerStatus |
| podPhase v1.PodPhase |
| expectReady v1.PodCondition |
| }{ |
| { |
| spec: nil, |
| containerStatuses: nil, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.ContainersReady, v1.ConditionFalse, UnknownContainerStatuses, ""), |
| }, |
| { |
| spec: &v1.PodSpec{}, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.ContainersReady, v1.ConditionTrue, "", ""), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| }, |
| }, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.ContainersReady, v1.ConditionFalse, ContainersNotReady, "containers with unknown status: [1234]"), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| {Name: "5678"}, |
| }, |
| }, |
| containerStatuses: []v1.ContainerStatus{ |
| getReadyStatus("1234"), |
| getReadyStatus("5678"), |
| }, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.ContainersReady, v1.ConditionTrue, "", ""), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| {Name: "5678"}, |
| }, |
| }, |
| containerStatuses: []v1.ContainerStatus{ |
| getReadyStatus("1234"), |
| }, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.ContainersReady, v1.ConditionFalse, ContainersNotReady, "containers with unknown status: [5678]"), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| {Name: "5678"}, |
| }, |
| }, |
| containerStatuses: []v1.ContainerStatus{ |
| getReadyStatus("1234"), |
| getNotReadyStatus("5678"), |
| }, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.ContainersReady, v1.ConditionFalse, ContainersNotReady, "containers with unready status: [5678]"), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| }, |
| }, |
| containerStatuses: []v1.ContainerStatus{ |
| getNotReadyStatus("1234"), |
| }, |
| podPhase: v1.PodSucceeded, |
| expectReady: getPodCondition(v1.ContainersReady, v1.ConditionFalse, PodCompleted, ""), |
| }, |
| } |
| |
| for i, test := range tests { |
| ready := GenerateContainersReadyCondition(test.spec, test.containerStatuses, test.podPhase) |
| if !reflect.DeepEqual(ready, test.expectReady) { |
| t.Errorf("On test case %v, expectReady:\n%+v\ngot\n%+v\n", i, test.expectReady, ready) |
| } |
| } |
| } |
| |
| func TestGeneratePodReadyCondition(t *testing.T) { |
| tests := []struct { |
| spec *v1.PodSpec |
| conditions []v1.PodCondition |
| containerStatuses []v1.ContainerStatus |
| podPhase v1.PodPhase |
| expectReady v1.PodCondition |
| }{ |
| { |
| spec: nil, |
| conditions: nil, |
| containerStatuses: nil, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionFalse, UnknownContainerStatuses, ""), |
| }, |
| { |
| spec: &v1.PodSpec{}, |
| conditions: nil, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionTrue, "", ""), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| }, |
| }, |
| conditions: nil, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionFalse, ContainersNotReady, "containers with unknown status: [1234]"), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| {Name: "5678"}, |
| }, |
| }, |
| conditions: nil, |
| containerStatuses: []v1.ContainerStatus{ |
| getReadyStatus("1234"), |
| getReadyStatus("5678"), |
| }, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionTrue, "", ""), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| {Name: "5678"}, |
| }, |
| }, |
| conditions: nil, |
| containerStatuses: []v1.ContainerStatus{ |
| getReadyStatus("1234"), |
| }, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionFalse, ContainersNotReady, "containers with unknown status: [5678]"), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| {Name: "5678"}, |
| }, |
| }, |
| conditions: nil, |
| containerStatuses: []v1.ContainerStatus{ |
| getReadyStatus("1234"), |
| getNotReadyStatus("5678"), |
| }, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionFalse, ContainersNotReady, "containers with unready status: [5678]"), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| }, |
| }, |
| conditions: nil, |
| containerStatuses: []v1.ContainerStatus{ |
| getNotReadyStatus("1234"), |
| }, |
| podPhase: v1.PodSucceeded, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionFalse, PodCompleted, ""), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| ReadinessGates: []v1.PodReadinessGate{ |
| {ConditionType: v1.PodConditionType("gate1")}, |
| }, |
| }, |
| conditions: nil, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionFalse, ReadinessGatesNotReady, `corresponding condition of pod readiness gate "gate1" does not exist.`), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| ReadinessGates: []v1.PodReadinessGate{ |
| {ConditionType: v1.PodConditionType("gate1")}, |
| }, |
| }, |
| conditions: []v1.PodCondition{ |
| getPodCondition("gate1", v1.ConditionFalse, "", ""), |
| }, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionFalse, ReadinessGatesNotReady, `the status of pod readiness gate "gate1" is not "True", but False`), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| ReadinessGates: []v1.PodReadinessGate{ |
| {ConditionType: v1.PodConditionType("gate1")}, |
| }, |
| }, |
| conditions: []v1.PodCondition{ |
| getPodCondition("gate1", v1.ConditionTrue, "", ""), |
| }, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionTrue, "", ""), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| ReadinessGates: []v1.PodReadinessGate{ |
| {ConditionType: v1.PodConditionType("gate1")}, |
| {ConditionType: v1.PodConditionType("gate2")}, |
| }, |
| }, |
| conditions: []v1.PodCondition{ |
| getPodCondition("gate1", v1.ConditionTrue, "", ""), |
| }, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionFalse, ReadinessGatesNotReady, `corresponding condition of pod readiness gate "gate2" does not exist.`), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| ReadinessGates: []v1.PodReadinessGate{ |
| {ConditionType: v1.PodConditionType("gate1")}, |
| {ConditionType: v1.PodConditionType("gate2")}, |
| }, |
| }, |
| conditions: []v1.PodCondition{ |
| getPodCondition("gate1", v1.ConditionTrue, "", ""), |
| getPodCondition("gate2", v1.ConditionFalse, "", ""), |
| }, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionFalse, ReadinessGatesNotReady, `the status of pod readiness gate "gate2" is not "True", but False`), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| ReadinessGates: []v1.PodReadinessGate{ |
| {ConditionType: v1.PodConditionType("gate1")}, |
| {ConditionType: v1.PodConditionType("gate2")}, |
| }, |
| }, |
| conditions: []v1.PodCondition{ |
| getPodCondition("gate1", v1.ConditionTrue, "", ""), |
| getPodCondition("gate2", v1.ConditionTrue, "", ""), |
| }, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionTrue, "", ""), |
| }, |
| { |
| spec: &v1.PodSpec{ |
| Containers: []v1.Container{ |
| {Name: "1234"}, |
| }, |
| ReadinessGates: []v1.PodReadinessGate{ |
| {ConditionType: v1.PodConditionType("gate1")}, |
| }, |
| }, |
| conditions: []v1.PodCondition{ |
| getPodCondition("gate1", v1.ConditionTrue, "", ""), |
| }, |
| containerStatuses: []v1.ContainerStatus{getNotReadyStatus("1234")}, |
| podPhase: v1.PodRunning, |
| expectReady: getPodCondition(v1.PodReady, v1.ConditionFalse, ContainersNotReady, "containers with unready status: [1234]"), |
| }, |
| } |
| |
| for i, test := range tests { |
| ready := GeneratePodReadyCondition(test.spec, test.conditions, test.containerStatuses, test.podPhase) |
| if !reflect.DeepEqual(ready, test.expectReady) { |
| t.Errorf("On test case %v, expectReady:\n%+v\ngot\n%+v\n", i, test.expectReady, ready) |
| } |
| } |
| } |
| |
| func TestGeneratePodInitializedCondition(t *testing.T) { |
| noInitContainer := &v1.PodSpec{} |
| oneInitContainer := &v1.PodSpec{ |
| InitContainers: []v1.Container{ |
| {Name: "1234"}, |
| }, |
| } |
| twoInitContainer := &v1.PodSpec{ |
| InitContainers: []v1.Container{ |
| {Name: "1234"}, |
| {Name: "5678"}, |
| }, |
| } |
| tests := []struct { |
| spec *v1.PodSpec |
| containerStatuses []v1.ContainerStatus |
| podPhase v1.PodPhase |
| expected v1.PodCondition |
| }{ |
| { |
| spec: twoInitContainer, |
| containerStatuses: nil, |
| podPhase: v1.PodRunning, |
| expected: v1.PodCondition{ |
| Status: v1.ConditionFalse, |
| Reason: UnknownContainerStatuses, |
| }, |
| }, |
| { |
| spec: noInitContainer, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expected: v1.PodCondition{ |
| Status: v1.ConditionTrue, |
| Reason: "", |
| }, |
| }, |
| { |
| spec: oneInitContainer, |
| containerStatuses: []v1.ContainerStatus{}, |
| podPhase: v1.PodRunning, |
| expected: v1.PodCondition{ |
| Status: v1.ConditionFalse, |
| Reason: ContainersNotInitialized, |
| }, |
| }, |
| { |
| spec: twoInitContainer, |
| containerStatuses: []v1.ContainerStatus{ |
| getReadyStatus("1234"), |
| getReadyStatus("5678"), |
| }, |
| podPhase: v1.PodRunning, |
| expected: v1.PodCondition{ |
| Status: v1.ConditionTrue, |
| Reason: "", |
| }, |
| }, |
| { |
| spec: twoInitContainer, |
| containerStatuses: []v1.ContainerStatus{ |
| getReadyStatus("1234"), |
| }, |
| podPhase: v1.PodRunning, |
| expected: v1.PodCondition{ |
| Status: v1.ConditionFalse, |
| Reason: ContainersNotInitialized, |
| }, |
| }, |
| { |
| spec: twoInitContainer, |
| containerStatuses: []v1.ContainerStatus{ |
| getReadyStatus("1234"), |
| getNotReadyStatus("5678"), |
| }, |
| podPhase: v1.PodRunning, |
| expected: v1.PodCondition{ |
| Status: v1.ConditionFalse, |
| Reason: ContainersNotInitialized, |
| }, |
| }, |
| { |
| spec: oneInitContainer, |
| containerStatuses: []v1.ContainerStatus{ |
| getReadyStatus("1234"), |
| }, |
| podPhase: v1.PodSucceeded, |
| expected: v1.PodCondition{ |
| Status: v1.ConditionTrue, |
| Reason: PodCompleted, |
| }, |
| }, |
| } |
| |
| for _, test := range tests { |
| test.expected.Type = v1.PodInitialized |
| condition := GeneratePodInitializedCondition(test.spec, test.containerStatuses, test.podPhase) |
| assert.Equal(t, test.expected.Type, condition.Type) |
| assert.Equal(t, test.expected.Status, condition.Status) |
| assert.Equal(t, test.expected.Reason, condition.Reason) |
| |
| } |
| } |
| |
| func getPodCondition(conditionType v1.PodConditionType, status v1.ConditionStatus, reason, message string) v1.PodCondition { |
| return v1.PodCondition{ |
| Type: conditionType, |
| Status: status, |
| Reason: reason, |
| Message: message, |
| } |
| } |
| |
| func getReadyStatus(cName string) v1.ContainerStatus { |
| return v1.ContainerStatus{ |
| Name: cName, |
| Ready: true, |
| } |
| } |
| |
| func getNotReadyStatus(cName string) v1.ContainerStatus { |
| return v1.ContainerStatus{ |
| Name: cName, |
| Ready: false, |
| } |
| } |
