Skip to content

Commit 0e09a0d

Browse files
k8s-ci-robotk8s-ci-robot
authored
Merge pull request #213 from grantr/fake-client-scheme
Add scheme support to fake client
2 parents 0f0740d + 5558165 commit 0e09a0d

File tree

2 files changed

+141
-101
lines changed

2 files changed

+141
-101
lines changed

‎pkg/client/fake/client.go

Lines changed: 46 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,9 @@ package fake
1919
import (
2020
"context"
2121
"encoding/json"
22+
"fmt"
2223
"os"
24+
"strings"
2325

2426
"k8s.io/apimachinery/pkg/api/meta"
2527
"k8s.io/apimachinery/pkg/runtime"
@@ -38,14 +40,22 @@ var (
3840

3941
type fakeClient struct {
4042
tracker testing.ObjectTracker
43+
scheme *runtime.Scheme
4144
}
4245

4346
var _ client.Client = &fakeClient{}
4447

4548
// NewFakeClient creates a new fake client for testing.
4649
// You can choose to initialize it with a slice of runtime.Object.
4750
func NewFakeClient(initObjs ...runtime.Object) client.Client {
48-
tracker := testing.NewObjectTracker(scheme.Scheme, scheme.Codecs.UniversalDecoder())
51+
return NewFakeClientWithScheme(scheme.Scheme, initObjs...)
52+
}
53+
54+
// NewFakeClientWithScheme creates a new fake client with the given scheme
55+
// for testing.
56+
// You can choose to initialize it with a slice of runtime.Object.
57+
func NewFakeClientWithScheme(clientScheme *runtime.Scheme, initObjs ...runtime.Object) client.Client {
58+
tracker := testing.NewObjectTracker(clientScheme, scheme.Codecs.UniversalDecoder())
4959
for _, obj := range initObjs {
5060
err := tracker.Add(obj)
5161
if err != nil {
@@ -56,11 +66,12 @@ func NewFakeClient(initObjs ...runtime.Object) client.Client {
5666
}
5767
return &fakeClient{
5868
tracker: tracker,
69+
scheme: clientScheme,
5970
}
6071
}
6172

6273
func (c *fakeClient) Get(ctx context.Context, key client.ObjectKey, obj runtime.Object) error {
63-
gvr, err := getGVRFromObject(obj)
74+
gvr, err := getGVRFromObject(obj, c.scheme)
6475
if err != nil {
6576
return err
6677
}
@@ -78,7 +89,16 @@ func (c *fakeClient) Get(ctx context.Context, key client.ObjectKey, obj runtime.
7889
}
7990

8091
func (c *fakeClient) List(ctx context.Context, opts *client.ListOptions, list runtime.Object) error {
81-
gvk := opts.Raw.TypeMeta.GroupVersionKind()
92+
gvk, err := getGVKFromList(list, c.scheme)
93+
if err != nil {
94+
// The old fake client required GVK info in Raw.TypeMeta, so check there
95+
// before giving up
96+
if opts.Raw == nil || opts.Raw.TypeMeta.APIVersion == "" || opts.Raw.TypeMeta.Kind == "" {
97+
return err
98+
}
99+
gvk = opts.Raw.TypeMeta.GroupVersionKind()
100+
}
101+
82102
gvr, _ := meta.UnsafeGuessKindToResource(gvk)
83103
o, err := c.tracker.List(gvr, gvk, opts.Namespace)
84104
if err != nil {
@@ -94,7 +114,7 @@ func (c *fakeClient) List(ctx context.Context, opts *client.ListOptions, list ru
94114
}
95115

96116
func (c *fakeClient) Create(ctx context.Context, obj runtime.Object) error {
97-
gvr, err := getGVRFromObject(obj)
117+
gvr, err := getGVRFromObject(obj, c.scheme)
98118
if err != nil {
99119
return err
100120
}
@@ -106,7 +126,7 @@ func (c *fakeClient) Create(ctx context.Context, obj runtime.Object) error {
106126
}
107127

108128
func (c *fakeClient) Delete(ctx context.Context, obj runtime.Object, opts ...client.DeleteOptionFunc) error {
109-
gvr, err := getGVRFromObject(obj)
129+
gvr, err := getGVRFromObject(obj, c.scheme)
110130
if err != nil {
111131
return err
112132
}
@@ -119,7 +139,7 @@ func (c *fakeClient) Delete(ctx context.Context, obj runtime.Object, opts ...cli
119139
}
120140

121141
func (c *fakeClient) Update(ctx context.Context, obj runtime.Object) error {
122-
gvr, err := getGVRFromObject(obj)
142+
gvr, err := getGVRFromObject(obj, c.scheme)
123143
if err != nil {
124144
return err
125145
}
@@ -134,15 +154,33 @@ func (c *fakeClient) Status() client.StatusWriter {
134154
return &fakeStatusWriter{client: c}
135155
}
136156

137-
func getGVRFromObject(obj runtime.Object) (schema.GroupVersionResource, error) {
138-
gvk, err := apiutil.GVKForObject(obj, scheme.Scheme)
157+
func getGVRFromObject(obj runtime.Object, scheme *runtime.Scheme) (schema.GroupVersionResource, error) {
158+
gvk, err := apiutil.GVKForObject(obj, scheme)
139159
if err != nil {
140160
return schema.GroupVersionResource{}, err
141161
}
142162
gvr, _ := meta.UnsafeGuessKindToResource(gvk)
143163
return gvr, nil
144164
}
145165

166+
func getGVKFromList(list runtime.Object, scheme *runtime.Scheme) (schema.GroupVersionKind, error) {
167+
gvk, err := apiutil.GVKForObject(list, scheme)
168+
if err != nil {
169+
return schema.GroupVersionKind{}, err
170+
}
171+
172+
if gvk.Kind == "List" {
173+
return schema.GroupVersionKind{}, fmt.Errorf("cannot derive GVK for generic List type %T (kind %q)", list, gvk)
174+
}
175+
176+
if !strings.HasSuffix(gvk.Kind, "List") {
177+
return schema.GroupVersionKind{}, fmt.Errorf("non-list type %T (kind %q) passed as output", list, gvk)
178+
}
179+
// we need the non-list GVK, so chop off the "List" from the end of the kind
180+
gvk.Kind = gvk.Kind[:len(gvk.Kind)-4]
181+
return gvk, nil
182+
}
183+
146184
type fakeStatusWriter struct {
147185
client *fakeClient
148186
}

‎pkg/client/fake/client_test.go

Lines changed: 95 additions & 93 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,6 @@ limitations under the License.
1717
package fake
1818

1919
import (
20-
"encoding/json"
21-
2220
. "github.com/onsi/ginkgo"
2321
. "github.com/onsi/gomega"
2422

@@ -35,7 +33,7 @@ var _ = Describe("Fake client", func() {
3533
var cm *corev1.ConfigMap
3634
var cl client.Client
3735

38-
BeforeEach(func(done Done) {
36+
BeforeEach(func() {
3937
dep = &appsv1.Deployment{
4038
ObjectMeta: metav1.ObjectMeta{
4139
Name: "test-deployment",
@@ -51,106 +49,110 @@ var _ = Describe("Fake client", func() {
5149
"test-key": "test-value",
5250
},
5351
}
54-
cl = NewFakeClient(dep, cm)
55-
close(done)
56-
})
57-
58-
It("should be able to Get", func() {
59-
By("Getting a deployment")
60-
namespacedName := types.NamespacedName{
61-
Name: "test-deployment",
62-
Namespace: "ns1",
63-
}
64-
obj := &appsv1.Deployment{}
65-
err := cl.Get(nil, namespacedName, obj)
66-
Expect(err).To(BeNil())
67-
Expect(obj).To(Equal(dep))
6852
})
6953

70-
It("should be able to List", func() {
71-
By("Listing all deployments in a namespace")
72-
list := &metav1.List{}
73-
err := cl.List(nil, &client.ListOptions{
74-
Namespace: "ns1",
75-
Raw: &metav1.ListOptions{
76-
TypeMeta: metav1.TypeMeta{
77-
APIVersion: "apps/v1",
78-
Kind: "Deployment",
54+
AssertClientBehavior := func() {
55+
It("should be able to Get", func() {
56+
By("Getting a deployment")
57+
namespacedName := types.NamespacedName{
58+
Name: "test-deployment",
59+
Namespace: "ns1",
60+
}
61+
obj := &appsv1.Deployment{}
62+
err := cl.Get(nil, namespacedName, obj)
63+
Expect(err).To(BeNil())
64+
Expect(obj).To(Equal(dep))
65+
})
66+
67+
It("should be able to List", func() {
68+
By("Listing all deployments in a namespace")
69+
list := &appsv1.DeploymentList{}
70+
err := cl.List(nil, &client.ListOptions{
71+
Namespace: "ns1",
72+
}, list)
73+
Expect(err).To(BeNil())
74+
Expect(list.Items).To(HaveLen(1))
75+
Expect(list.Items).To(ConsistOf(*dep))
76+
})
77+
78+
It("should be able to Create", func() {
79+
By("Creating a new configmap")
80+
newcm := &corev1.ConfigMap{
81+
ObjectMeta: metav1.ObjectMeta{
82+
Name: "new-test-cm",
83+
Namespace: "ns2",
7984
},
80-
},
81-
}, list)
82-
Expect(err).To(BeNil())
83-
Expect(list.Items).To(HaveLen(1))
84-
j, err := json.Marshal(dep)
85-
Expect(err).To(BeNil())
86-
expectedDep := runtime.RawExtension{Raw: j}
87-
Expect(list.Items).To(ConsistOf(expectedDep))
88-
})
85+
}
86+
err := cl.Create(nil, newcm)
87+
Expect(err).To(BeNil())
8988

90-
It("should be able to Create", func() {
91-
By("Creating a new configmap")
92-
newcm := &corev1.ConfigMap{
93-
ObjectMeta: metav1.ObjectMeta{
89+
By("Getting the new configmap")
90+
namespacedName := types.NamespacedName{
9491
Name: "new-test-cm",
9592
Namespace: "ns2",
96-
},
97-
}
98-
err := cl.Create(nil, newcm)
99-
Expect(err).To(BeNil())
100-
101-
By("Getting the new configmap")
102-
namespacedName := types.NamespacedName{
103-
Name: "new-test-cm",
104-
Namespace: "ns2",
105-
}
106-
obj := &corev1.ConfigMap{}
107-
err = cl.Get(nil, namespacedName, obj)
108-
Expect(err).To(BeNil())
109-
Expect(obj).To(Equal(newcm))
110-
})
93+
}
94+
obj := &corev1.ConfigMap{}
95+
err = cl.Get(nil, namespacedName, obj)
96+
Expect(err).To(BeNil())
97+
Expect(obj).To(Equal(newcm))
98+
})
99+
100+
It("should be able to Update", func() {
101+
By("Updating a new configmap")
102+
newcm := &corev1.ConfigMap{
103+
ObjectMeta: metav1.ObjectMeta{
104+
Name: "test-cm",
105+
Namespace: "ns2",
106+
},
107+
Data: map[string]string{
108+
"test-key": "new-value",
109+
},
110+
}
111+
err := cl.Update(nil, newcm)
112+
Expect(err).To(BeNil())
111113

112-
It("should be able to Update", func() {
113-
By("Updating a new configmap")
114-
newcm := &corev1.ConfigMap{
115-
ObjectMeta: metav1.ObjectMeta{
114+
By("Getting the new configmap")
115+
namespacedName := types.NamespacedName{
116116
Name: "test-cm",
117117
Namespace: "ns2",
118-
},
119-
Data: map[string]string{
120-
"test-key": "new-value",
121-
},
122-
}
123-
err := cl.Update(nil, newcm)
124-
Expect(err).To(BeNil())
125-
126-
By("Getting the new configmap")
127-
namespacedName := types.NamespacedName{
128-
Name: "test-cm",
129-
Namespace: "ns2",
130-
}
131-
obj := &corev1.ConfigMap{}
132-
err = cl.Get(nil, namespacedName, obj)
133-
Expect(err).To(BeNil())
134-
Expect(obj).To(Equal(newcm))
118+
}
119+
obj := &corev1.ConfigMap{}
120+
err = cl.Get(nil, namespacedName, obj)
121+
Expect(err).To(BeNil())
122+
Expect(obj).To(Equal(newcm))
123+
})
124+
125+
It("should be able to Delete", func() {
126+
By("Deleting a deployment")
127+
err := cl.Delete(nil, dep)
128+
Expect(err).To(BeNil())
129+
130+
By("Listing all deployments in the namespace")
131+
list := &appsv1.DeploymentList{}
132+
err = cl.List(nil, &client.ListOptions{
133+
Namespace: "ns1",
134+
}, list)
135+
Expect(err).To(BeNil())
136+
Expect(list.Items).To(HaveLen(0))
137+
})
138+
}
139+
140+
Context("with default scheme.Scheme", func() {
141+
BeforeEach(func(done Done) {
142+
cl = NewFakeClient(dep, cm)
143+
close(done)
144+
})
145+
AssertClientBehavior()
135146
})
136147

137-
It("should be able to Delete", func() {
138-
By("Deleting a deployment")
139-
err := cl.Delete(nil, dep)
140-
Expect(err).To(BeNil())
141-
142-
By("Listing all deployments in the namespace")
143-
list := &metav1.List{}
144-
err = cl.List(nil, &client.ListOptions{
145-
Namespace: "ns1",
146-
Raw: &metav1.ListOptions{
147-
TypeMeta: metav1.TypeMeta{
148-
APIVersion: "apps/v1",
149-
Kind: "Deployment",
150-
},
151-
},
152-
}, list)
153-
Expect(err).To(BeNil())
154-
Expect(list.Items).To(HaveLen(0))
148+
Context("with given scheme", func() {
149+
BeforeEach(func(done Done) {
150+
scheme := runtime.NewScheme()
151+
corev1.AddToScheme(scheme)
152+
appsv1.AddToScheme(scheme)
153+
cl = NewFakeClientWithScheme(scheme, dep, cm)
154+
close(done)
155+
})
156+
AssertClientBehavior()
155157
})
156158
})

0 commit comments

Comments
 (0)