Add generated new SKU type data

Signed-off-by: Jack Francis <[email protected]>

Author: jackfrancis

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/Azure/go-autorest/autorest/to v0.4.0
 	github.com/google/go-cmp v0.5.9
 	github.com/pkg/errors v0.9.1
+	k8s.io/utils v0.0.0-20240821151609-f90d01438635
 )
 
 require (

go.sum

@@ -95,3 +95,5 @@ gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+k8s.io/utils v0.0.0-20240821151609-f90d01438635 h1:2wThSvJoW/Ncn9TmQEYXRnevZXi2duqHWf5OX9S3zjI=
+k8s.io/utils v0.0.0-20240821151609-f90d01438635/go.mod h1:OLgZIPagt7ERELqWJFomSt595RzquPNLL48iOWgYOg0=

hack/generate_vmsize_testdata.go

@@ -11,10 +11,10 @@ import (
 	"github.com/Azure/skewer/testdata"
 )
 
-func getSKUs(subscriptionID, region string) (map[string]testdata.SKUInfo, error) {
+func getSKUs(subscriptionID, region string) (map[string]testdata.SKUInfo, []testdata.SKU, error) {
 	authorizer, err := auth.NewAuthorizerFromCLI()
 	if err != nil {
-		return nil, err
+		return nil, nil, err
 	}
 
 	// Create a new compute client
@@ -24,25 +24,27 @@ func getSKUs(subscriptionID, region string) (map[string]testdata.SKUInfo, error)
 	// List SKUs for the specified region
 	skuList, err := client.List(context.Background(), region, "")
 	if err != nil {
-		return nil, err
+		return nil, nil, err
 	}
 
-	skus := map[string]testdata.SKUInfo{}
+	skuInfoMap := map[string]testdata.SKUInfo{}
+	skuSlice := []testdata.SKU{}
 	for _, sku := range skuList.Values() {
 		if *sku.ResourceType == "virtualMachines" {
-			if _, ok := skus[*sku.Name]; !ok {
+			if _, ok := skuInfoMap[*sku.Name]; !ok {
+				skuSlice = append(skuSlice, testdata.SKU{Name: sku.Name})
 				skuInfo := testdata.SKUInfo{
 					Size: *sku.Size,
 				}
-				skus[*sku.Name] = skuInfo
+				skuInfoMap[*sku.Name] = skuInfo
 			}
 		}
 	}
 
-	return skus, nil
+	return skuInfoMap, skuSlice, nil
 }
 
-const templateCode = `package testdata
+const vmSizeTestDataTemplate = `package testdata
 
 type SKUInfo struct {
 	Size string
@@ -57,22 +59,60 @@ var SKUData = map[string]SKUInfo{
 }
 `
 
-func generateAndSaveFile(skus map[string]testdata.SKUInfo) error {
-	file, err := os.Create("../testdata/generated_vmsize_testdata.go")
+const vmSKUsTestDataTemplate = `package testdata
+
+import (
+	"k8s.io/utils/ptr"
+)
+
+type SKU struct {
+	Name *string
+}
+
+var SKUs = []SKU{
+{{- range $i, $data := .}}
+	{
+		Name:     ptr.To("{{ $data.Name }}"),
+	},
+{{- end }}
+}
+`
+
+func generateAndSaveFiles(skuInfoMap map[string]testdata.SKUInfo, skuSlice []testdata.SKU) error {
+	vmSizeGeneratedFilename := "testdata/generated_vmsize_testdata.go"
+	file, err := os.Create(fmt.Sprintf("../%s", vmSizeGeneratedFilename))
 	if err != nil {
 		return err
 	}
 	defer file.Close()
 
-	tmpl, err := template.New("skus_template").Parse(templateCode)
+	tmpl, err := template.New("vmSize_test_data_template").Parse(vmSizeTestDataTemplate)
+	if err != nil {
+		return err
+	}
+
+	err = tmpl.Execute(file, skuInfoMap)
+	if err != nil {
+		return err
+	}
+	fmt.Printf("Generated and saved %s successfully!\n", vmSizeGeneratedFilename)
+
+	vmSKUsGeneratedFilename := "testdata/generated_vmskus_testdata.go"
+	file, err = os.Create(fmt.Sprintf("../%s", vmSKUsGeneratedFilename))
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+	tmpl, err = template.New("vmSKUs_test_data_template").Parse(vmSKUsTestDataTemplate)
 	if err != nil {
 		return err
 	}
 
-	err = tmpl.Execute(file, skus)
+	err = tmpl.Execute(file, skuSlice)
 	if err != nil {
 		return err
 	}
+	fmt.Printf("Generated and saved %s successfully!\n", vmSKUsGeneratedFilename)
 
 	return nil
 }
@@ -87,17 +127,15 @@ func main() {
 		region = "eastus" // Default region if not provided in the environment variable
 	}
 
-	skus, err := getSKUs(subscriptionID, region)
+	skuInfoMap, skuSlice, err := getSKUs(subscriptionID, region)
 	if err != nil {
 		fmt.Println("Error fetching SKUs:", err)
 		return
 	}
 
-	err = generateAndSaveFile(skus)
+	err = generateAndSaveFiles(skuInfoMap, skuSlice)
 	if err != nil {
 		fmt.Println("Error generating and saving file:", err)
 		return
 	}
-
-	fmt.Println("Generated and saved skudata/skus_generated.go successfully!")
 }