spinifex/tests/e2e/harness/eks.go

//go:build e2e

package harness

import (
	"errors"
	"fmt"
	"strings"
	"testing"
	"time"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/awserr"
	"github.com/aws/aws-sdk-go/service/eks"
)

// WaitForEKSClusterActive polls DescribeCluster until status == ACTIVE. Control
// plane provisioning boots a K3s server VM + NLB, so default timeout is 10min.
func WaitForEKSClusterActive(t *testing.T, c *AWSClient, name string, opts ...PollOpt) *eks.Cluster {
	t.Helper()
	cfg := applyOpts(pollCfg{timeout: 10 * time.Minute, interval: 5 * time.Second}, opts...)
	var last *eks.Cluster
	EventuallyErr(t, func() error {
		out, err := c.EKS.DescribeCluster(&eks.DescribeClusterInput{Name: aws.String(name)})
		if err != nil {
			return fmt.Errorf("describe-cluster %s: %w", name, err)
		}
		last = out.Cluster
		state := aws.StringValue(last.Status)
		switch state {
		case eks.ClusterStatusActive:
			return nil
		case eks.ClusterStatusFailed:
			return fmt.Errorf("%s entered FAILED state", name)
		default:
			return fmt.Errorf("%s status=%s want=ACTIVE", name, state)
		}
	}, cfg.timeout, cfg.interval)
	t.Logf("cluster %s reached status ACTIVE", name)
	return last
}

// WaitForEKSClusterDeleted polls DescribeCluster until it returns
// ResourceNotFoundException (cluster fully gone).
func WaitForEKSClusterDeleted(t *testing.T, c *AWSClient, name string, opts ...PollOpt) {
	t.Helper()
	cfg := applyOpts(pollCfg{timeout: 10 * time.Minute, interval: 5 * time.Second}, opts...)
	EventuallyErr(t, func() error {
		_, err := c.EKS.DescribeCluster(&eks.DescribeClusterInput{Name: aws.String(name)})
		if err == nil {
			return fmt.Errorf("%s still exists", name)
		}
		if isClusterNotFound(err) {
			return nil
		}
		return fmt.Errorf("describe-cluster %s: %w", name, err)
	}, cfg.timeout, cfg.interval)
	t.Logf("cluster %s deleted", name)
}

// isClusterNotFound reports whether err is the EKS "cluster gone" error. The
// SDK constant is "ResourceNotFoundException" but awsgw emits the doubled
// "ResourceNotFoundExceptionException" on the wire, so match by substring.
func isClusterNotFound(err error) bool {
	if err == nil {
		return false
	}
	var aerr awserr.Error
	if errors.As(err, &aerr) {
		return strings.Contains(aerr.Code(), "ResourceNotFound")
	}
	return false
}