def patch_retrieve_token_for_user(unit_test):
def _retrieve_token_for_user(*args, **kwargs): # pylint: disable=unused-argument
return 'Bearer', 'top-secret-token-for-you', None
mock_in_unit_test(unit_test,
'azure.cli.core._profile.CredsCache.retrieve_token_for_user',
_retrieve_token_for_user)
def patch_uuid_str(unit_test):
def _mock_get_uuid_str(*args, **kwargs): # pylint: disable=unused-argument
return MOCK_JOB_ID
mock_in_unit_test(unit_test,
'azure.cli.command_modules.dla.custom._get_uuid_str',
_mock_get_uuid_str)
def patch_get_current_system_username(unit_test):
def _get_current_system_username(*args, **kwargs): # pylint: disable=unused-argument
from .utilities import create_random_name
return create_random_name(prefix='example_')
mock_in_unit_test(
unit_test, 'azure.cli.core.local_context._get_current_system_username',
_get_current_system_username)
def patch_retrieve_token_for_user(unit_test):
def _retrieve_token_for_user(*args, **kwargs): # pylint: disable=unused-argument
return 'Bearer', 'top-secret-token-for-you', None
mock_in_unit_test(
unit_test,
'azure.cli.core._profile.CredsCache.retrieve_token_for_user',
_retrieve_token_for_user)
def patch_ids(unit_test):
def _mock_uuid(*args, **kwargs): # pylint: disable=unused-argument
return MOCK_UUID
mock_in_unit_test(unit_test, f'{TELEMETRY_SESSION_OBJECT}.correlation_id',
_mock_uuid())
mock_in_unit_test(unit_test,
f'{TELEMETRY_MODULE}._get_azure_subscription_id',
_mock_uuid)
def patch_long_run_operation_delay(unit_test):
def _shortcut_long_run_operation(*args, **kwargs): # pylint: disable=unused-argument
return
mock_in_unit_test(unit_test,
'msrestazure.azure_operation.AzureOperationPoller._delay',
_shortcut_long_run_operation)
mock_in_unit_test(unit_test,
'azure.cli.core.commands.LongRunningOperation._delay',
_shortcut_long_run_operation)
def patch_long_run_operation_delay(unit_test):
def _shortcut_long_run_operation(*args, **kwargs): # pylint: disable=unused-argument
return
mock_in_unit_test(
unit_test, 'msrestazure.azure_operation.AzureOperationPoller._delay',
_shortcut_long_run_operation)
mock_in_unit_test(unit_test,
'azure.cli.core.commands.LongRunningOperation._delay',
_shortcut_long_run_operation)
def patch_main_exception_handler(unit_test):
from vcr.errors import CannotOverwriteExistingCassetteException
def _handle_main_exception(ex):
if isinstance(ex, CannotOverwriteExistingCassetteException):
# This exception usually caused by a no match HTTP request. This is a product error
# that is caused by change of SDK invocation.
raise ex
raise CliExecutionError(ex)
mock_in_unit_test(unit_test, 'azure.cli.core.util.handle_exception', _handle_main_exception)
def patch_retrieve_token_for_user(unit_test):
def _retrieve_token_for_user(*args, **kwargs): # pylint: disable=unused-argument
import datetime
fake_token = 'top-secret-token-for-you'
return 'Bearer', fake_token, {
"tokenType": "Bearer",
"expiresIn": 3600,
"expiresOn": (datetime.datetime.now() + datetime.timedelta(hours=1)).strftime("%Y-%m-%d %H:%M:%S.%f"),
"resource": args[3],
"accessToken": fake_token,
"refreshToken": fake_token
}
mock_in_unit_test(unit_test,
'azure.cli.core._profile.CredsCache.retrieve_token_for_user',
_retrieve_token_for_user)
def patch_load_cached_subscriptions(unit_test):
def _handle_load_cached_subscription(*args, **kwargs): # pylint: disable=unused-argument
return [{
"id": MOCKED_SUBSCRIPTION_ID,
"user": {
"name": MOCKED_USER_NAME,
"type": "user"
},
"state": "Enabled",
"name": "Example",
"tenantId": MOCKED_TENANT_ID,
"isDefault": True}]
mock_in_unit_test(unit_test,
'azure.cli.core._profile.Profile.load_cached_subscriptions',
_handle_load_cached_subscription)
def patch_progress_controller(unit_test):
def _mock_pass(*args, **kwargs): # pylint: disable=unused-argument
pass
mock_in_unit_test(
unit_test, 'azure.cli.core.commands.progress.ProgressHook.update', _mock_pass)
mock_in_unit_test(
unit_test, 'azure.cli.core.commands.progress.ProgressHook.add', _mock_pass)
mock_in_unit_test(
unit_test, 'azure.cli.core.commands.progress.ProgressHook.end', _mock_pass)
mock_in_unit_test(
unit_test, 'azure.cli.command_modules.storage.custom._update_progress', _mock_pass)
def patch_progress_controller(unit_test):
def _mock_pass(*args, **kwargs): # pylint: disable=unused-argument
pass
mock_in_unit_test(unit_test,
'azure.cli.core.commands.progress.ProgressHook.update',
_mock_pass)
mock_in_unit_test(unit_test,
'azure.cli.core.commands.progress.ProgressHook.add',
_mock_pass)
mock_in_unit_test(unit_test,
'azure.cli.core.commands.progress.ProgressHook.end',
_mock_pass)
mock_in_unit_test(
unit_test, 'azure.cli.command_modules.storage.custom._update_progress',
_mock_pass)
mock_in_unit_test(
unit_test,
'azure.cli.core.commands.LongRunningOperation._template_progress',
_mock_pass)
def patch_progress_controller(unit_test):
def _mock_pass(*args, **kwargs): # pylint: disable=unused-argument
pass
mock_in_unit_test(
unit_test, 'azure.cli.core.commands.progress.ProgressHook.update', _mock_pass)
mock_in_unit_test(
unit_test, 'azure.cli.core.commands.progress.ProgressHook.add', _mock_pass)
mock_in_unit_test(
unit_test, 'azure.cli.core.commands.progress.ProgressHook.end', _mock_pass)
def patch_progress_controller(unit_test):
def _mock_pass(*args, **kwargs): # pylint: disable=unused-argument
pass
def _mock_get_hook(_):
return _mock_pass
mock_in_unit_test(unit_test,
'azure.cli.core.commands.progress.ProgressHook.update',
_mock_pass)
mock_in_unit_test(unit_test,
'azure.cli.core.commands.progress.ProgressHook.add',
_mock_pass)
mock_in_unit_test(unit_test,
'azure.cli.core.commands.progress.ProgressHook.end',
_mock_pass)
mock_in_unit_test(
unit_test,
'azure.cli.command_modules.storage.operations.blob.get_update_progress_fn',
_mock_get_hook)
def patch_version(unit_test):
mock_in_unit_test(unit_test, 'azure.cli.core.__version__', MOCK_VERSION)
def patch_uuid_str(unit_test):
def _mock_get_uuid_str(*args, **kwargs): # pylint: disable=unused-argument
return MOCK_JOB_ID
mock_in_unit_test(unit_test, 'azure.cli.command_modules.dla.custom._get_uuid_str', _mock_get_uuid_str)
def patch_telemetry(unit_test):
mock_in_unit_test(unit_test,
'azure.cli.core.telemetry.is_telemetry_enabled',
lambda: True)
