ChangeFeed: Adds MalformedContinuationToken SubstatusCode to exception (

#4583)
Azure · Jul 15, 2024 · 6e1d40d · 6e1d40d
1 parent 3fd2ce6
commit 6e1d40d
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/Microsoft.Azure.Cosmos/src/Query/Core/ExceptionToCosmosException.cs b/Microsoft.Azure.Cosmos/src/Query/Core/ExceptionToCosmosException.cs
@@ -157,7 +157,10 @@ internal override CosmosException Visit(
             {
                 return CosmosExceptionFactory.CreateBadRequestException(
                     message: malformedChangeFeedContinuationTokenException.Message,
-                    headers: new Headers(),
+                    headers: new Headers()
+                    {
+                        SubStatusCode = Documents.SubStatusCodes.MalformedContinuationToken
+                    },
                     stackTrace: malformedChangeFeedContinuationTokenException.StackTrace,
                     innerException: malformedChangeFeedContinuationTokenException,
                     trace: trace);

diff --git a/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/CosmosItemTests.cs b/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/CosmosItemTests.cs
@@ -3319,6 +3319,23 @@ await CosmosItemTests.GivenItemAsyncWhenMissingMemberHandlingIsErrorThenExpectsC
                     id: itemIdThatWillNotExist,
                     partitionKey: new Cosmos.PartitionKey(partitionKeyValue),
                     cancellationToken: cancellationToken));
+        }
+
+        [TestMethod]
+        public async Task MalformedChangeFeedContinuationTokenSubStatusCodeTest()
+        {
+            FeedIterator badIterator = this.Container.GetChangeFeedStreamIterator(
+                    ChangeFeedStartFrom.ContinuationToken("AMalformedContinuationToken"),
+                    ChangeFeedMode.Incremental,
+                    new ChangeFeedRequestOptions()
+                    {
+                        PageSizeHint = 100
+                    });
+
+            ResponseMessage response = await badIterator.ReadNextAsync();
+
+            Assert.AreEqual(HttpStatusCode.BadRequest, response.StatusCode);
+            Assert.AreEqual(SubStatusCodes.MalformedContinuationToken, response.Headers.SubStatusCode);
         }
 
         private static async Task GivenItemStreamAsyncWhenMissingMemberHandlingIsErrorThenExpectsCosmosExceptionTestAsync(