fix: Resolve errors for resource not found (#633)

Co-authored-by: yl5722 <[email protected]> Co-authored-by: Alex Leites <[email protected]>
Azure · Jan 23, 2025 · 5a4e5d6 · 5a4e5d6
1 parent ff53ecc
commit 5a4e5d6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pkg/providers/instance/armutils.go b/pkg/providers/instance/armutils.go
@@ -126,7 +126,8 @@ func deleteNicIfExists(ctx context.Context, client NetworkInterfacesAPI, rg, nic
 func deleteVirtualMachineIfExists(ctx context.Context, client VirtualMachinesAPI, rg, vmName string) error {
 	_, err := client.Get(ctx, rg, vmName, nil)
 	if err != nil {
-		if sdkerrors.IsNotFoundErr(err) {
+		azErr := sdkerrors.IsResponseError(err)
+		if azErr != nil && (azErr.ErrorCode == "NotFound" || azErr.ErrorCode == "ResourceNotFound") {
 			return nil
 		}
 		return err

diff --git a/pkg/providers/instance/instance.go b/pkg/providers/instance/instance.go
@@ -148,7 +148,8 @@ func (p *Provider) Get(ctx context.Context, vmName string) (*armcompute.VirtualM
 	var err error
 
 	if vm, err = p.AZClient.virtualMachinesClient.Get(ctx, p.resourceGroup, vmName, nil); err != nil {
-		if sdkerrors.IsNotFoundErr(err) {
+		azErr := sdkerrors.IsResponseError(err)
+		if azErr != nil && (azErr.ErrorCode == "NotFound" || azErr.ErrorCode == "ResourceNotFound") {
 			return nil, corecloudprovider.NewNodeClaimNotFoundError(err)
 		}
 		return nil, fmt.Errorf("failed to get VM instance, %w", err)