add field selector enabled fake client

Author: answer1991

pkg/client/fake/field_selector_client.go

@@ -0,0 +1,106 @@
+/*
+Copyright 2020 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 fake
+
+import (
+	"context"
+
+	"k8s.io/apimachinery/pkg/api/meta"
+	"k8s.io/apimachinery/pkg/fields"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+
+	"sigs.k8s.io/controller-runtime/pkg/client/apiutil"
+)
+
+var _ client.Client = &fieldSelectorEnableClient{}
+
+// NewFieldSelectorEnableClient wrapper a fake client that List method enables field selector
+func NewFieldSelectorEnableClient(client client.Client, scheme *runtime.Scheme, obj runtime.Object, indexers map[string]client.IndexerFunc) (client.Client, error) {
+	gvk, err := apiutil.GVKForObject(obj, scheme)
+	if err != nil {
+		return nil, err
+	}
+
+	return &fieldSelectorEnableClient{
+		Client:   client,
+		scheme:   scheme,
+		gvk:      gvk,
+		indexers: indexers,
+	}, nil
+}
+
+type fieldSelectorEnableClient struct {
+	client.Client
+	scheme *runtime.Scheme
+
+	gvk schema.GroupVersionKind
+
+	indexers map[string]client.IndexerFunc
+}
+
+func (f *fieldSelectorEnableClient) List(ctx context.Context, list runtime.Object, opts ...client.ListOption) error {
+	if err := f.Client.List(ctx, list, opts...); err != nil {
+		return err
+	}
+
+	if len(opts) == 0 {
+		return nil
+	}
+
+	objs, err := meta.ExtractList(list)
+	if err != nil {
+		return err
+	}
+
+	if len(objs) == 0 {
+		return nil
+	}
+
+	if gvk, err := apiutil.GVKForObject(objs[0], f.scheme); err != nil {
+		return err
+	} else if gvk != f.gvk {
+		return nil
+	}
+
+	options := client.ListOptions{}
+	options.ApplyOptions(opts)
+
+	if options.FieldSelector == nil {
+		return nil
+	}
+
+	filtered := make([]runtime.Object, 0)
+	for i := range objs {
+		objFields := map[string]string{}
+
+		for k, indexer := range f.indexers {
+			values := indexer(objs[i])
+			if len(values) != 1 {
+				continue
+			}
+			objFields[k] = values[0]
+		}
+
+		if options.FieldSelector.Matches(fields.Set(objFields)) {
+			filtered = append(filtered, objs[i])
+		}
+	}
+
+	return meta.SetList(list, filtered)
+}

pkg/client/fake/field_selector_client_test.go

@@ -0,0 +1,231 @@
+/*
+Copyright 2020 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 fake
+
+import (
+	"context"
+	"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/fields"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/client-go/kubernetes/scheme"
+
+	"sigs.k8s.io/controller-runtime/pkg/client"
+)
+
+func TestListByFieldSelector(t *testing.T) {
+	pod := &corev1.Pod{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "foo",
+			Namespace: "bar",
+		},
+		Spec: corev1.PodSpec{
+			NodeName:           "node1",
+			ServiceAccountName: "sa1",
+		},
+	}
+
+	node := &corev1.Node{
+		ObjectMeta: metav1.ObjectMeta{
+			Name: "foo",
+		},
+		Spec: corev1.NodeSpec{
+			PodCIDR:    "cidr1",
+			ProviderID: "provider1",
+		},
+	}
+
+	fakeClient := NewFakeClientWithScheme(scheme.Scheme, pod, node)
+
+	podIndexers := map[string]client.IndexerFunc{
+		"spec.nodeName": func(o runtime.Object) []string {
+			po, ok := o.(*corev1.Pod)
+			if !ok {
+				return nil
+			}
+
+			return []string{po.Spec.NodeName}
+		},
+		"spec.serviceAccountName": func(o runtime.Object) []string {
+			po, ok := o.(*corev1.Pod)
+			if !ok {
+				return nil
+			}
+
+			return []string{po.Spec.ServiceAccountName}
+		},
+	}
+
+	nodeIndexers := map[string]client.IndexerFunc{
+		"spec.podCIDR": func(o runtime.Object) []string {
+			no, ok := o.(*corev1.Node)
+			if !ok {
+				return nil
+			}
+
+			return []string{no.Spec.PodCIDR}
+		},
+		"spec.providerID": func(o runtime.Object) []string {
+			no, ok := o.(*corev1.Node)
+			if !ok {
+				return nil
+			}
+
+			return []string{no.Spec.ProviderID}
+		},
+	}
+
+	fieldSelectorClient, err := NewFieldSelectorEnableClient(fakeClient, scheme.Scheme, &corev1.Pod{}, podIndexers)
+	if err != nil {
+		t.Fatalf("unexpect init client err: %v", err)
+	}
+
+	fieldSelectorClient, err = NewFieldSelectorEnableClient(fieldSelectorClient, scheme.Scheme, &corev1.Node{}, nodeIndexers)
+	if err != nil {
+		t.Fatalf("unexpect init client err: %v", err)
+	}
+
+	cases := []struct {
+		obj                runtime.Object
+		name               string
+		opts               client.ListOption
+		expectObjectsCount int
+	}{
+		{
+			name:               "list Pod using nil fieldSelector",
+			obj:                &corev1.PodList{},
+			opts:               nil,
+			expectObjectsCount: 1,
+		},
+		{
+			name:               "list Node using nil fieldSelector",
+			obj:                &corev1.NodeList{},
+			opts:               nil,
+			expectObjectsCount: 1,
+		},
+		{
+			name: "list Pod does not match fieldSelector name",
+			obj:  &corev1.PodList{},
+			opts: client.MatchingFields(fields.Set{
+				"spec.nodeName1": "node1",
+			}),
+			expectObjectsCount: 0,
+		},
+		{
+			name: "list Node does not match fieldSelector name",
+			obj:  &corev1.NodeList{},
+			opts: client.MatchingFields(fields.Set{
+				"spec.podCIDR1": "node1",
+			}),
+			expectObjectsCount: 0,
+		},
+		{
+			name: "list Pod does not match fieldSelector value",
+			obj:  &corev1.PodList{},
+			opts: client.MatchingFields(fields.Set{
+				"spec.nodeName": "node2",
+			}),
+			expectObjectsCount: 0,
+		},
+		{
+			name: "list Node does not match fieldSelector value",
+			obj:  &corev1.NodeList{},
+			opts: client.MatchingFields(fields.Set{
+				"spec.podCIDR": "bar2",
+			}),
+			expectObjectsCount: 0,
+		},
+		{
+			name: "list Pod match fieldSelector one name and value",
+			obj:  &corev1.PodList{},
+			opts: client.MatchingFields(fields.Set{
+				"spec.nodeName": "node1",
+			}),
+			expectObjectsCount: 1,
+		},
+		{
+			name: "list Node match fieldSelector one name and value",
+			obj:  &corev1.NodeList{},
+			opts: client.MatchingFields(fields.Set{
+				"spec.podCIDR": "cidr1",
+			}),
+			expectObjectsCount: 1,
+		},
+		{
+			name: "list Pod match fieldSelector one name and value, and does not match one value",
+			obj:  &corev1.PodList{},
+			opts: client.MatchingFields(fields.Set{
+				"spec.nodeName":           "node1",
+				"spec.serviceAccountName": "sa2",
+			}),
+			expectObjectsCount: 0,
+		},
+		{
+			name: "list Node match fieldSelector one name and value, and does not match one value",
+			obj:  &corev1.NodeList{},
+			opts: client.MatchingFields(fields.Set{
+				"spec.podCIDR":    "cidr1",
+				"spec.providerID": "provider2",
+			}),
+			expectObjectsCount: 0,
+		},
+		{
+			name: "list Pod match fieldSelector two name and value",
+			obj:  &corev1.PodList{},
+			opts: client.MatchingFields(fields.Set{
+				"spec.nodeName":           "node1",
+				"spec.serviceAccountName": "sa1",
+			}),
+			expectObjectsCount: 1,
+		},
+		{
+			name: "list Node match fieldSelector two name and value",
+			obj:  &corev1.NodeList{},
+			opts: client.MatchingFields(fields.Set{
+				"spec.podCIDR":    "cidr1",
+				"spec.providerID": "provider1",
+			}),
+			expectObjectsCount: 1,
+		},
+	}
+
+	for _, tc := range cases {
+		t.Run(tc.name, func(t *testing.T) {
+			if tc.opts != nil {
+				if err := fieldSelectorClient.List(context.TODO(), tc.obj, tc.opts); err != nil {
+					t.Fatalf("unexpect list err: %v", err)
+				}
+			} else {
+				if err := fieldSelectorClient.List(context.TODO(), tc.obj); err != nil {
+					t.Fatalf("unexpect list err: %v", err)
+				}
+			}
+
+			objs, err := meta.ExtractList(tc.obj)
+			if err != nil {
+				t.Fatalf("unexpect extra list err: %v", err)
+			}
+
+			if len(objs) != tc.expectObjectsCount {
+				t.Errorf("expect list object result count: %d, got: %d", tc.expectObjectsCount, len(objs))
+			}
+		})
+	}
+}