[Event Hubs] Partition Context fix (Azure#38265)

* fix * test fix
azure-sdk · Aug 17, 2023 · d4245e5 · d4245e5
1 parent 5041d7c
commit d4245e5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/...t.Azure.WebJobs.Extensions.EventHubs/src/Listeners/EventHubListener.PartitionProcessor.cs b/...t.Azure.WebJobs.Extensions.EventHubs/src/Listeners/EventHubListener.PartitionProcessor.cs
@@ -282,6 +282,7 @@ private async Task MonitorCachedEvents(DateTimeOffset? lastCheckpointTime, Cance
                         var details = GetOperationDetails(_mostRecentPartitionContext, "MaxWaitTimeElapsed");
                         _logger.LogDebug($"Partition Processor has waited MaxWaitTime since last invocation and is attempting to invoke function on all held events ({details})");
 
+                        UpdateCheckpointContext(triggerEvents, _mostRecentPartitionContext);
                         await TriggerExecute(triggerEvents, _mostRecentPartitionContext, _functionExecutionToken).ConfigureAwait(false);
                         if (!backgroundCancellationTokenSource.Token.IsCancellationRequested)
                         {

diff --git a/sdk/eventhub/Microsoft.Azure.WebJobs.Extensions.EventHubs/tests/EventHubListenerTests.cs b/sdk/eventhub/Microsoft.Azure.WebJobs.Extensions.EventHubs/tests/EventHubListenerTests.cs
@@ -112,10 +112,10 @@ public async Task ProcessEvents_MultipleDispatch_CheckpointsCorrectly(int batchC
                 Times.Exactly(expected));
         }
 
-        [TestCase(1, 29)]
-        [TestCase(5, 5)]
-        [TestCase(10, 2)]
-        [TestCase(30, 0)]
+        [TestCase(1, 30)]
+        [TestCase(5, 6)]
+        [TestCase(10, 3)]
+        [TestCase(30, 1)]
         [TestCase(35, 0)]
         public async Task ProcessEvents_MultipleDispatch_MinBatch_CheckpointsCorrectly_NoCheckpoint(int batchCheckpointFrequency, int expected)
         {
@@ -221,10 +221,10 @@ public async Task ProcessEvents_MultipleDispatch_MinBatch_CheckpointsCorrectly_R
                 Times.AtLeastOnce);
         }
 
-        [TestCase(1, 29)]
-        [TestCase(5, 5)]
-        [TestCase(10, 2)]
-        [TestCase(30, 0)]
+        [TestCase(1, 30)]
+        [TestCase(5, 6)]
+        [TestCase(10, 3)]
+        [TestCase(30, 1)]
         [TestCase(35, 0)]
         public async Task ProcessEvents_MultipleDispatch_MinBatch_CheckpointsCorrectly_OldCheckpoint(int batchCheckpointFrequency, int expected)
         {