//go:build unit || all
// +build unit all

package eks

import (
	"testing"

	"github.com/onsi/gomega"
	"github.com/tidbcloud/infra-provider/pkg/k8s"
)

func TestEnsureDMNodeRoleARNReady(t *testing.T) {
	g := gomega.NewWithT(t)

	// empty ARN should request a healthy dependency wait
	res := ensureDMNodeRoleARNReady("")
	g.Expect(res.RawError()).NotTo(gomega.BeNil())
	g.Expect(k8s.ShouldRequeue(res.RawError()).RequeueAfter).To(gomega.Equal(iamReadyRequeueInterval))

	// non-empty ARN should be done
	res = ensureDMNodeRoleARNReady("arn:aws:iam::123456789012:role/dm-role")
	g.Expect(res.RawError()).To(gomega.BeNil())
}