def test_http_response_error(self):
status_codes = [x for x in range(400, 405)] + [500, 1000, None]
cli_errors = [
BadRequestError,
UnauthorizedError,
UnclassifiedUserFault,
ForbiddenError,
ResourceNotFoundError,
AzureInternalError,
ServiceError,
ServiceError,
]
status_code_cli_error_pairs = list(zip(status_codes, cli_errors))
azure_error = HttpResponseError()
for idx, status_code_cli_error_pair in enumerate(
status_code_cli_error_pairs, 1):
# get mapped error
status_code = status_code_cli_error_pair[0]
azure_error.status_code = status_code
azure_error.message = f"error_msg_{idx}"
mapped_error = helpers.map_azure_error_to_cli_error(azure_error)
# get mock error
cli_error = status_code_cli_error_pair[1]
mock_error = cli_error(f"error_msg_{idx}")
self.check_error_equality(mapped_error, mock_error)
def get_cluster_snapshot(cli_ctx, subscription_id, resource_group_name, snapshot_name):
snapshot_client = get_mc_snapshots_client(cli_ctx, subscription_id)
try:
snapshot = snapshot_client.get(resource_group_name, snapshot_name)
# track 2 sdk raise exception from azure.core.exceptions
except AzureError as ex:
if "not found" in ex.message:
raise ResourceNotFoundError("Managed cluster snapshot '{}' not found.".format(snapshot_name))
raise map_azure_error_to_cli_error(ex)
return snapshot
def get_nodepool_snapshot(cli_ctx, resource_group_name, snapshot_name):
snapshot_client = cf_nodepool_snapshots(cli_ctx)
try:
snapshot = snapshot_client.get(resource_group_name, snapshot_name)
# track 2 sdk raise exception from azure.core.exceptions
except AzureError as ex:
if "not found" in ex.message:
raise ResourceNotFoundError(
"Snapshot '{}' not found.".format(snapshot_name))
raise map_azure_error_to_cli_error(ex)
return snapshot
def test_azure_error(self):
azure_error = AzureError("test_error_msg")
cli_error = helpers.map_azure_error_to_cli_error(azure_error)
mock_error = ServiceError("test_error_msg")
self.check_error_equality(cli_error, mock_error)
def test_service_request_error(self):
azure_error = ServiceRequestError("test_error_msg")
cli_error = helpers.map_azure_error_to_cli_error(azure_error)
mock_error = ClientRequestError("test_error_msg")
self.check_error_equality(cli_error, mock_error)
def test_service_response_error(self):
azure_error = ServiceResponseError("test_error_msg")
cli_error = map_azure_error_to_cli_error(azure_error)
mock_error = AzureResponseError("test_error_msg")
self.check_error_equality(cli_error, mock_error)
