Throwing communicationException when it is connectionError (#6942)

Multiple errors can happen when link/session is closing. Throw ServiceBusCommunicationException instead of ServiceBusException.

Fixes #6097

sdk/servicebus/Microsoft.Azure.ServiceBus/src/Amqp/AmqpExceptionHelper.cs

@@ -206,6 +206,9 @@ public static Exception GetClientException(Exception exception, string reference
                 case OperationCanceledException operationCanceledException when operationCanceledException.InnerException is AmqpException amqpException:
                     return amqpException.Error.ToMessagingContractException(connectionError);
 
+                case OperationCanceledException _ when connectionError:
+                    return new ServiceBusCommunicationException(message, aggregateException);
+
                 case OperationCanceledException _:
                     return new ServiceBusException(true, message, aggregateException);
 
@@ -216,6 +219,11 @@ public static Exception GetClientException(Exception exception, string reference
                     return new ServiceBusCommunicationException(message, aggregateException);
             }
 
+            if (connectionError)
+            {
+                return new ServiceBusCommunicationException(message, aggregateException);
+            }
+
             return aggregateException;
         }
 

sdk/servicebus/Microsoft.Azure.ServiceBus/src/Core/MessageReceiver.cs

@@ -1065,7 +1065,7 @@ protected virtual async Task<IList<Message>> OnReceiveAsync(int maxMessageCount,
             }
             catch (Exception exception)
             {
-                throw AmqpExceptionHelper.GetClientException(exception, receiveLink?.GetTrackingId(), null, receiveLink?.Session.IsClosing() ?? false);
+                throw AmqpExceptionHelper.GetClientException(exception, receiveLink?.GetTrackingId(), null, receiveLink?.IsClosing() ?? false);
             }
         }