mapper_gcp_services_test.go

Documentation: edge-infra.dev/pkg/edge/api/graph/mapper

     1  package mapper
     2  
     3  import (
     4  	"testing"
     5  
     6  	"edge-infra.dev/pkg/edge/api/graph/model"
     7  
     8  	"github.com/stretchr/testify/assert"
     9  	"google.golang.org/api/compute/v1"
    10  )
    11  
    12  // Test ToServiceAccount if it is not empty
    13  func TestToServiceAccountNotEmpty(t *testing.T) {
    14  	servAcc, _ := ToServiceAccount(getTestServiceAccountSecret("test-service-account", fakeServiceAccountSecret))
    15  	verifyServiceAccountEqualModel(servAcc, getTestServiceAccount(), t)
    16  }
    17  
    18  // Test ToServiceAccount if it is incorrect
    19  func TestToServiceAccountIsIncorrect(t *testing.T) {
    20  	badServAcc, _ := ToServiceAccount(getTestServiceAccountSecret("test-service-account", badServiceAccount))
    21  	if badServAcc != nil {
    22  		t.Fatalf("result = %v. Should equal nil", badServAcc)
    23  	}
    24  }
    25  
    26  // Test ToZones
    27  func TestToZones(t *testing.T) {
    28  	zoneList := ToZones(getTestZoneList())
    29  	strZones := getTestStringArr("us-east1-a", "us-central2-b")
    30  	verifyZonesEqualModel(zoneList, strZones, t)
    31  
    32  	//test empty array
    33  	zoneEmptyList := ToZones(&compute.ZoneList{})
    34  	assert.Equal(t, []*string{}, zoneEmptyList)
    35  }
    36  
    37  // Test ToMachineTypes
    38  func TestToMachineTypes(t *testing.T) {
    39  	machineList := ToMachineTypes(getTestMachineList())
    40  	testMachineInfo := getTestMachineInfo("n1-standard-2", 4, 60)
    41  	for _, machineType := range machineList {
    42  		verifyMachineTypeInforEqualModel(machineType, testMachineInfo, t)
    43  	}
    44  
    45  	//test empty array
    46  	machineList = ToMachineTypes(&compute.MachineTypeList{})
    47  	assert.Equal(t, []*model.MachineTypeInfo{}, machineList)
    48  }
    49  
    50  // Test ToGkeVersions
    51  func TestToGkeVersions(t *testing.T) {
    52  	gkeList := ToGkeVersions(getTestGkeVersions())
    53  	strGkeVersions := getTestStringArr("1.4-gke-test", "1.4.5-gke-test")
    54  	verifyZonesEqualModel(gkeList, strGkeVersions, t)
    55  
    56  	//test empty array
    57  	gkeList = ToGkeVersions([]string{})
    58  	assert.Equal(t, []*string{}, gkeList)
    59  }
    60
View as plain text