mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-27 05:27:21 +00:00
184 lines
4.5 KiB
Go
184 lines
4.5 KiB
Go
/*
|
|
Copyright 2018 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 pod
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"reflect"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
v1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
"k8s.io/client-go/kubernetes/fake"
|
|
)
|
|
|
|
func TestPatchPodStatus(t *testing.T) {
|
|
ns := "ns"
|
|
name := "name"
|
|
uid := types.UID("myuid")
|
|
client := &fake.Clientset{}
|
|
client.CoreV1().Pods(ns).Create(context.TODO(), &v1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: ns,
|
|
Name: name,
|
|
},
|
|
}, metav1.CreateOptions{})
|
|
|
|
testCases := []struct {
|
|
description string
|
|
mutate func(input v1.PodStatus) v1.PodStatus
|
|
expectUnchanged bool
|
|
expectedPatchBytes []byte
|
|
}{
|
|
{
|
|
"no change",
|
|
func(input v1.PodStatus) v1.PodStatus { return input },
|
|
true,
|
|
[]byte(fmt.Sprintf(`{"metadata":{"uid":"myuid"}}`)),
|
|
},
|
|
{
|
|
"message change",
|
|
func(input v1.PodStatus) v1.PodStatus {
|
|
input.Message = "random message"
|
|
return input
|
|
},
|
|
false,
|
|
[]byte(fmt.Sprintf(`{"metadata":{"uid":"myuid"},"status":{"message":"random message"}}`)),
|
|
},
|
|
{
|
|
"pod condition change",
|
|
func(input v1.PodStatus) v1.PodStatus {
|
|
input.Conditions[0].Status = v1.ConditionFalse
|
|
return input
|
|
},
|
|
false,
|
|
[]byte(fmt.Sprintf(`{"metadata":{"uid":"myuid"},"status":{"$setElementOrder/conditions":[{"type":"Ready"},{"type":"PodScheduled"}],"conditions":[{"status":"False","type":"Ready"}]}}`)),
|
|
},
|
|
{
|
|
"additional init container condition",
|
|
func(input v1.PodStatus) v1.PodStatus {
|
|
input.InitContainerStatuses = []v1.ContainerStatus{
|
|
{
|
|
Name: "init-container",
|
|
Ready: true,
|
|
},
|
|
}
|
|
return input
|
|
},
|
|
false,
|
|
[]byte(fmt.Sprintf(`{"metadata":{"uid":"myuid"},"status":{"initContainerStatuses":[{"image":"","imageID":"","lastState":{},"name":"init-container","ready":true,"restartCount":0,"state":{}}]}}`)),
|
|
},
|
|
}
|
|
for _, tc := range testCases {
|
|
t.Run(tc.description, func(t *testing.T) {
|
|
_, patchBytes, unchanged, err := PatchPodStatus(context.TODO(), client, ns, name, uid, getPodStatus(), tc.mutate(getPodStatus()))
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if unchanged != tc.expectUnchanged {
|
|
t.Errorf("unexpected change: %t", unchanged)
|
|
}
|
|
if !reflect.DeepEqual(patchBytes, tc.expectedPatchBytes) {
|
|
t.Errorf("expect patchBytes: %q, got: %q\n", tc.expectedPatchBytes, patchBytes)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func getPodStatus() v1.PodStatus {
|
|
return v1.PodStatus{
|
|
Phase: v1.PodRunning,
|
|
Conditions: []v1.PodCondition{
|
|
{
|
|
Type: v1.PodReady,
|
|
Status: v1.ConditionTrue,
|
|
},
|
|
{
|
|
Type: v1.PodScheduled,
|
|
Status: v1.ConditionTrue,
|
|
},
|
|
},
|
|
ContainerStatuses: []v1.ContainerStatus{
|
|
{
|
|
Name: "container1",
|
|
Ready: true,
|
|
},
|
|
{
|
|
Name: "container2",
|
|
Ready: true,
|
|
},
|
|
},
|
|
Message: "Message",
|
|
}
|
|
}
|
|
|
|
func TestReplaceOrAppendPodCondition(t *testing.T) {
|
|
cType := v1.PodConditionType("ExampleType")
|
|
testCases := []struct {
|
|
description string
|
|
conditions []v1.PodCondition
|
|
condition v1.PodCondition
|
|
wantConditions []v1.PodCondition
|
|
}{
|
|
{
|
|
description: "append",
|
|
conditions: []v1.PodCondition{},
|
|
condition: v1.PodCondition{
|
|
Type: cType,
|
|
Status: v1.ConditionTrue,
|
|
},
|
|
wantConditions: []v1.PodCondition{
|
|
{
|
|
Type: cType,
|
|
Status: v1.ConditionTrue,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
description: "replace",
|
|
conditions: []v1.PodCondition{
|
|
{
|
|
Type: cType,
|
|
Status: v1.ConditionTrue,
|
|
},
|
|
},
|
|
condition: v1.PodCondition{
|
|
Type: cType,
|
|
Status: v1.ConditionFalse,
|
|
},
|
|
wantConditions: []v1.PodCondition{
|
|
{
|
|
Type: cType,
|
|
Status: v1.ConditionFalse,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for _, tc := range testCases {
|
|
t.Run(tc.description, func(t *testing.T) {
|
|
gotConditions := ReplaceOrAppendPodCondition(tc.conditions, &tc.condition)
|
|
if diff := cmp.Diff(tc.wantConditions, gotConditions); diff != "" {
|
|
t.Errorf("Unexpected conditions: %s", diff)
|
|
}
|
|
})
|
|
}
|
|
}
|