132 lines
2.8 KiB
Go
132 lines
2.8 KiB
Go
package pod
|
|
|
|
import (
|
|
"github.com/google/go-cmp/cmp"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/informers"
|
|
"k8s.io/client-go/kubernetes/fake"
|
|
"kubesphere.io/kubesphere/pkg/api"
|
|
"kubesphere.io/kubesphere/pkg/apiserver/query"
|
|
"kubesphere.io/kubesphere/pkg/models/resources/v1alpha3"
|
|
"testing"
|
|
)
|
|
|
|
func TestListPods(t *testing.T) {
|
|
tests := []struct {
|
|
description string
|
|
namespace string
|
|
query *query.Query
|
|
expected *api.ListResult
|
|
expectedErr error
|
|
}{
|
|
{
|
|
"test name filter",
|
|
"default",
|
|
&query.Query{
|
|
Pagination: &query.Pagination{
|
|
Limit: 10,
|
|
Offset: 0,
|
|
},
|
|
SortBy: query.FieldName,
|
|
Ascending: false,
|
|
Filters: map[query.Field]query.Value{query.FieldNamespace: query.Value("default")},
|
|
},
|
|
&api.ListResult{
|
|
Items: []interface{}{foo4, foo3, foo2, foo1},
|
|
TotalItems: len(pods),
|
|
},
|
|
nil,
|
|
},
|
|
{
|
|
"test pvcName filter",
|
|
"default",
|
|
&query.Query{
|
|
Pagination: &query.Pagination{
|
|
Limit: 10,
|
|
Offset: 0,
|
|
},
|
|
SortBy: query.FieldName,
|
|
Ascending: false,
|
|
Filters: map[query.Field]query.Value{
|
|
query.FieldNamespace: query.Value("default"),
|
|
filedPVCName: query.Value(foo4.Spec.Volumes[0].PersistentVolumeClaim.ClaimName),
|
|
},
|
|
},
|
|
&api.ListResult{
|
|
Items: []interface{}{foo4},
|
|
TotalItems: 1,
|
|
},
|
|
nil,
|
|
},
|
|
}
|
|
|
|
getter := prepare()
|
|
|
|
for _, test := range tests {
|
|
got, err := getter.List(test.namespace, test.query)
|
|
if test.expectedErr != nil && err != test.expectedErr {
|
|
t.Errorf("expected error, got nothing")
|
|
} else if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if diff := cmp.Diff(got, test.expected); diff != "" {
|
|
t.Errorf("%T differ (-got, +want): %s", test.expected, diff)
|
|
}
|
|
}
|
|
}
|
|
|
|
var (
|
|
foo1 = &corev1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "foo1",
|
|
Namespace: "default",
|
|
},
|
|
}
|
|
foo2 = &corev1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "foo2",
|
|
Namespace: "default",
|
|
},
|
|
}
|
|
foo3 = &corev1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "foo3",
|
|
Namespace: "default",
|
|
},
|
|
}
|
|
foo4 = &corev1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "foo4",
|
|
Namespace: "default",
|
|
},
|
|
Spec: corev1.PodSpec{
|
|
Volumes: []corev1.Volume{
|
|
{
|
|
Name: "data",
|
|
VolumeSource: corev1.VolumeSource{
|
|
PersistentVolumeClaim: &corev1.PersistentVolumeClaimVolumeSource{
|
|
ClaimName: "pvc-1",
|
|
ReadOnly: false,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
pods = []interface{}{foo1, foo2, foo3, foo4}
|
|
)
|
|
|
|
func prepare() v1alpha3.Interface {
|
|
|
|
client := fake.NewSimpleClientset()
|
|
informer := informers.NewSharedInformerFactory(client, 0)
|
|
|
|
for _, pod := range pods {
|
|
_ = informer.Core().V1().Pods().Informer().GetIndexer().Add(pod)
|
|
}
|
|
|
|
return New(informer)
|
|
}
|