package utils_test

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/tidbcloud/auto-deploy/internal/utils"
)

func TestGetAzureAvailabilityZonesN(t *testing.T) {
	zones, err := utils.GetAzureAvailabilityZonesN("japaneast", 3)
	assert.NoError(t, err)
	assert.Equal(t, []string{"japaneast-1", "japaneast-2", "japaneast-3"}, zones)
	zones, err = utils.GetAzureAvailabilityZonesN("eastus2", 3)
	assert.NoError(t, err)
	assert.Equal(t, []string{"eastus2-1", "eastus2-2", "eastus2-3"}, zones)
}