fixup! add CancellationToken support

Particular · Feb 18, 2021 · 319cefa · 319cefa
1 parent 067418b
commit 319cefa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/NServiceBus.TransportTests/NServiceBusTransportTest.cs b/src/NServiceBus.TransportTests/NServiceBusTransportTest.cs
@@ -83,15 +83,15 @@ protected async Task StartPump(Func<MessageContext, Task> onMessage, Func<ErrorC
                 InputQueueName,
                 string.Empty,
                 new StartupDiagnosticEntries(),
-                (message, ex) =>
+                (message, ex, cancellationToken) =>
                 {
                     if (onCriticalError == null)
                     {
                         testCancellationTokenSource.Cancel();
                         Assert.Fail($"{message}{Environment.NewLine}{ex}");
                     }
 
-                    onCriticalError(message, ex);
+                    onCriticalError(message, ex, cancellationToken);
                 },
                 true);