CosmosOperationCanceledException: Adds serializable functionality

Microsoft.Azure.Cosmos/src/Resource/CosmosExceptions/CosmosOperationCanceledException.cs

@@ -6,18 +6,25 @@ namespace Microsoft.Azure.Cosmos
 {
     using System;
     using System.Collections;
+    using System.Diagnostics;
+    using System.Drawing.Printing;
+    using System.Runtime.ConstrainedExecution;
+    using System.Runtime.Serialization;
     using System.Threading;
     using global::Azure.Core.Pipeline;
+    using Microsoft.Azure.Cosmos.ChangeFeed.LeaseManagement;
     using Microsoft.Azure.Cosmos.Diagnostics;
     using Microsoft.Azure.Cosmos.Telemetry;
     using Microsoft.Azure.Cosmos.Telemetry.Diagnostics;
     using Microsoft.Azure.Cosmos.Tracing;
+    using Newtonsoft.Json;
 
     /// <summary>
     /// The exception that is thrown in a thread upon cancellation of an operation that
     ///  the thread was executing. This extends the OperationCanceledException to include the
     ///  diagnostics of the operation that was canceled.
-    /// </summary>
+    /// </summary> 
+    [Serializable]
     public class CosmosOperationCanceledException : OperationCanceledException
     {
         private readonly OperationCanceledException originalException;
@@ -63,6 +70,22 @@ internal CosmosOperationCanceledException(
             this.lazyMessage = this.CreateLazyMessage();
         }
 
+        /// <summary>
+        /// Initializes a new instance of the <see cref="CosmosOperationCanceledException"/> class.
+        /// </summary>
+        /// <param name="info">The SerializationInfo object that holds serialized object data for the exception being thrown.</param>
+        /// <param name="context">The StreamingContext that contains contextual information about the source or destination.</param>
+        protected CosmosOperationCanceledException(SerializationInfo info, StreamingContext context)
+            : base(info, context)
+        {
+            this.originalException = (OperationCanceledException)info.GetValue("originalException", typeof(OperationCanceledException));
+            this.tokenCancellationRequested = (bool)info.GetValue("tokenCancellationRequested", typeof(bool));
+            this.lazyMessage = new Lazy<string>(() => (string)info.GetValue("lazyMessage", typeof(string)));
+            this.toStringMessage = new Lazy<string>(() => (string)info.GetValue("toStringMessage", typeof(string)));
+            //Diagnostics cannot be serialized
+            this.Diagnostics = new CosmosTraceDiagnostics(NoOpTrace.Singleton);
+        }
+
         /// <inheritdoc/>
         public override string Source
         {
@@ -111,7 +134,7 @@ private Lazy<string> CreateToStringMessage()
         {
             return new Lazy<string>(() => $"{this.originalException}{Environment.NewLine}Cancellation Token has expired: {this.tokenCancellationRequested}. Learn more at: https://aka.ms/cosmosdb-tsg-request-timeout{Environment.NewLine}CosmosDiagnostics: {this.Diagnostics}");
         }
-
+         
         /// <summary>
         /// RecordOtelAttributes
         /// </summary>
@@ -123,5 +146,19 @@ internal static void RecordOtelAttributes(CosmosOperationCanceledException excep
             scope.AddAttribute(OpenTelemetryAttributeKeys.RequestDiagnostics, exception.Diagnostics);
             scope.AddAttribute(OpenTelemetryAttributeKeys.ExceptionMessage, exception.GetBaseException().Message);
         }
+
+        /// <summary>
+        /// Sets the System.Runtime.Serialization.SerializationInfo with information about the exception.
+        /// </summary>
+        /// <param name="info">The SerializationInfo object that holds serialized object data for the exception being thrown.</param>
+        /// <param name="context">The StreamingContext that contains contextual information about the source or destination.</param>
+        public override void GetObjectData(SerializationInfo info, StreamingContext context)
+        {
+            base.GetObjectData(info, context);
+            info.AddValue("originalException", this.originalException);
+            info.AddValue("tokenCancellationRequested", this.tokenCancellationRequested);
+            info.AddValue("lazyMessage", this.lazyMessage.Value);
+            info.AddValue("toStringMessage", this.toStringMessage.Value);
+        }
     }
 }

Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/CosmosOperationCanceledTests.cs

@@ -0,0 +1,45 @@
+//------------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+//------------------------------------------------------------
+
+namespace Microsoft.Azure.Cosmos.Tests
+{
+    using System;
+    using System.Collections.Generic;
+    using System.Linq;
+    using System.Runtime.Serialization.Formatters.Binary;
+    using System.Text;
+    using System.Threading.Tasks;
+    using Microsoft.Azure.Cosmos.Diagnostics;
+    using Microsoft.Azure.Cosmos.Tracing; //what is this? for the no op trace
+    using Microsoft.IdentityModel.Tokens;
+    using Microsoft.VisualStudio.TestTools.UnitTesting;
+    using Newtonsoft.Json;
+
+    [TestClass]
+    public class CosmosOperationCanceledTests
+    {
+
+        [TestMethod]
+        public void SerializationValidation()
+        {
+            //create test exception
+            CosmosOperationCanceledException originalException = new CosmosOperationCanceledException(
+                new OperationCanceledException("error message"),
+                new CosmosTraceDiagnostics(NoOpTrace.Singleton));
+
+            //serialize exception
+            string serialized = JsonConvert.SerializeObject(originalException);
+
+            CosmosOperationCanceledException deserializedExceptoin = 
+                JsonConvert.DeserializeObject<CosmosOperationCanceledException>(serialized);
+
+            //Asserts
+            Assert.AreEqual(originalException.ToString(), deserializedExceptoin.ToString());
+            Assert.AreEqual(originalException.Message, deserializedExceptoin.Message);
+            Assert.AreEqual(originalException.GetBaseException().Message, deserializedExceptoin.GetBaseException().Message);
+            Assert.AreEqual(originalException.GetBaseException().ToString(), deserializedExceptoin.GetBaseException().ToString());
+            Assert.AreEqual(originalException.GetBaseException().HResult, deserializedExceptoin.GetBaseException().HResult);
+        }
+    }
+}