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)