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_telemetry(unit_test): mock_in_unit_test(unit_test, 'azure.cli.core.telemetry.is_telemetry_enabled', lambda: True)