[Internal] Gates: Fixes bug where CosmosOperaitonCancellationException is now thrown in CancellationTokenTests

Microsoft.Azure.Cosmos/src/Handler/AbstractRetryHandler.cs

@@ -53,6 +53,10 @@ public override async Task<ResponseMessage> SendAsync(
 
                 throw;
             }
+            catch (OperationCanceledException ex)
+            {
+                throw new CosmosOperationCanceledException(ex, request.Trace);
+            }
             finally
             {
                 request.OnBeforeSendRequestActions -= retryPolicyInstance.OnBeforeSendRequest;

Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/ClientConfigurationDiagnosticTest.cs

@@ -154,10 +154,9 @@ public async Task VerifyDiagnosticsOrderTest()
             }
             catch (CosmosOperationCanceledException oce)
             {
-                IReadOnlyList<ITrace> children = ((CosmosTraceDiagnostics)oce.Diagnostics).Value.Children;
-                ITrace exceptionChild = children[^1];
-                Assert.AreEqual("CosmosOperationCanceledException", exceptionChild.Name);
-                Assert.IsNotNull(exceptionChild.Data["Operation Cancelled Exception"]);
+                //check that the exception child exists in the trace diagnostics
+                Assert.IsTrue(oce.Diagnostics.ToString().Contains("CosmosOperationCanceledException"));
+                Assert.IsTrue(oce.Diagnostics.ToString().Contains("Operation Cancelled Exception"));
             }
         }
     }