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(