Feat: Add assertions for service key presence in ServiceCollection

Author: AnnaSasDev

src/CodeOfChaos.Testing.TUnit/CodeOfChaos.Testing.TUnit.csproj

@@ -30,6 +30,7 @@
 
     <ItemGroup>
         <PackageReference Include="JetBrains.Annotations" Version="2024.3.0" />
+        <PackageReference Include="Microsoft.Extensions.DependencyInjection.Abstractions" Version="9.0.4" />
         <PackageReference Include="TUnit.Assertions" Version="0.19.86" />
         <PackageReference Include="TUnit.Core" Version="0.19.86" />
     </ItemGroup>

src/CodeOfChaos.Testing.TUnit/Conditions/Library/ContainsServiceKeyCondition.cs

@@ -0,0 +1,23 @@
+// ---------------------------------------------------------------------------------------------------------------------
+// Imports
+// ---------------------------------------------------------------------------------------------------------------------
+using Microsoft.Extensions.DependencyInjection;
+using TUnit.Assertions.AssertConditions;
+
+namespace CodeOfChaos.Testing.TUnit.Conditions.Library;
+// ---------------------------------------------------------------------------------------------------------------------
+// Code
+// ---------------------------------------------------------------------------------------------------------------------
+public class ContainsServiceKeyCondition<TServiceKey> : BaseAssertCondition<ServiceCollection> {
+    // -----------------------------------------------------------------------------------------------------------------
+    // Methods
+    // -----------------------------------------------------------------------------------------------------------------
+    protected override string GetExpectation() => $"to have a registered service \"{typeof(TServiceKey).Name}\"";
+    
+    protected override ValueTask<AssertionResult> GetResult(ServiceCollection? actualValue, Exception? exception, AssertionMetadata assertionMetadata) {
+        if (actualValue is null) return AssertionResult.Fail($"{nameof(ServiceCollection)} is null");
+        return actualValue.Any(d => d.ServiceType == typeof(TServiceKey)) 
+            ? AssertionResult.Passed 
+            : FailWithMessage($"No service with type {typeof(TServiceKey).Name} has been registered.");
+    }
+}

src/CodeOfChaos.Testing.TUnit/Conditions/Library/DoesNotContainServiceKeyCondition.cs

@@ -0,0 +1,23 @@
+// ---------------------------------------------------------------------------------------------------------------------
+// Imports
+// ---------------------------------------------------------------------------------------------------------------------
+using Microsoft.Extensions.DependencyInjection;
+using TUnit.Assertions.AssertConditions;
+
+namespace CodeOfChaos.Testing.TUnit.Conditions.Library;
+// ---------------------------------------------------------------------------------------------------------------------
+// Code
+// ---------------------------------------------------------------------------------------------------------------------
+public class DoesNotContainServiceKeyCondition<TServiceKey> : BaseAssertCondition<ServiceCollection> {
+    // -----------------------------------------------------------------------------------------------------------------
+    // Methods
+    // -----------------------------------------------------------------------------------------------------------------
+    protected override string GetExpectation() => $"to have a registered service \"{typeof(TServiceKey).Name}\"";
+    
+    protected override ValueTask<AssertionResult> GetResult(ServiceCollection? actualValue, Exception? exception, AssertionMetadata assertionMetadata) {
+        if (actualValue is null) return AssertionResult.Fail($"{nameof(ServiceCollection)} is null");
+        return actualValue.All(d => d.ServiceType != typeof(TServiceKey))
+            ? AssertionResult.Passed 
+            : FailWithMessage($"No service with type {typeof(TServiceKey).Name} has been registered.");
+    }
+}

src/CodeOfChaos.Testing.TUnit/Extensions/TUnitExtensionsServiceCollection.cs

@@ -0,0 +1,29 @@
+// ---------------------------------------------------------------------------------------------------------------------
+// Imports
+// ---------------------------------------------------------------------------------------------------------------------
+using CodeOfChaos.Testing.TUnit.Conditions.Library;
+using Microsoft.Extensions.DependencyInjection;
+using System.Runtime.CompilerServices;
+using TUnit.Assertions.AssertConditions.Interfaces;
+using TUnit.Assertions.AssertionBuilders;
+
+namespace CodeOfChaos.Testing.TUnit;
+// ---------------------------------------------------------------------------------------------------------------------
+// Code
+// ---------------------------------------------------------------------------------------------------------------------
+// ReSharper disable once InconsistentNaming
+public static class TUnitExtensionsServiceCollection {
+    public static InvokableValueAssertionBuilder<ServiceCollection> ContainsServiceKey<TServiceKey>(this IValueSource<ServiceCollection> valueSource, [CallerArgumentExpression(nameof(valueSource))] string doNotPopulateThisValue1 = "") {
+        return valueSource.RegisterAssertion(
+            new ContainsServiceKeyCondition<TServiceKey>(),
+            [doNotPopulateThisValue1]
+        );
+    }
+    
+    public static InvokableValueAssertionBuilder<ServiceCollection> DoesNotContainServiceKey<TServiceKey>(this IValueSource<ServiceCollection> valueSource, [CallerArgumentExpression(nameof(valueSource))] string doNotPopulateThisValue1 = "") {
+        return valueSource.RegisterAssertion(
+            new DoesNotContainServiceKeyCondition<TServiceKey>(),
+            [doNotPopulateThisValue1]
+        );
+    }
+}

tests/Tests.CodeOfChaos.Testing.TUnit/TUnitExtensionsServiceCollectionTests.cs

@@ -0,0 +1,61 @@
+// ---------------------------------------------------------------------------------------------------------------------
+// Imports
+// ---------------------------------------------------------------------------------------------------------------------
+using CodeOfChaos.Testing.TUnit;
+using Microsoft.Extensions.DependencyInjection;
+using TUnit.Assertions.Exceptions;
+
+namespace Tests.CodeOfChaos.Testing.TUnit;
+
+// ---------------------------------------------------------------------------------------------------------------------
+// Code
+// ---------------------------------------------------------------------------------------------------------------------
+public class TUnitExtensionsServiceCollectionTests {
+    [Test]
+    public async Task ContainsServiceKey_ShouldAssert() {
+        // Arrange
+        var services = new ServiceCollection();
+        
+        // Act
+        services.AddSingleton<string, string>();
+
+        // Assert
+        await Assert.That(services).ContainsServiceKey<string>();
+    }
+    
+    [Test]
+    public async Task ContainsServiceKey_ShouldThrowAssertion() {
+        // Arrange
+        var services = new ServiceCollection();
+        
+        // Act
+        services.AddSingleton<string, string>();
+        
+        // Assert
+        await Assert.ThrowsAsync<AssertionException>(async () => await Assert.That(services).ContainsServiceKey<int>());
+    }
+    
+    [Test]
+    public async Task DoesNotContainServiceKey_ShouldAssert() {
+        // Arrange
+        var services = new ServiceCollection();
+        
+        // Act
+        services.AddSingleton<string, string>();
+
+        // Assert
+        await Assert.That(services).DoesNotContainServiceKey<int>();
+    }
+    
+    [Test]
+    public async Task DoesNotContainServiceKey_ShouldThrowAssertion() {
+        // Arrange
+        var services = new ServiceCollection();
+        
+        // Act
+        services.AddSingleton<string, string>();
+        
+        // Assert
+        await Assert.ThrowsAsync<AssertionException>(async () => await Assert.That(services).DoesNotContainServiceKey<string>());
+    }
+}