test: Improve EnsureInitializedAsync_ShouldThrowException_WhenProvide…

…rDoesNotExist test
open-feature · Oct 14, 2024 · 572d67b · 572d67b
1 parent 84dd0e0
commit 572d67b
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/test/OpenFeature.DependencyInjection.Tests/FeatureLifecycleManagerTests.cs b/test/OpenFeature.DependencyInjection.Tests/FeatureLifecycleManagerTests.cs
@@ -16,9 +16,6 @@ public FeatureLifecycleManagerTests()
         Api.Instance.SetContext(null);
         Api.Instance.ClearHooks();
 
-        //_mockApi = Substitute.ForPartsOf<Api>();
-        //Api.Instance.Returns(_mockApi);
-
         _mockServiceProvider = Substitute.For<IServiceProvider>();
 
         _systemUnderTest = new FeatureLifecycleManager(
@@ -48,10 +45,11 @@ public async Task EnsureInitializedAsync_ShouldThrowException_WhenProviderDoesNo
         // Arrange
         _mockServiceProvider.GetService(typeof(FeatureProvider)).Returns(null as FeatureProvider);
 
-        // Act & Assert
-        var exception = await Assert.ThrowsAsync<InvalidOperationException>(async () =>
-            await _systemUnderTest.EnsureInitializedAsync().ConfigureAwait(false));
+        // Act
+        var act = () => _systemUnderTest.EnsureInitializedAsync().AsTask();
 
+        // Assert
+        var exception = await Assert.ThrowsAsync<InvalidOperationException>(act);
         exception.Message.Should().Be("Feature provider is not registered in the service collection.");
     }
 }