Add more e2e tests with example controller (#519)

* Extend test cases for verifying the setup

* Add tests case with many objects

* Add test case for removing shard

* Add test case for adding shard

* Add test case for graceful shard termination

* Add test case for shard failure detection

Author: timebertt

cmd/checksum-controller/reconciler.go

@@ -21,6 +21,7 @@ import (
 	"crypto/sha256"
 	"encoding/hex"
 	"fmt"
+	"maps"
 
 	corev1 "k8s.io/api/core/v1"
 	apiequality "k8s.io/apimachinery/pkg/api/equality"
@@ -112,6 +113,7 @@ func (r *Reconciler) Reconcile(ctx context.Context, req reconcile.Request) (reco
 		ObjectMeta: metav1.ObjectMeta{
 			Name:      "checksums-" + secret.Name,
 			Namespace: secret.Namespace,
+			Labels:    maps.Clone(secret.Labels),
 		},
 		Data: make(map[string]string, len(secret.Data)),
 	}

hack/test-e2e.sh

@@ -6,4 +6,4 @@ set -o errexit
 
 source "$(dirname "$0")/test-e2e.env"
 
-ginkgo run --timeout=1h --poll-progress-after=60s --poll-progress-interval=30s --randomize-all --randomize-suites --keep-going --vv "$@"
+ginkgo run --timeout=1h --poll-progress-after=60s --poll-progress-interval=30s --randomize-all --randomize-suites --keep-going -v --show-node-events "$@"

pkg/utils/test/matchers/object.go

@@ -22,16 +22,26 @@ import (
 )
 
 // HaveName succeeds if the actual object has a matching name.
-func HaveName(name interface{}) gomegatypes.GomegaMatcher {
+func HaveName(name string) gomegatypes.GomegaMatcher {
 	return HaveField("ObjectMeta.Name", name)
 }
 
+// HaveNames succeeds if the actual list consists of matching names.
+func HaveNames(names ...string) gomegatypes.GomegaMatcher {
+	matchers := make([]any, len(names))
+	for i, n := range names {
+		matchers[i] = HaveName(n)
+	}
+
+	return HaveField("Items", ConsistOf(matchers...))
+}
+
 // HaveLabel succeeds if the actual object has a label with a matching key.
-func HaveLabel(key interface{}) gomegatypes.GomegaMatcher {
+func HaveLabel(key any) gomegatypes.GomegaMatcher {
 	return HaveField("ObjectMeta.Labels", HaveKey(key))
 }
 
 // HaveLabelWithValue succeeds if the actual object has a label with a matching key and value.
-func HaveLabelWithValue(key, value interface{}) gomegatypes.GomegaMatcher {
+func HaveLabelWithValue(key, value any) gomegatypes.GomegaMatcher {
 	return HaveField("ObjectMeta.Labels", HaveKeyWithValue(key, value))
 }

test/e2e/e2e_suite_test.go

@@ -69,6 +69,9 @@ var _ = BeforeSuite(func() {
 
 	restConfig, err := config.GetConfig()
 	Expect(err).NotTo(HaveOccurred())
+	// gotta go fast during tests
+	restConfig.QPS = 100
+	restConfig.Burst = 150
 
 	scheme := runtime.NewScheme()
 	schemeBuilder := runtime.NewSchemeBuilder(