Check separately

kubernetes-sigs · Jan 8, 2024 · 405a1d9 · 405a1d9
1 parent 62f9fb3
commit 405a1d9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/e2e/aks_marketplace.go b/test/e2e/aks_marketplace.go
@@ -78,7 +78,11 @@ func AKSMarketplaceExtensionSpec(ctx context.Context, inputGetter func() AKSMark
 				},
 			},
 		}
-		g.Expect(mgmtClient.Update(ctx, infraControlPlane)).To(Succeed())
+	}, input.WaitExtensionIntervals...).Should(Succeed())
+
+	Eventually(func(g Gomega) {
+		err = mgmtClient.Get(ctx, client.ObjectKey{Namespace: input.Cluster.Spec.ControlPlaneRef.Namespace, Name: input.Cluster.Spec.ControlPlaneRef.Name}, infraControlPlane)
+		g.Expect(err).NotTo(HaveOccurred())
 		g.Expect(conditions.IsTrue(infraControlPlane, infrav1.AKSExtensionsReadyCondition)).To(BeTrue())
 	}, input.WaitExtensionIntervals...).Should(Succeed())