package constant

import (
	"testing"

	. "github.com/onsi/gomega"
)

func TestFreeTierProviderType(t *testing.T) {
	g := NewGomegaWithT(t)

	g.Expect(FreeTierProviderType("")).To(Equal(ProviderTypeAWSFreeTier))
	g.Expect(FreeTierProviderType("aws")).To(Equal(ProviderTypeAWSFreeTier))
	g.Expect(FreeTierProviderType("alicloud")).To(Equal(ProviderTypeAliCloudFreeTier))
	g.Expect(FreeTierProviderType("gcp")).To(Equal(ProviderTypeGCPFreeTier))
	g.Expect(FreeTierProviderType("azure")).To(Equal("azure-free-tier"))
	g.Expect(AllFreeTierProviderTypes()).To(Equal([]string{
		ProviderTypeAWSFreeTier,
		ProviderTypeAliCloudFreeTier,
		ProviderTypeGCPFreeTier,
	}))
}