package azure

import (
	"fmt"
	"testing"

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

func TestConvertToWorkloadIdentityLogin(t *testing.T) {
	testData := []struct {
		name        string
		input       []byte
		expected    []byte
		expectedErr error
	}{
		{
			name:        "no authInfos found in kubeconfig",
			input:       []byte(``),
			expectedErr: fmt.Errorf("no authInfos found in kubeconfig"),
		},
		{
			name:        "failed to load kubeconfig",
			input:       []byte(`invalid`),
			expectedErr: fmt.Errorf("failed to load kubeconfig"),
		},
		{
			name:     "success",
			input:    []byte(testdata.OriginKubeConfig),
			expected: []byte(testdata.ExpectedKubeConfig),
		},
	}

	for _, tt := range testData {
		t.Run(tt.name, func(t *testing.T) {
			actual, err := ConvertToWorkloadIdentityLogin(tt.input)
			if tt.expectedErr != nil {
				assert.ErrorContains(t, err, tt.expectedErr.Error())
				return
			}
			assert.Nil(t, err)
			assert.Equal(t, string(tt.expected), string(actual))
		})
	}
}