Adapt e2e tests

Author: timebertt

test/e2e/e2e_suite_test.go

@@ -17,6 +17,7 @@ limitations under the License.
 package e2e
 
 import (
+	"context"
 	"testing"
 	"time"
 
@@ -64,5 +65,8 @@ var _ = BeforeSuite(func() {
 	testClient, err = client.New(restConfig, client.Options{Scheme: scheme})
 	Expect(err).NotTo(HaveOccurred())
 
+	clientContext, clientCancel := context.WithCancel(context.Background())
 	komega.SetClient(testClient)
+	komega.SetContext(clientContext)
+	DeferCleanup(clientCancel)
 })

test/e2e/example_test.go

@@ -32,20 +32,18 @@ import (
 	. "github.com/timebertt/kubernetes-controller-sharding/pkg/utils/test/matchers"
 )
 
-var _ = Describe("Example Shard", Label("example"), Ordered, func() {
-	const controllerRingName = "example"
+var _ = Describe("Example Controller", Label("checksum-controller"), Ordered, func() {
+	const controllerRingName = "checksum-controller"
 
-	var (
-		controllerRing *shardingv1alpha1.ControllerRing
-	)
+	var controllerRing *shardingv1alpha1.ControllerRing
 
 	BeforeAll(func() {
 		controllerRing = &shardingv1alpha1.ControllerRing{ObjectMeta: metav1.ObjectMeta{Name: controllerRingName}}
 	})
 
 	Describe("setup", func() {
 		It("the Deployment should be healthy", func(ctx SpecContext) {
-			deployment := &appsv1.Deployment{ObjectMeta: metav1.ObjectMeta{Name: "shard", Namespace: metav1.NamespaceDefault}}
+			deployment := &appsv1.Deployment{ObjectMeta: metav1.ObjectMeta{Name: controllerRingName, Namespace: metav1.NamespaceDefault}}
 
 			Eventually(ctx, func(g Gomega) {
 				g.Expect(Get(deployment)()).To(Succeed())