Improve Azure client functions and add base URL

Issue: rancher/aks-operator#98 Signed-off-by: Michal Jura <mjura@suse.com>
mjura · Sep 29, 2023 · 2c02db5 · 2c02db5
1 parent 0ff5fe8
commit 2c02db5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/api/norman/customization/aks/listers.go b/pkg/api/norman/customization/aks/listers.go
@@ -55,7 +55,7 @@ func NewVirtualMachineSKUClient(cap *Capabilities) (*skus.ResourceSkusClient, er
 		return nil, err
 	}
 
-	skusClient := skus.NewResourceSkusClient(cap.SubscriptionID)
+	skusClient := skus.NewResourceSkusClientWithBaseURI(cap.BaseURL, cap.SubscriptionID)
 	skusClient.Authorizer = authorizer
 
 	return &skusClient, nil
@@ -67,7 +67,7 @@ func NewVirtualMachineClient(cap *Capabilities) (*compute.VirtualMachineSizesCli
 		return nil, err
 	}
 
-	virtualMachine := compute.NewVirtualMachineSizesClient(cap.SubscriptionID)
+	virtualMachine := compute.NewVirtualMachineSizesClientWithBaseURI(cap.BaseURL, cap.SubscriptionID)
 	virtualMachine.Authorizer = authorizer
 
 	return &virtualMachine, nil
@@ -115,7 +115,7 @@ func NewSubscriptionServiceClient(cap *Capabilities) (*subscription.Subscription
 		return nil, err
 	}
 
-	subscriptionService := subscription.NewSubscriptionsClient()
+	subscriptionService := subscription.NewSubscriptionsClientWithBaseURI(cap.BaseURL)
 	subscriptionService.Authorizer = authorizer
 
 	return &subscriptionService, nil