Skip to content

Query: Fixes malformed continuation token exception type and message #3917

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
microsoft-github-policy-service merged 12 commits into from
Jul 10, 2023
Merged

Query: Fixes malformed continuation token exception type and message #3917

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
9308e69
Fixed malformed continuation token issue where Exception was not Cosm…
akotalwar Jun 15, 2023
97c928b
Fixed incorrect indentation
akotalwar Jun 15, 2023
4574231
Added type check for incoming exception
akotalwar Jun 16, 2023
f7d91e9
Replaced if/else with extra catch block
akotalwar Jun 16, 2023
17089de
Merge branch 'master' into users/akotalwar/BugFixForNonSpecificContTo…
akotalwar Jun 17, 2023
176d9da
Moved fix to a higher point in the call stack
akotalwar Jun 26, 2023
3c7522c
Merge branch 'users/akotalwar/BugFixForNonSpecificContTokenException'…
akotalwar Jun 26, 2023
afa7717
Merge branch 'master' into users/akotalwar/BugFixForNonSpecificContTo…
akotalwar Jun 26, 2023
a96930d
Removed unused Usings
akotalwar Jun 26, 2023
c112a12
Merge branch 'users/akotalwar/BugFixForNonSpecificContTokenException'…
akotalwar Jun 26, 2023
a75fcbd
Updated test code
akotalwar Jul 7, 2023
6d0f5ca
Merge branch 'master' into users/akotalwar/BugFixForNonSpecificContTo…
ealsur Jul 10, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 23 additions & 2 deletions Microsoft.Azure.Cosmos/src/ReadFeed/ReadFeedIteratorCore.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,13 +9,16 @@ namespace Microsoft.Azure.Cosmos.ReadFeed
using System.Threading;
using System.Threading.Tasks;
using Microsoft.Azure.Cosmos.CosmosElements;
using Microsoft.Azure.Cosmos.Diagnostics;
using Microsoft.Azure.Cosmos.Json;
using Microsoft.Azure.Cosmos.Pagination;
using Microsoft.Azure.Cosmos.Query.Core;
using Microsoft.Azure.Cosmos.Query.Core.Exceptions;
using Microsoft.Azure.Cosmos.Query.Core.Monads;
using Microsoft.Azure.Cosmos.ReadFeed.Pagination;
using Microsoft.Azure.Cosmos.Resource.CosmosExceptions;
using Microsoft.Azure.Cosmos.Routing;
using Microsoft.Azure.Cosmos.Tracing;
using Microsoft.Azure.Documents;

/// <summary>
/// Cosmos feed stream iterator. This is used to get the query responses with a Stream content
Expand Down Expand Up @@ -113,7 +116,25 @@ public ReadFeedIteratorCore(
else
{
CosmosString tokenAsString = (CosmosString)token;
state = ReadFeedState.Continuation(CosmosElement.Parse(tokenAsString.Value));
try
{
state = ReadFeedState.Continuation(CosmosElement.Parse(tokenAsString.Value));
}
catch (Exception exception) when (exception.InnerException is JsonParseException)
{
MalformedContinuationTokenException malformedContinuationTokenException = new MalformedContinuationTokenException(exception.Message);
throw CosmosExceptionFactory.CreateBadRequestException(
message: $"Malformed Continuation Token: {tokenAsString}.",
headers: CosmosQueryResponseMessageHeaders.ConvertToQueryHeaders(
new Headers(),
default,
default,
(int)SubStatusCodes.MalformedContinuationToken,
default),
stackTrace: exception.StackTrace,
innerException: malformedContinuationTokenException,
trace: null);
}
}

FeedRangeState<ReadFeedState> feedRangeState = new FeedRangeState<ReadFeedState>(feedRange, state);
Expand Down
26 changes: 26 additions & 0 deletions Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/CosmosItemLinqTests.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -211,6 +211,32 @@ public async Task ItemLINQQueryTest()
Assert.AreEqual(itemList[1].id, queriable.ToList()[0].id);
}

[TestMethod]
public void ItemLINQQueryWithInvalidContinuationTokenTest()
{
string malformedString = "Malformed String";
FeedIterator<ToDoActivity> feedIterator = this.Container.GetItemLinqQueryable<ToDoActivity>().ToFeedIterator();

while (feedIterator.HasMoreResults)
{
IOrderedQueryable<ToDoActivity> querable = this.Container.GetItemLinqQueryable<ToDoActivity>(
continuationToken: malformedString);
try
{
FeedIterator<ToDoActivity> iterator = querable.ToFeedIterator();
}
catch (CosmosException exception)
{
Assert.IsTrue(exception.StatusCode == System.Net.HttpStatusCode.BadRequest);
Assert.IsTrue(exception.SubStatusCode == (int)Documents.SubStatusCodes.MalformedContinuationToken);
Assert.IsTrue(exception.Message.Contains(malformedString));
return;
}

Assert.Fail("Should never reach till here, hence ensuring that an exception is always recieved");
}
}

[TestMethod]
public async Task ItemLINQQueryWithContinuationTokenTest()
{
Expand Down