From 912f9e7adcbcc23881fcc7f0f86cf27d279f33ac Mon Sep 17 00:00:00 2001
From: Matias Quaranta <ealsur@hotmail.com>
Date: Thu, 6 Apr 2023 12:22:46 -0700
Subject: [PATCH 1/2] Passing properties

---
 Microsoft.Azure.Cosmos/src/GatewayStoreClient.cs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/Microsoft.Azure.Cosmos/src/GatewayStoreClient.cs b/Microsoft.Azure.Cosmos/src/GatewayStoreClient.cs
index c779dc7855..e151ffcd94 100644
--- a/Microsoft.Azure.Cosmos/src/GatewayStoreClient.cs
+++ b/Microsoft.Azure.Cosmos/src/GatewayStoreClient.cs
@@ -320,6 +320,14 @@ private async ValueTask<HttpRequestMessage> PrepareRequestMessageAsync(
                 }
             }
 
+            if (request.Properties != null)
+            {
+                foreach (KeyValuePair<string, object> property in request.Properties)
+                {
+                    requestMessage.Properties.Add(property);
+                }
+            }
+
             // add activityId
             Guid activityId = System.Diagnostics.Trace.CorrelationManager.ActivityId;
             Debug.Assert(activityId != Guid.Empty);

From 6ce7cbd9c9ce3660b49d20548ae2240a7fd59a6d Mon Sep 17 00:00:00 2001
From: Matias Quaranta <ealsur@hotmail.com>
Date: Thu, 6 Apr 2023 12:42:00 -0700
Subject: [PATCH 2/2] test

---
 .../GatewayStoreModelTest.cs                  | 61 +++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/GatewayStoreModelTest.cs b/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/GatewayStoreModelTest.cs
index 43078d1866..36b350a751 100644
--- a/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/GatewayStoreModelTest.cs
+++ b/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/GatewayStoreModelTest.cs
@@ -161,6 +161,67 @@ public async Task TestRetries()
 
         }
 
+        /// <summary>
+        /// Verifies that if the DCE has Properties set, the HttpRequestMessage has them too. Used on ThinClient.
+        /// </summary>
+        [TestMethod]
+        public async Task PassesPropertiesFromDocumentServiceRequest()
+        {
+            IDictionary<string, object> properties = new Dictionary<string, object>()
+            {
+                {"property1", Guid.NewGuid() },
+                {"property2", Guid.NewGuid().ToString() }
+            };
+
+            Func<HttpRequestMessage, Task<HttpResponseMessage>> sendFunc = request =>
+            {
+                Assert.AreEqual(properties.Count, request.Properties.Count);
+                foreach (KeyValuePair<string, object> item in properties)
+                {
+                    Assert.AreEqual(item.Value, request.Properties[item.Key]);
+                }
+
+                return Task.FromResult(new HttpResponseMessage(HttpStatusCode.OK) );
+            };
+
+            Mock<IDocumentClientInternal> mockDocumentClient = new Mock<IDocumentClientInternal>();
+            mockDocumentClient.Setup(client => client.ServiceEndpoint).Returns(new Uri("https://foo"));
+
+            using GlobalEndpointManager endpointManager = new GlobalEndpointManager(mockDocumentClient.Object, new ConnectionPolicy());
+            ISessionContainer sessionContainer = new SessionContainer(string.Empty);
+            DocumentClientEventSource eventSource = DocumentClientEventSource.Instance;
+            HttpMessageHandler messageHandler = new MockMessageHandler(sendFunc);
+            using GatewayStoreModel storeModel = new GatewayStoreModel(
+                endpointManager,
+                sessionContainer,
+                ConsistencyLevel.Eventual,
+                eventSource,
+                null,
+                MockCosmosUtil.CreateCosmosHttpClient(() => new HttpClient(messageHandler)));
+
+            using (new ActivityScope(Guid.NewGuid()))
+            {
+                using (DocumentServiceRequest request =
+                DocumentServiceRequest.Create(
+                    Documents.OperationType.Query,
+                    Documents.ResourceType.Document,
+                    new Uri("https://foo.com/dbs/db1/colls/coll1", UriKind.Absolute),
+                    new MemoryStream(Encoding.UTF8.GetBytes("content1")),
+                    AuthorizationTokenType.PrimaryMasterKey,
+                    null))
+                {
+                    // Add properties to the DCE
+                    request.Properties = new Dictionary<string, object>();
+                    foreach (KeyValuePair<string, object> property in properties)
+                    {
+                        request.Properties.Add(property.Key, property.Value);
+                    }
+
+                    await storeModel.ProcessMessageAsync(request);
+                }
+            }
+        }
+
         [TestMethod]
         public async Task TestApplySessionForMasterOperation()
         {