package azure

import (
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/tidbcloud/infra-provider/regional-server/test/testdata"
)

func TestConvertToWorkloadIdentityLogin(t *testing.T) {
	converted, err := ConvertToWorkloadIdentityLogin([]byte(testdata.OriginKubeConfig))
	require.NoError(t, err)
	require.Equal(t, testdata.ExpectedKubeConfig, string(converted))
}

func TestConvertToWorkloadIdentityLoginCrossTenant(t *testing.T) {
	converted, err := ConvertToWorkloadIdentityLoginCrossTenant(
		[]byte(testdata.OriginKubeConfig),
		"tidbcloud-multi-tenant-app",
		"customer-tenant-id",
	)
	require.NoError(t, err)
	require.Equal(t, testdata.ExpectedCrossTenantKubeConfig, string(converted))
}