auto migration cases

Author: sunzhaohua2

e2e/framework/machineset.go

@@ -8,9 +8,16 @@ import (
 	corev1 "k8s.io/api/core/v1"
 	apierrors "k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/client-go/discovery"
+	"k8s.io/client-go/dynamic"
+	"k8s.io/client-go/rest"
+	"k8s.io/client-go/scale"
 	"k8s.io/utils/pointer"
 	clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
 	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/client/apiutil"
+	"sigs.k8s.io/controller-runtime/pkg/client/config"
 )
 
 type machineSetParams struct {
@@ -188,3 +195,55 @@ func GetMachinesFromMachineSet(cl client.Client, machineSet *clusterv1.MachineSe
 	}
 	return machinesForSet, nil
 }
+
+
+// ScaleMachineSet scales a machineSet with a given name to the given number of replicas.
+func ScaleMachineSet(name string, replicas int) error {
+	scaleClient, err := getScaleClient()
+	if err != nil {
+		return fmt.Errorf("error calling getScaleClient %w", err)
+	}
+
+	scale, err := scaleClient.Scales(CAPINamespace).Get(ctx, schema.GroupResource{Group: "cluster.x-k8s.io", Resource: "MachineSet"}, name, metav1.GetOptions{})
+	if err != nil {
+		return fmt.Errorf("error calling scaleClient.Scales get: %w", err)
+	}
+
+	scaleUpdate := scale.DeepCopy()
+	scaleUpdate.Spec.Replicas = int32(replicas)
+
+	_, err = scaleClient.Scales(CAPINamespace).Update(ctx, schema.GroupResource{Group: "cluster.x-k8s.io", Resource: "MachineSet"}, scaleUpdate, metav1.UpdateOptions{})
+	if err != nil {
+		return fmt.Errorf("error calling scaleClient.Scales update: %w", err)
+	}
+
+	return nil
+}
+
+// getScaleClient returns a ScalesGetter object to manipulate scale subresources.
+func getScaleClient() (scale.ScalesGetter, error) {
+	cfg, err := config.GetConfig()
+	if err != nil {
+		return nil, fmt.Errorf("error getting config %w", err)
+	}
+
+	httpClient, err := rest.HTTPClientFor(cfg)
+	if err != nil {
+		return nil, fmt.Errorf("error calling rest.HTTPClientFor %w", err)
+	}
+
+	mapper, err := apiutil.NewDynamicRESTMapper(cfg, httpClient)
+	if err != nil {
+		return nil, fmt.Errorf("error calling NewDiscoveryRESTMapper %w", err)
+	}
+
+	discovery := discovery.NewDiscoveryClientForConfigOrDie(cfg)
+	scaleKindResolver := scale.NewDiscoveryScaleKindResolver(discovery)
+
+	scaleClient, err := scale.NewForConfig(cfg, mapper, dynamic.LegacyAPIPathResolverFunc, scaleKindResolver)
+	if err != nil {
+		return nil, fmt.Errorf("error calling building scale client %w", err)
+	}
+
+	return scaleClient, nil
+}

e2e/framework/machinetemplate.go

@@ -0,0 +1,41 @@
+package framework
+
+import (
+	awsv1 "sigs.k8s.io/cluster-api-provider-aws/v2/api/v1beta2"
+	azurev1 "sigs.k8s.io/cluster-api-provider-azure/api/v1beta1"
+	gcpv1 "sigs.k8s.io/cluster-api-provider-gcp/api/v1beta1"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+)
+
+// GetAWSMachineTemplate gets awsMachineTemplate by its name from the default cluster API namespace.
+func GetAWSMachineTemplate(cl client.Client, name string) (*awsv1.AWSMachineTemplate, error) {
+	var awsMachineTemplate = &awsv1.AWSMachineTemplate{}
+
+	key := client.ObjectKey{Namespace: CAPINamespace, Name: name}
+	if err := cl.Get(ctx, key, awsMachineTemplate); err != nil {
+		return nil, err
+	}
+	return awsMachineTemplate, nil
+}
+
+// GetAzureMachineTemplate gets azureMachineTemplate by its name from the default cluster API namespace.
+func GetAzureMachineTemplate(cl client.Client, name string) (*azurev1.AzureMachineTemplate, error) {
+	var azureMachineTemplate = &azurev1.AzureMachineTemplate{}
+
+	key := client.ObjectKey{Namespace: CAPINamespace, Name: name}
+	if err := cl.Get(ctx, key, azureMachineTemplate); err != nil {
+		return nil, err
+	}
+	return azureMachineTemplate, nil
+}
+
+// GetGCPMachineTemplate gets gcpMachineTemplate by its name from the default cluster API namespace.
+func GetGCPMachineTemplate(cl client.Client, name string) (*gcpv1.GCPMachineTemplate, error) {
+	var gcpMachineTemplate = &gcpv1.GCPMachineTemplate{}
+
+	key := client.ObjectKey{Namespace: CAPINamespace, Name: name}
+	if err := cl.Get(ctx, key, gcpMachineTemplate); err != nil {
+		return nil, err
+	}
+	return gcpMachineTemplate, nil
+}

e2e/framework/util.go

@@ -1,11 +1,13 @@
 package framework
 
 import (
+	"context"
 	"fmt"
 	"net/url"
 	"strconv"
 
 	configv1 "github.com/openshift/api/config/v1"
+	"k8s.io/apimachinery/pkg/types"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 )
 
@@ -32,3 +34,13 @@ func GetControlPlaneHostAndPort(cl client.Client) (string, int32, error) {
 
 	return apiUrl.Hostname(), int32(port), nil
 }
+
+// IsTechPreviewNoUpgrade checks if a cluster is a TechPreviewNoUpgrade cluster
+func IsTechPreviewNoUpgrade(ctx context.Context,  cl client.Client) bool {
+	featureGate := &configv1.FeatureGate{}
+	if err := cl.Get(ctx, types.NamespacedName{Name: "cluster"}, featureGate); err != nil {
+		return false
+	}
+
+	return featureGate.Spec.FeatureSet == configv1.TechPreviewNoUpgrade
+}