Properly refresh continuation token for enumerating all entities

Fixes #53

Author: kzu

src/TableStorage/DocumentRepository`1.cs

@@ -138,6 +138,8 @@ async IAsyncEnumerable<T> EnumerateBinaryAsync(Expression<Func<IDocumentEntity,
                 var segment = await table.ExecuteQuerySegmentedAsync(query, continuation, cancellation)
                     .ConfigureAwait(false);
 
+                continuation = segment.ContinuationToken;
+
                 foreach (var entity in segment)
                 {
                     if (entity != null && entity.Document != null)
@@ -214,6 +216,8 @@ async IAsyncEnumerable<T> EnumerateStringAsync(Expression<Func<IDocumentEntity,
                 var segment = await table.ExecuteQuerySegmentedAsync(query, continuation, cancellation)
                     .ConfigureAwait(false);
 
+                continuation = segment.ContinuationToken;
+
                 foreach (var entity in segment)
                 {
                     if (entity != null && entity.Document != null)

src/TableStorage/TableEntityRepository.cs

@@ -77,6 +77,8 @@ public async IAsyncEnumerable<ITableEntity> EnumerateAsync(string? partitionKey
                 var segment = await table.ExecuteQuerySegmentedAsync(query, continuation, cancellation)
                     .ConfigureAwait(false);
 
+                continuation = segment.ContinuationToken;
+
                 foreach (var entity in segment)
                     if (entity != null)
                         yield return entity;

src/TableStorage/TableRepository`1.cs

@@ -109,7 +109,9 @@ public async IAsyncEnumerable<T> EnumerateAsync(string? partitionKey = default,
             {
                 var segment = await table.ExecuteQuerySegmentedAsync(query, continuation, cancellation)
                     .ConfigureAwait(false);
-                
+
+                continuation = segment.ContinuationToken;
+
                 foreach (var entity in segment)
                     if (entity != null)
                         yield return ToEntity(entity);

src/Tests/RepositoryTests.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.Azure.Cosmos.Table;
 using Xunit;