Fix unit test

Signed-off-by: Lubron Zhan <[email protected]>
vmware-tanzu · Mar 13, 2024 · 5697b6a · 5697b6a
1 parent 12a35d3
commit 5697b6a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/controllers/machine/machine_controller_intg_test.go b/controllers/machine/machine_controller_intg_test.go
@@ -93,10 +93,11 @@ func intgTestMachineController() {
 			})
 			It("Corresponding Endpoints should be created", func() {
 				ep := &corev1.Endpoints{}
-				Eventually(func() error {
-					return ctx.Client.Get(ctx.Context, client.ObjectKey{Name: cluster.Namespace + "-" + cluster.Name + "-control-plane", Namespace: cluster.Namespace}, ep)
-				}).Should(Succeed())
 				Eventually(func() int {
+					err := ctx.Client.Get(ctx.Context, client.ObjectKey{Name: cluster.Namespace + "-" + cluster.Name + "-control-plane", Namespace: cluster.Namespace}, ep)
+					if err != nil {
+						return 0
+					}
 					if len(ep.Subsets) == 0 {
 						return 0
 					}