Wait for md rollout after rebase

Signed-off-by: Danil-Grigorev <[email protected]>

test/e2e/clusterclass_changes.go

@@ -790,10 +790,10 @@ func rebaseClusterClassAndWait(ctx context.Context, input rebaseClusterClassAndW
 		// NOTE: We only wait until the change is rolled out to the MachineDeployment objects and not to the worker machines
 		// to speed up the test and focus the test on the ClusterClass feature.
 		log.Logf("Waiting for MachineDeployment rollout for MachineDeploymentTopology %q (class %q) to complete.", mdTopology.Name, mdTopology.Class)
-		Eventually(func() error {
+		Eventually(func(g Gomega) error {
 			// Get MachineDeployment for the current MachineDeploymentTopology.
 			mdList := &clusterv1.MachineDeploymentList{}
-			Expect(mgmtClient.List(ctx, mdList, client.InNamespace(input.Cluster.Namespace), client.MatchingLabels{
+			g.Expect(mgmtClient.List(ctx, mdList, client.InNamespace(input.Cluster.Namespace), client.MatchingLabels{
 				clusterv1.ClusterTopologyMachineDeploymentNameLabel: mdTopology.Name,
 			})).To(Succeed())
 			if len(mdList.Items) != 1 {
@@ -810,6 +810,10 @@ func rebaseClusterClassAndWait(ctx context.Context, input rebaseClusterClassAndW
 				return errors.Errorf("label %q should be %q, but is %q", testWorkerLabelName, mdTopology.Class, labelValue)
 			}
 
+			g.Expect(md.Spec.Replicas).To(Equal(md.Status.Replicas), "Machine deployment does not have expected replicas")
+			g.Expect(md.Spec.Replicas).To(Equal(md.Status.AvailableReplicas), "Machine deployment does not have expected available replicas")
+			g.Expect(md.Spec.Replicas).To(Equal(md.Status.ReadyReplicas), "Machine deployment does not have expected ready replicas")
+
 			return nil
 		}, input.WaitForMachineDeployments...).Should(BeNil())
 	}