update

kubernetes-sigs · Sep 13, 2023 · b5ae572 · b5ae572
1 parent e5fefd2
commit b5ae572
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/controllers/serviceaccount_controller_unit_test.go b/controllers/serviceaccount_controller_unit_test.go
@@ -45,10 +45,13 @@ func unitTestsReconcileNormal() {
 	)
 
 	JustBeforeEach(func() {
+		controllerCtx = helpers.NewUnitTestContextForController(namespace, vsphereCluster, false, initObjects, nil)
 		// Note: The service account provider requires a reference to the vSphereCluster hence the need to create
 		// a fake vSphereCluster in the test and pass it to during context setup.
-		reconciler = ServiceAccountReconciler{}
-		controllerCtx = helpers.NewUnitTestContextForController(namespace, vsphereCluster, false, initObjects, nil)
+		reconciler = ServiceAccountReconciler{
+			// Note: This also sets the client.Client.
+			ControllerContext: controllerCtx.ControllerContext,
+		}
 		_, err := reconciler.ReconcileNormal(ctx, controllerCtx.GuestClusterContext)
 		Expect(err).NotTo(HaveOccurred())
 

diff --git a/controllers/svcdiscovery_controller_unit_test.go b/controllers/svcdiscovery_controller_unit_test.go
@@ -46,6 +46,10 @@ func serviceDiscoveryUnitTestsReconcileNormal() {
 	JustBeforeEach(func() {
 		vsphereCluster = fake.NewVSphereCluster(namespace)
 		controllerCtx = helpers.NewUnitTestContextForController(namespace, &vsphereCluster, false, initObjects, nil)
+		reconciler = serviceDiscoveryReconciler{
+			// Note: This also sets the client.Client.
+			ControllerContext: controllerCtx.ControllerContext,
+		}
 		_, err := reconciler.ReconcileNormal(ctx, controllerCtx.GuestClusterContext)
 		Expect(err).NotTo(HaveOccurred())