fix(vm): lazy-import shlex/ResourceExistsError and fix Invoke mock target

Author: Ashutosh0x

src/azure-cli/azure/cli/command_modules/vm/custom.py

@@ -15,7 +15,6 @@
 import json
 import os
 import uuid
-import shlex
 from datetime import datetime, timedelta
 
 import requests
@@ -25,7 +24,7 @@
 
 from knack.log import get_logger
 from knack.util import CLIError
-from azure.core.exceptions import ResourceExistsError
+
 from azure.cli.core.azclierror import (
     ResourceNotFoundError,
     ValidationError,
@@ -6602,6 +6601,8 @@ def _get_vm_and_rg(cmd, vm_name, rg=None):
 
 
 def vm_cp(cmd, source, destination, storage_account=None, container_name='azvmcp'):
+    import shlex
+    from azure.core.exceptions import ResourceExistsError
     from azure.cli.command_modules.storage.operations.blob import upload_blob, download_blob
     from azure.cli.command_modules.storage.util import create_short_lived_blob_sas_v2
     from azure.cli.command_modules.storage._client_factory import cf_sa, cf_sa_for_keys, cf_blob_service

src/azure-cli/azure/cli/command_modules/vm/tests/latest/test_vm_cp_unit.py

@@ -71,7 +71,7 @@ def test_vm_cp_upload_basic(self, mock_upload, mock_sas, mock_blob_factory, mock
         mock_container.get_blob_client.return_value = mock_blob
 
         # Execute
-        with mock.patch('azure.cli.command_modules.vm.custom.Invoke') as mock_invoke:
+        with mock.patch('azure.cli.command_modules.vm.aaz.latest.vm.run_command.Invoke') as mock_invoke:
             mock_invoke.return_value.return_value = {'value': [{'message': 'success'}]}
             vm_cp(cmd, source="local.txt", destination="myrg:myvm:/tmp/remote.txt")
 
@@ -123,7 +123,7 @@ def test_vm_cp_download_basic(self, mock_download, mock_blob_factory, mock_keys_
         mock_container.get_blob_client.return_value = mock_blob
 
         # Execute
-        with mock.patch('azure.cli.command_modules.vm.custom.Invoke') as mock_invoke:
+        with mock.patch('azure.cli.command_modules.vm.aaz.latest.vm.run_command.Invoke') as mock_invoke:
             mock_invoke.return_value.return_value = {'value': [{'message': 'success'}]}
             vm_cp(cmd, source="myrg:myvm:/tmp/remote.txt", destination="local.txt")