Skip to content

Commit 314acc0

Browse files
authored
Add support for new Additive Capabilities (#32)
* Regenerate testdata * Add tests * Fix bug in testdata generation tool that caused v2 and v1 to be different. Moved both to scan all regions (not every VM size is available in eastus, better to scan everything it gives a more robust test)
1 parent 1831119 commit 314acc0

8 files changed

Lines changed: 3918 additions & 321 deletions

File tree

hack/generate_vmsize_testdata.go

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ import (
1111
"github.com/Azure/skewer/testdata"
1212
)
1313

14-
func getSKUs(subscriptionID, region string) (map[string]testdata.SKUInfo, error) {
14+
func getSKUs(subscriptionID string) (map[string]testdata.SKUInfo, error) {
1515
authorizer, err := auth.NewAuthorizerFromCLI()
1616
if err != nil {
1717
return nil, err
@@ -22,7 +22,7 @@ func getSKUs(subscriptionID, region string) (map[string]testdata.SKUInfo, error)
2222
client.Authorizer = authorizer
2323

2424
// List SKUs for the specified region
25-
skuList, err := client.List(context.Background(), region, "")
25+
skuList, err := client.List(context.Background(), "", "")
2626
if err != nil {
2727
return nil, err
2828
}
@@ -51,7 +51,7 @@ type SKUInfo struct {
5151
var SKUData = map[string]SKUInfo{
5252
{{- range $key, $value := .}}
5353
"{{ $key }}": {
54-
Size: "{{ $value.Size }}",
54+
Size: "{{ $value.Size }}",
5555
},
5656
{{- end }}
5757
}
@@ -81,13 +81,7 @@ func main() {
8181
// Get the subscription ID from the environment variable or use a default value
8282
subscriptionID := os.Getenv("AZURE_SUBSCRIPTION_ID")
8383

84-
// Get the region from the environment variable or use a default value
85-
region := os.Getenv("AZURE_REGION")
86-
if region == "" {
87-
region = "eastus" // Default region if not provided in the environment variable
88-
}
89-
90-
skus, err := getSKUs(subscriptionID, region)
84+
skus, err := getSKUs(subscriptionID)
9185
if err != nil {
9286
fmt.Println("Error fetching SKUs:", err)
9387
return

0 commit comments

Comments
 (0)