refactor: Dispose in smoke tests

linkdotnet · Feb 23, 2024 · 3576cac · 3576cac
1 parent 6315641
commit 3576cac
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/tests/LinkDotNet.Blog.IntegrationTests/SmokeTests.cs b/tests/LinkDotNet.Blog.IntegrationTests/SmokeTests.cs
@@ -1,11 +1,12 @@
-using System.Threading.Tasks;
+using System;
+using System.Threading.Tasks;
 using LinkDotNet.Blog.Infrastructure.Persistence;
 using LinkDotNet.Blog.Web;
 using Microsoft.AspNetCore.Mvc.Testing;
 
 namespace LinkDotNet.Blog.IntegrationTests;
 
-public class SmokeTests : IClassFixture<WebApplicationFactory<Program>>
+public sealed class SmokeTests : IClassFixture<WebApplicationFactory<Program>>, IDisposable, IAsyncDisposable
 {
     private readonly WebApplicationFactory<Program> factory;
 
@@ -60,4 +61,14 @@ public async Task ShouldAllowDotsForFreeTextSearch()
 
         result.IsSuccessStatusCode.Should().BeTrue();
     }
+
+    public async ValueTask DisposeAsync()
+    {
+        if (factory != null)
+        {
+            await factory.DisposeAsync();
+        }
+    }
+
+    public void Dispose() => factory?.Dispose();
 }