Skip to content

Commit 18a8fbc

Browse files
committed
Update tests to use new List signature
Now that the fake client no longer needs Raw, these tests can use typed instead of untyped List objects.
1 parent 778d809 commit 18a8fbc

File tree

2 files changed

+15
-76
lines changed

2 files changed

+15
-76
lines changed

pkg/admission/cert/writer/secret_test.go

Lines changed: 14 additions & 75 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ limitations under the License.
1717
package writer
1818

1919
import (
20-
"encoding/json"
2120
"fmt"
2221

2322
. "github.com/onsi/ginkgo"
@@ -26,7 +25,6 @@ import (
2625
admissionregistration "k8s.io/api/admissionregistration/v1beta1"
2726
corev1 "k8s.io/api/core/v1"
2827
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
29-
"k8s.io/apimachinery/pkg/runtime"
3028
"k8s.io/apimachinery/pkg/types"
3129
"sigs.k8s.io/controller-runtime/pkg/admission/cert/generator"
3230
fakegenerator "sigs.k8s.io/controller-runtime/pkg/admission/cert/generator/fake"
@@ -108,7 +106,6 @@ var _ = Describe("SecretCertWriter", func() {
108106
var certWriter CertWriter
109107
var secretCertWriter *SecretCertWriter
110108
var secret *corev1.Secret
111-
var expectedSecret runtime.RawExtension
112109

113110
Context("Failed to EnsureCerts", func() {
114111
BeforeEach(func(done Done) {
@@ -201,43 +198,21 @@ var _ = Describe("SecretCertWriter", func() {
201198
It("should default it and return no error", func() {
202199
err := certWriter.EnsureCerts(mwc)
203200
Expect(err).NotTo(HaveOccurred())
204-
list := &corev1.List{}
205-
err = cl.List(nil, &client.ListOptions{
206-
Namespace: "namespace-bar",
207-
Raw: &metav1.ListOptions{
208-
TypeMeta: metav1.TypeMeta{
209-
APIVersion: "v1",
210-
Kind: "Secret",
211-
},
212-
},
213-
}, list)
201+
list := &corev1.SecretList{}
202+
err = cl.List(nil, list, client.InNamespace("namespace-bar"))
214203
Expect(err).NotTo(HaveOccurred())
215204
Expect(list.Items).To(HaveLen(1))
216205
})
217206
})
218207

219208
Context("no existing secret", func() {
220-
BeforeEach(func(done Done) {
221-
j, _ := json.Marshal(secret)
222-
expectedSecret = runtime.RawExtension{Raw: j}
223-
close(done)
224-
})
225-
226209
It("should create new secrets with certs", func() {
227210
err := certWriter.EnsureCerts(mwc)
228211
Expect(err).NotTo(HaveOccurred())
229-
list := &corev1.List{}
230-
err = cl.List(nil, &client.ListOptions{
231-
Namespace: "namespace-bar",
232-
Raw: &metav1.ListOptions{
233-
TypeMeta: metav1.TypeMeta{
234-
APIVersion: "v1",
235-
Kind: "Secret",
236-
},
237-
},
238-
}, list)
212+
list := &corev1.SecretList{}
213+
err = cl.List(nil, list, client.InNamespace("namespace-bar"))
239214
Expect(err).NotTo(HaveOccurred())
240-
Expect(list.Items).To(ConsistOf(expectedSecret))
215+
Expect(list.Items).To(ConsistOf(*secret))
241216
Expect(list.Items).To(HaveLen(1))
242217
})
243218
})
@@ -246,12 +221,6 @@ var _ = Describe("SecretCertWriter", func() {
246221
var oldSecret *corev1.Secret
247222

248223
Context("cert is invalid", func() {
249-
BeforeEach(func(done Done) {
250-
j, _ := json.Marshal(secret)
251-
expectedSecret = runtime.RawExtension{Raw: j}
252-
close(done)
253-
})
254-
255224
Describe("cert in secret is incomplete", func() {
256225
BeforeEach(func(done Done) {
257226
oldSecret = secret.DeepCopy()
@@ -264,18 +233,10 @@ var _ = Describe("SecretCertWriter", func() {
264233
It("should replace with new certs", func() {
265234
err := certWriter.EnsureCerts(mwc)
266235
Expect(err).NotTo(HaveOccurred())
267-
list := &corev1.List{}
268-
err = cl.List(nil, &client.ListOptions{
269-
Namespace: "namespace-bar",
270-
Raw: &metav1.ListOptions{
271-
TypeMeta: metav1.TypeMeta{
272-
APIVersion: "v1",
273-
Kind: "Secret",
274-
},
275-
},
276-
}, list)
236+
list := &corev1.SecretList{}
237+
err = cl.List(nil, list, client.InNamespace("namespace-bar"))
277238
Expect(err).NotTo(HaveOccurred())
278-
Expect(list.Items).To(ConsistOf(expectedSecret))
239+
Expect(list.Items).To(ConsistOf(*secret))
279240
Expect(list.Items).To(HaveLen(1))
280241
})
281242
})
@@ -296,18 +257,10 @@ var _ = Describe("SecretCertWriter", func() {
296257
It("should replace with new certs", func() {
297258
err := certWriter.EnsureCerts(mwc)
298259
Expect(err).NotTo(HaveOccurred())
299-
list := &corev1.List{}
300-
err = cl.List(nil, &client.ListOptions{
301-
Namespace: "namespace-bar",
302-
Raw: &metav1.ListOptions{
303-
TypeMeta: metav1.TypeMeta{
304-
APIVersion: "v1",
305-
Kind: "Secret",
306-
},
307-
},
308-
}, list)
260+
list := &corev1.SecretList{}
261+
err = cl.List(nil, list, client.InNamespace("namespace-bar"))
309262
Expect(err).NotTo(HaveOccurred())
310-
Expect(list.Items).To(ConsistOf(expectedSecret))
263+
Expect(list.Items).To(ConsistOf(*secret))
311264
Expect(list.Items).To(HaveLen(1))
312265
})
313266
})
@@ -320,8 +273,6 @@ var _ = Describe("SecretCertWriter", func() {
320273
ServerKeyName: []byte(certs2.Key),
321274
ServerCertName: []byte(certs2.Cert),
322275
}
323-
j, _ := json.Marshal(oldSecret)
324-
expectedSecret = runtime.RawExtension{Raw: j}
325276
cl = fake.NewFakeClient(oldSecret)
326277
secretCertWriter.Client = cl
327278
close(done)
@@ -335,8 +286,6 @@ var _ = Describe("SecretCertWriter", func() {
335286
ServerKeyName: []byte(certs2.Key),
336287
ServerCertName: []byte(certs2.Cert),
337288
}
338-
j, _ := json.Marshal(oldSecret)
339-
expectedSecret = runtime.RawExtension{Raw: j}
340289

341290
cl = fake.NewFakeClient(oldSecret)
342291
secretCertWriter.Client = cl
@@ -345,19 +294,11 @@ var _ = Describe("SecretCertWriter", func() {
345294
It("should keep the secret", func() {
346295
err := certWriter.EnsureCerts(mwc)
347296
Expect(err).NotTo(HaveOccurred())
348-
list := &corev1.List{}
349-
err = cl.List(nil, &client.ListOptions{
350-
Namespace: "namespace-bar",
351-
Raw: &metav1.ListOptions{
352-
TypeMeta: metav1.TypeMeta{
353-
APIVersion: "v1",
354-
Kind: "Secret",
355-
},
356-
},
357-
}, list)
297+
list := &corev1.SecretList{}
298+
err = cl.List(nil, list, client.InNamespace("namespace-bar"))
358299
Expect(err).NotTo(HaveOccurred())
359300
Expect(list.Items).To(HaveLen(1))
360-
Expect(list.Items[0]).To(Equal(expectedSecret))
301+
Expect(list.Items).To(ConsistOf(*oldSecret))
361302
})
362303
})
363304

@@ -370,8 +311,6 @@ var _ = Describe("SecretCertWriter", func() {
370311
//ServerKeyName: []byte(expiringKeyPEM),
371312
//ServerCertName: []byte(expiringCertPEM),
372313
}
373-
//j, _ := json.Marshal(someNewValidSecret)
374-
//expectedSecret = runtime.RawExtension{Raw: j}
375314

376315
cl = fake.NewFakeClient(oldSecret)
377316
secretCertWriter.Client = cl

pkg/builder/example_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ func (a *ReplicaSetReconciler) Reconcile(req reconcile.Request) (reconcile.Resul
7777

7878
// List the Pods matching the PodTemplate Labels
7979
pods := &corev1.PodList{}
80-
err = a.List(context.TODO(), client.InNamespace(req.Namespace).MatchingLabels(rs.Spec.Template.Labels), pods)
80+
err = a.List(context.TODO(), pods, client.InNamespace(req.Namespace), client.MatchingLabels(rs.Spec.Template.Labels))
8181
if err != nil {
8282
return reconcile.Result{}, err
8383
}

0 commit comments

Comments
 (0)