migtation case

update

go mod

Revert "go mod"

This reverts commit 34be27d.

Author: huali9

e2e/framework/machine.go

@@ -1,11 +1,19 @@
 package framework
 
 import (
+	"context"
 	"fmt"
+	"time"
 
+	. "github.com/onsi/gomega"
+	apierrors "k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/util/wait"
+	"k8s.io/klog/v2"
+	awsv1 "sigs.k8s.io/cluster-api-provider-aws/v2/api/v1beta2"
 	clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
 	"sigs.k8s.io/controller-runtime/pkg/client"
+	runtimeclient "sigs.k8s.io/controller-runtime/pkg/client"
 )
 
 // GetMachines gets a list of machines from the default cluster API namespace.
@@ -41,6 +49,30 @@ func GetMachines(cl client.Client, selectors ...*metav1.LabelSelector) ([]*clust
 	return machines, nil
 }
 
+// GetMachine get a machine by its name from the cluster API namespace.
+func GetMachine(cl client.Client, name string) (*clusterv1.Machine, error) {
+	machine := &clusterv1.Machine{}
+	key := client.ObjectKey{Namespace: CAPINamespace, Name: name}
+
+	if err := cl.Get(context.Background(), key, machine); err != nil {
+		return nil, fmt.Errorf("error querying api for machine object: %w", err)
+	}
+
+	return machine, nil
+}
+
+// GetAWSMachine get a awsmachine by its name from the cluster API namespace.
+func GetAWSMachine(cl client.Client, name string) (*awsv1.AWSMachine, error) {
+	machine := &awsv1.AWSMachine{}
+	key := client.ObjectKey{Namespace: CAPINamespace, Name: name}
+
+	if err := cl.Get(context.Background(), key, machine); err != nil {
+		return nil, fmt.Errorf("error querying api for awsmachine object: %w", err)
+	}
+
+	return machine, nil
+}
+
 // FilterRunningMachines returns a slice of only those Machines in the input
 // that are in the "Running" phase.
 func FilterRunningMachines(machines []*clusterv1.Machine) []*clusterv1.Machine {
@@ -54,3 +86,33 @@ func FilterRunningMachines(machines []*clusterv1.Machine) []*clusterv1.Machine {
 
 	return result
 }
+
+// DeleteMachines deletes the specified machines and returns an error on failure.
+func DeleteMachines(cl client.Client, machines ...*clusterv1.Machine) error {
+	return wait.PollUntilContextTimeout(ctx, RetryShort, time.Minute, true, func(ctx context.Context) (bool, error) {
+		for _, machine := range machines {
+			if err := cl.Delete(ctx, machine); err != nil {
+				klog.Errorf("Error querying api for machine object %q: %v, retrying...", machine.Name, err)
+				return false, err
+			}
+		}
+
+		return true, nil
+	})
+}
+
+// WaitForMachinesDeleted polls until the given Machines are not found.
+func WaitForMachinesDeleted(cl client.Client, machines ...*clusterv1.Machine) {
+	Eventually(func() bool {
+		for _, m := range machines {
+			if err := cl.Get(context.Background(), runtimeclient.ObjectKey{
+				Name:      m.GetName(),
+				Namespace: m.GetNamespace(),
+			}, &clusterv1.Machine{}); !apierrors.IsNotFound(err) {
+				return false // Not deleted, or other error.
+			}
+		}
+
+		return true // Everything was deleted.
+	}, WaitLong, RetryMedium).Should(BeTrue(), "error encountered while waiting for Machines to be deleted.")
+}