Added IAsyncDisposable to TestServiceProvider (#292)

Author: egil

CHANGELOG.md

@@ -98,6 +98,8 @@ List of new features.
 
   By [@egil](https://github.com/egil) in [#288](https://github.com/egil/bUnit/pull/288).
 
+- Added support for registering services in bUnits `Services` collection that implements `IAsyncDisposable`. Suggested by [@jmaillet](https://github.com/jmaillet) in [#249](https://github.com/egil/bUnit/issues/249). 
+
 ### Changed
 List of changes in existing functionality.
 

src/bunit.core/TestServiceProvider.cs

@@ -13,7 +13,7 @@ public sealed class TestServiceProvider : IServiceProvider, IServiceCollection,
 	{
 		private readonly IServiceCollection _serviceCollection;
 		private ServiceProvider? _serviceProvider;
-		
+
 		/// <summary>
 		/// Gets whether this <see cref="TestServiceProvider"/> has been initialized, and 
 		/// no longer will accept calls to the <c>AddService</c>'s methods.
@@ -78,7 +78,14 @@ public object GetService(Type serviceType)
 		/// <inheritdoc/>
 		public void Dispose()
 		{
-			_serviceProvider?.Dispose();
+			if (_serviceProvider is null) return;
+
+			var disposedTask = _serviceProvider.DisposeAsync().AsTask();
+
+			if (!disposedTask.IsCompleted)
+				disposedTask.GetAwaiter().GetResult();
+
+			_serviceProvider.Dispose();
 		}
 
 		/// <inheritdoc/>

tests/bunit.core.tests/TestServiceProviderTest.cs

@@ -7,7 +7,7 @@
 
 namespace Bunit
 {
-	public class TestServiceProviderTest
+	public partial class TestServiceProviderTest
 	{
 		class DummyService { }
 		class AnotherDummyService { }

tests/bunit.core.tests/TestServiceProviderTest.net5.cs

@@ -0,0 +1,28 @@
+#if NET5_0
+using System;
+using System.Threading.Tasks;
+using Microsoft.Extensions.DependencyInjection;
+using Shouldly;
+using Xunit;
+
+namespace Bunit
+{
+	public partial class TestServiceProviderTest
+	{
+		[Fact(DisplayName = "Can correctly dispose of async disposable service")]
+		public void Net5Test001()
+		{
+			var sut = new TestServiceProvider();
+			sut.AddScoped<AsyncDisposableService>();
+			sut.GetService<AsyncDisposableService>();
+
+			Should.NotThrow(() => sut.Dispose());
+		}
+
+		class AsyncDisposableService : IAsyncDisposable
+		{
+			public ValueTask DisposeAsync() => ValueTask.CompletedTask;
+		}
+	}
+}
+#endif