mirror of
https://github.com/optim-enterprises-bv/kubernetes.git
synced 2025-11-04 04:08:16 +00:00
478 lines
10 KiB
Go
478 lines
10 KiB
Go
/*
|
|
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 kubectl
|
|
|
|
import (
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
corev1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/kubernetes/pkg/kubectl/scheme"
|
|
)
|
|
|
|
func encodeOrDie(obj runtime.Object) []byte {
|
|
data, err := runtime.Encode(scheme.Codecs.LegacyCodec(corev1.SchemeGroupVersion), obj)
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
return data
|
|
}
|
|
|
|
func TestSortingPrinter(t *testing.T) {
|
|
intPtr := func(val int32) *int32 { return &val }
|
|
|
|
a := &corev1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "a",
|
|
},
|
|
}
|
|
|
|
b := &corev1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "b",
|
|
},
|
|
}
|
|
|
|
c := &corev1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "c",
|
|
},
|
|
}
|
|
|
|
tests := []struct {
|
|
obj runtime.Object
|
|
sort runtime.Object
|
|
field string
|
|
name string
|
|
expectedErr string
|
|
}{
|
|
{
|
|
name: "in-order-already",
|
|
obj: &corev1.PodList{
|
|
Items: []corev1.Pod{
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "a",
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "b",
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "c",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
sort: &corev1.PodList{
|
|
Items: []corev1.Pod{
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "a",
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "b",
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "c",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
field: "{.metadata.name}",
|
|
},
|
|
{
|
|
name: "reverse-order",
|
|
obj: &corev1.PodList{
|
|
Items: []corev1.Pod{
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "b",
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "c",
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "a",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
sort: &corev1.PodList{
|
|
Items: []corev1.Pod{
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "a",
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "b",
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "c",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
field: "{.metadata.name}",
|
|
},
|
|
{
|
|
name: "random-order-timestamp",
|
|
obj: &corev1.PodList{
|
|
Items: []corev1.Pod{
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
CreationTimestamp: metav1.Unix(300, 0),
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
CreationTimestamp: metav1.Unix(100, 0),
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
CreationTimestamp: metav1.Unix(200, 0),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
sort: &corev1.PodList{
|
|
Items: []corev1.Pod{
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
CreationTimestamp: metav1.Unix(100, 0),
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
CreationTimestamp: metav1.Unix(200, 0),
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
CreationTimestamp: metav1.Unix(300, 0),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
field: "{.metadata.creationTimestamp}",
|
|
},
|
|
{
|
|
name: "random-order-numbers",
|
|
obj: &corev1.ReplicationControllerList{
|
|
Items: []corev1.ReplicationController{
|
|
{
|
|
Spec: corev1.ReplicationControllerSpec{
|
|
Replicas: intPtr(5),
|
|
},
|
|
},
|
|
{
|
|
Spec: corev1.ReplicationControllerSpec{
|
|
Replicas: intPtr(1),
|
|
},
|
|
},
|
|
{
|
|
Spec: corev1.ReplicationControllerSpec{
|
|
Replicas: intPtr(9),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
sort: &corev1.ReplicationControllerList{
|
|
Items: []corev1.ReplicationController{
|
|
{
|
|
Spec: corev1.ReplicationControllerSpec{
|
|
Replicas: intPtr(1),
|
|
},
|
|
},
|
|
{
|
|
Spec: corev1.ReplicationControllerSpec{
|
|
Replicas: intPtr(5),
|
|
},
|
|
},
|
|
{
|
|
Spec: corev1.ReplicationControllerSpec{
|
|
Replicas: intPtr(9),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
field: "{.spec.replicas}",
|
|
},
|
|
{
|
|
name: "v1.List in order",
|
|
obj: &corev1.List{
|
|
Items: []runtime.RawExtension{
|
|
{Raw: encodeOrDie(a)},
|
|
{Raw: encodeOrDie(b)},
|
|
{Raw: encodeOrDie(c)},
|
|
},
|
|
},
|
|
sort: &corev1.List{
|
|
Items: []runtime.RawExtension{
|
|
{Raw: encodeOrDie(a)},
|
|
{Raw: encodeOrDie(b)},
|
|
{Raw: encodeOrDie(c)},
|
|
},
|
|
},
|
|
field: "{.metadata.name}",
|
|
},
|
|
{
|
|
name: "v1.List in reverse",
|
|
obj: &corev1.List{
|
|
Items: []runtime.RawExtension{
|
|
{Raw: encodeOrDie(c)},
|
|
{Raw: encodeOrDie(b)},
|
|
{Raw: encodeOrDie(a)},
|
|
},
|
|
},
|
|
sort: &corev1.List{
|
|
Items: []runtime.RawExtension{
|
|
{Raw: encodeOrDie(a)},
|
|
{Raw: encodeOrDie(b)},
|
|
{Raw: encodeOrDie(c)},
|
|
},
|
|
},
|
|
field: "{.metadata.name}",
|
|
},
|
|
{
|
|
name: "some-missing-fields",
|
|
obj: &unstructured.UnstructuredList{
|
|
Object: map[string]interface{}{
|
|
"kind": "List",
|
|
"apiVersion": "v1",
|
|
},
|
|
Items: []unstructured.Unstructured{
|
|
{
|
|
Object: map[string]interface{}{
|
|
"kind": "ReplicationController",
|
|
"apiVersion": "v1",
|
|
"status": map[string]interface{}{
|
|
"availableReplicas": 2,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Object: map[string]interface{}{
|
|
"kind": "ReplicationController",
|
|
"apiVersion": "v1",
|
|
"status": map[string]interface{}{},
|
|
},
|
|
},
|
|
{
|
|
Object: map[string]interface{}{
|
|
"kind": "ReplicationController",
|
|
"apiVersion": "v1",
|
|
"status": map[string]interface{}{
|
|
"availableReplicas": 1,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
sort: &unstructured.UnstructuredList{
|
|
Object: map[string]interface{}{
|
|
"kind": "List",
|
|
"apiVersion": "v1",
|
|
},
|
|
Items: []unstructured.Unstructured{
|
|
{
|
|
Object: map[string]interface{}{
|
|
"kind": "ReplicationController",
|
|
"apiVersion": "v1",
|
|
"status": map[string]interface{}{},
|
|
},
|
|
},
|
|
{
|
|
Object: map[string]interface{}{
|
|
"kind": "ReplicationController",
|
|
"apiVersion": "v1",
|
|
"status": map[string]interface{}{
|
|
"availableReplicas": 1,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Object: map[string]interface{}{
|
|
"kind": "ReplicationController",
|
|
"apiVersion": "v1",
|
|
"status": map[string]interface{}{
|
|
"availableReplicas": 2,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
field: "{.status.availableReplicas}",
|
|
},
|
|
{
|
|
name: "all-missing-fields",
|
|
obj: &unstructured.UnstructuredList{
|
|
Object: map[string]interface{}{
|
|
"kind": "List",
|
|
"apiVersion": "v1",
|
|
},
|
|
Items: []unstructured.Unstructured{
|
|
{
|
|
Object: map[string]interface{}{
|
|
"kind": "ReplicationController",
|
|
"apiVersion": "v1",
|
|
"status": map[string]interface{}{
|
|
"replicas": 0,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Object: map[string]interface{}{
|
|
"kind": "ReplicationController",
|
|
"apiVersion": "v1",
|
|
"status": map[string]interface{}{
|
|
"replicas": 0,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
field: "{.status.availableReplicas}",
|
|
expectedErr: "couldn't find any field with path \"{.status.availableReplicas}\" in the list of objects",
|
|
},
|
|
{
|
|
name: "model-invalid-fields",
|
|
obj: &corev1.ReplicationControllerList{
|
|
Items: []corev1.ReplicationController{
|
|
{
|
|
Status: corev1.ReplicationControllerStatus{},
|
|
},
|
|
{
|
|
Status: corev1.ReplicationControllerStatus{},
|
|
},
|
|
{
|
|
Status: corev1.ReplicationControllerStatus{},
|
|
},
|
|
},
|
|
},
|
|
field: "{.invalid}",
|
|
expectedErr: "couldn't find any field with path \"{.invalid}\" in the list of objects",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
sort := &SortingPrinter{SortField: tt.field, Decoder: scheme.Codecs.UniversalDecoder()}
|
|
err := sort.sortObj(tt.obj)
|
|
if err != nil {
|
|
if len(tt.expectedErr) > 0 {
|
|
if strings.Contains(err.Error(), tt.expectedErr) {
|
|
return
|
|
}
|
|
t.Fatalf("%s: expected error containing: %q, got: \"%v\"", tt.name, tt.expectedErr, err)
|
|
}
|
|
t.Fatalf("%s: unexpected error: %v", tt.name, err)
|
|
}
|
|
if len(tt.expectedErr) > 0 {
|
|
t.Fatalf("%s: expected error containing: %q, got none", tt.name, tt.expectedErr)
|
|
}
|
|
if !reflect.DeepEqual(tt.obj, tt.sort) {
|
|
t.Errorf("[%s]\nexpected:\n%v\nsaw:\n%v", tt.name, tt.sort, tt.obj)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRuntimeSortLess(t *testing.T) {
|
|
var testobj runtime.Object
|
|
|
|
testobj = &corev1.PodList{
|
|
Items: []corev1.Pod{
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "b",
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "c",
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "a",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
testobjs, err := meta.ExtractList(testobj)
|
|
if err != nil {
|
|
t.Fatalf("ExtractList testobj got unexpected error: %v", err)
|
|
}
|
|
|
|
testfield := "{.metadata.name}"
|
|
testruntimeSort := NewRuntimeSort(testfield, testobjs)
|
|
tests := []struct {
|
|
name string
|
|
runtimeSort *RuntimeSort
|
|
i int
|
|
j int
|
|
expectResult bool
|
|
expectErr bool
|
|
}{
|
|
{
|
|
name: "test less true",
|
|
runtimeSort: testruntimeSort,
|
|
i: 0,
|
|
j: 1,
|
|
expectResult: true,
|
|
},
|
|
{
|
|
name: "test less false",
|
|
runtimeSort: testruntimeSort,
|
|
i: 1,
|
|
j: 2,
|
|
expectResult: false,
|
|
},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
result := test.runtimeSort.Less(test.i, test.j)
|
|
if result != test.expectResult {
|
|
t.Errorf("case[%d]:%s Expected result: %v, Got result: %v", i, test.name, test.expectResult, result)
|
|
}
|
|
})
|
|
}
|
|
}
|