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_ids(unit_test): def _mock_uuid(*args, **kwargs): # pylint: disable=unused-argument return MOCK_UUID mock_in_unit_test(unit_test, TELEMETRY_CORRELATION_ID_PATCH_TARGET, _mock_uuid()) mock_in_unit_test(unit_test, TELEMETRY_AZURE_SUBSCRIPTION_ID_PATCH_TARGET, _mock_uuid)
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_telemetry(unit_test): mock_in_unit_test(unit_test, 'azure.cli.core.telemetry.is_telemetry_enabled', lambda: True)
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, TELEMETRY_IS_ENABLED_PATCH_TARGET, lambda: True)
def patch_version(unit_test): mock_in_unit_test(unit_test, AZURE_CLI_CORE_VERSION_PATCH_TARGET, MOCK_CORE_CLI_VERSION)