use dispose async in testhelper

6 years ago · 5ee3faa856
1 changed files with 2 additions and 2 deletions
--- a/test/E2ETest/TestHelpers.cs
+++ b/test/E2ETest/TestHelpers.cs
@ -104,7 +104,7 @@ namespace E2ETest
            using var cancellation = new CancellationTokenSource(WaitForServicesTimeout);
            try
            {
-                using (cancellation.Token.Register(() => startedTask.TrySetCanceled()))
+                await using (cancellation.Token.Register(() => startedTask.TrySetCanceled()))
                {
                    await startedTask.Task;
                }
@ -160,7 +160,7 @@ namespace E2ETest
            using var cancellation = new CancellationTokenSource(WaitForServicesTimeout);
            try
            {
-                using (cancellation.Token.Register(() => stoppedTask.TrySetCanceled()))
+                await using (cancellation.Token.Register(() => stoppedTask.TrySetCanceled()))
                {
                    await stoppedTask.Task;
                }