package mapper

import (
	"testing"

	"edge-infra.dev/pkg/edge/api/graph/model"

	"github.com/stretchr/testify/assert"
	"google.golang.org/api/compute/v1"
)

// Test ToServiceAccount if it is not empty
func TestToServiceAccountNotEmpty(t *testing.T) {
	servAcc, _ := ToServiceAccount(getTestServiceAccountSecret("test-service-account", fakeServiceAccountSecret))
	verifyServiceAccountEqualModel(servAcc, getTestServiceAccount(), t)
}

// Test ToServiceAccount if it is incorrect
func TestToServiceAccountIsIncorrect(t *testing.T) {
	badServAcc, _ := ToServiceAccount(getTestServiceAccountSecret("test-service-account", badServiceAccount))
	if badServAcc != nil {
		t.Fatalf("result = %v. Should equal nil", badServAcc)
	}
}

// Test ToZones
func TestToZones(t *testing.T) {
	zoneList := ToZones(getTestZoneList())
	strZones := getTestStringArr("us-east1-a", "us-central2-b")
	verifyZonesEqualModel(zoneList, strZones, t)

	//test empty array
	zoneEmptyList := ToZones(&compute.ZoneList{})
	assert.Equal(t, []*string{}, zoneEmptyList)
}

// Test ToMachineTypes
func TestToMachineTypes(t *testing.T) {
	machineList := ToMachineTypes(getTestMachineList())
	testMachineInfo := getTestMachineInfo("n1-standard-2", 4, 60)
	for _, machineType := range machineList {
		verifyMachineTypeInforEqualModel(machineType, testMachineInfo, t)
	}

	//test empty array
	machineList = ToMachineTypes(&compute.MachineTypeList{})
	assert.Equal(t, []*model.MachineTypeInfo{}, machineList)
}

// Test ToGkeVersions
func TestToGkeVersions(t *testing.T) {
	gkeList := ToGkeVersions(getTestGkeVersions())
	strGkeVersions := getTestStringArr("1.4-gke-test", "1.4.5-gke-test")
	verifyZonesEqualModel(gkeList, strGkeVersions, t)

	//test empty array
	gkeList = ToGkeVersions([]string{})
	assert.Equal(t, []*string{}, gkeList)
}