Use xunit async init

cyanfish · Apr 1, 2024 · d1be1c7 · d1be1c7
1 parent 1991b7f
commit d1be1c7
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/NAPS2.Sdk.Tests/Remoting/ScanServerTestsBase.cs b/NAPS2.Sdk.Tests/Remoting/ScanServerTestsBase.cs
@@ -8,11 +8,12 @@
 using NAPS2.Scan.Internal;
 using NAPS2.Sdk.Tests.Mocks;
 using NSubstitute;
+using Xunit;
 using Xunit.Abstractions;
 
 namespace NAPS2.Sdk.Tests.Remoting;
 
-public class ScanServerTestsBase : ContextualTests
+public class ScanServerTestsBase : ContextualTests, IAsyncLifetime
 {
     protected const int TIMEOUT = 60_000;
 
@@ -40,19 +41,16 @@ public ScanServerTestsBase(ITestOutputHelper testOutputHelper,
         ScanningContext.Logger.LogDebug("Display name: {Name}", displayName);
         var serverDevice = new ScanDevice(ScanOptionsValidator.SystemDefaultDriver, "testID", "testName");
         _server.RegisterDevice(serverDevice, displayName);
-        _server.Start().Wait();
 
         // Set up a client ScanController for scanning through EsclScanDriver -> network -> ScanServer
         _client = new ScanController(ScanningContext);
         var uuid = new ScanServerDevice { Device = serverDevice, Name = displayName }.GetUuid(_server.InstanceId);
         _clientDevice = new ScanDevice(Driver.Escl, uuid, displayName);
     }
 
-    public override void Dispose()
-    {
-        _server.Stop().Wait();
-        base.Dispose();
-    }
+    public Task InitializeAsync() => _server.Start();
+
+    public Task DisposeAsync() => _server.Stop();
 
     protected async Task<bool> TryFindClientDevice()
     {