From bf2989fc1e2a784286a495ee512b33cdb903bec7 Mon Sep 17 00:00:00 2001 From: Jordan Brown Date: Mon, 13 Jul 2020 15:04:33 -0400 Subject: [PATCH] Fix JobHandler tests --- .../Jobs/TestJobHandler.cs | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/tests/Tgstation.Server.Host.Tests/Jobs/TestJobHandler.cs b/tests/Tgstation.Server.Host.Tests/Jobs/TestJobHandler.cs index 3326de98fd..a4143ca8cc 100644 --- a/tests/Tgstation.Server.Host.Tests/Jobs/TestJobHandler.cs +++ b/tests/Tgstation.Server.Host.Tests/Jobs/TestJobHandler.cs @@ -38,12 +38,12 @@ namespace Tgstation.Server.Host.Jobs.Tests var tcs = new TaskCompletionSource(); currentWaitTask = tcs.Task; cts.Cancel(); - using(var handler = new JobHandler(TestJob)) - { - await Assert.ThrowsExceptionAsync(() => handler.Wait(cts.Token)).ConfigureAwait(false); - tcs.SetResult(null); - await handler.Wait(default).ConfigureAwait(false); - } + using var handler = new JobHandler(TestJob); + await Assert.ThrowsExceptionAsync(() => handler.Wait(cts.Token)).ConfigureAwait(false); + handler.Start(); + await Assert.ThrowsExceptionAsync(() => handler.Wait(cts.Token)).ConfigureAwait(false); + tcs.SetResult(null); + await handler.Wait(default).ConfigureAwait(false); } Assert.IsFalse(cancelled); } @@ -70,6 +70,7 @@ namespace Tgstation.Server.Host.Jobs.Tests cancelled = false; using(var handler = new JobHandler(TestJob)) { + handler.Start(); handler.Cancel(); tcs.SetResult(null); await handler.Wait(default).ConfigureAwait(false);