Fixes tests

ramonsmits · ramonsmits · commit d5d6c3ae1719 · 2025-03-11T10:33:26.000+01:00
diff --git a/src/ServiceControl/Operations/ErrorIngestion.cs b/src/ServiceControl/Operations/ErrorIngestion.cs
@@ -269,6 +269,7 @@ async Task OnMessage(MessageContext messageContext, CancellationToken cancellati
             }
 
             var taskCompletionSource = new TaskCompletionSource<bool>(TaskCreationOptions.RunContinuationsAsynchronously);
+            messageContext.SetTaskCompletionSource(taskCompletionSource);
 
             // Ideally we want to propagate the cancellationToken to the batch handling
             // but cancellation in only cancelled when endpointInstance.Stop is cancelled, not when invoked.

Original file line number	Diff line number	Diff line change
`@@ -269,6 +269,7 @@ async Task OnMessage(MessageContext messageContext, CancellationToken cancellati`
`269`	`269`	`}`
`270`	`270`
`271`	`271`	`var taskCompletionSource = new TaskCompletionSource<bool>(TaskCreationOptions.RunContinuationsAsynchronously);`
	`272`	`+ messageContext.SetTaskCompletionSource(taskCompletionSource);`
`272`	`273`
`273`	`274`	`// Ideally we want to propagate the cancellationToken to the batch handling`
`274`	`275`	`// but cancellation in only cancelled when endpointInstance.Stop is cancelled, not when invoked.`