fix: addressing comments

Author: dibahlfi

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

@@ -305,7 +305,7 @@ public virtual async Task<ResponseMessage> SendAsync(
                             // For epk range filtering we can end up in one of 3 cases:
                             if (overlappingRanges.Count > 1)
                             {
-                                //If we are running a query and our provided partition key results in a hash that resolves to more than one EPKRanges then its a valid use case
+                                //If we are running a query plan and our provided partition key results in a hash that resolves to more than one EPKRanges then its a valid use case
                                 bool isQueryOperation = request.ResourceType == ResourceType.Document && request.OperationType == OperationType.QueryPlan;
                                 if (!isQueryOperation)
                                 {
@@ -321,9 +321,6 @@ public virtual async Task<ResponseMessage> SendAsync(
 
                                     return goneException.ToCosmosResponseMessage(request);
                                 }
-                                // For query operations spanning multiple partitions, we don't set PartitionKeyRangeId
-                                // because the query engine needs to handle this case differently
-                                // Just continue without setting PartitionKeyRangeId
                             }
                             // overlappingRanges.Count == 1
                             else

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

@@ -51,13 +51,13 @@ public async Task ValidatePassingOverlappingRangesInQueryPlanDoesntThrowAnExcept
             // Mock PartitionKeyRangeCache
             List<PartitionKeyRange> overlappingRanges = new List<PartitionKeyRange>
             {
-                    new PartitionKeyRange { Id = "0", MinInclusive = "A", MaxExclusive = "Z" },
-                    new PartitionKeyRange { Id = "1", MinInclusive = "M", MaxExclusive = "Z" }
+                new PartitionKeyRange { Id = "0", MinInclusive = "0D4DC2CD8F49C65A8E0C5306B61B4343", MaxExclusive = "0DCEB8CE51C6BFE84F4BD9409F69B9BB2164DEBD78C50C850E0C1E3E3F0579ED" },
+                new PartitionKeyRange { Id = "1", MinInclusive = "0DCEB8CE51C6BFE84F4BD9409F69B9BB2164DEBD78C50C850E0C1E3E3F0579ED", MaxExclusive = "1080F600C27CF98DC13F8639E94E7676" }
             };
             PartitionKeyRangeCache pkRangeCache = new TestPartitionKeyRangeCache(overlappingRanges);
 
             // FeedRangeEpk for the test
-            FeedRangeEpk feedRange = new FeedRangeEpk(new Documents.Routing.Range<string>("A", "Z", true, false));
+            FeedRangeEpk feedRange = new FeedRangeEpk(new Documents.Routing.Range<string>("0DCEB8CE51C6BFE84F4BD9409F69B9BB", "0DCEB8CE51C6BFE84F4BD9409F69B9BBFF", true, false));
             RequestInvokerHandler invoker = new RequestInvokerHandler(client, null, null, null)
             {
                 InnerHandler = new TestHandler((request, token) => TestHandler.ReturnSuccess())