Fix: IUnitOfWorkRepository

Author: AnnaSasDev

src/CodeOfChaos.Types.UnitOfWork.Contracts/IUnitOfWork.cs

@@ -20,5 +20,5 @@ public interface IUnitOfWork : IAsyncDisposable {
 
     ValueTask<TDbContext> GetDbContextAsync<TDbContext>(CancellationToken ct = default) where TDbContext : DbContext;
 
-    TRepo GetRepository<TRepo>() where TRepo : class, IToUnitOfWorkRepository;
+    TRepo GetRepository<TRepo>() where TRepo : class, IUnitOfWorkRepository;
 }

src/CodeOfChaos.Types.UnitOfWork.Contracts/ICanAttachToUnitOfWork.cs renamed to src/CodeOfChaos.Types.UnitOfWork.Contracts/IUnitOfWorkRepository.cs

@@ -5,4 +5,4 @@ namespace CodeOfChaos.Types.UnitOfWork;
 // ---------------------------------------------------------------------------------------------------------------------
 // Code
 // ---------------------------------------------------------------------------------------------------------------------
-public interface IToUnitOfWorkRepository;
+public interface IUnitOfWorkRepository;

src/CodeOfChaos.Types.UnitOfWork/UnitOfWork.cs

@@ -14,7 +14,7 @@ namespace CodeOfChaos.Types.UnitOfWork;
 public class UnitOfWork<TDbContext>(IDbContextFactory<TDbContext> dbContextFactory, IServiceScope serviceScope) : IUnitOfWork where TDbContext : DbContext{
     private readonly AsyncLazy<TDbContext> _db = new(async ct => await dbContextFactory.CreateDbContextAsync(ct));
     private IDbContextTransaction? _transaction;
-    private ConcurrentDictionary<Type, IToUnitOfWorkRepository> AttachedRepositories { get; } = [];
+    private ConcurrentDictionary<Type, IUnitOfWorkRepository> AttachedRepositories { get; } = [];
 
     // -----------------------------------------------------------------------------------------------------------------
     // Methods
@@ -84,8 +84,8 @@ public virtual async ValueTask<T> GetDbContextAsync<T>(CancellationToken ct = de
         return dbContext as T ?? throw new InvalidCastException($"Cannot cast DbContext of type '{dbContext.GetType()}' to '{typeof(T)}'");
     }
 
-    public virtual TRepo GetRepository<TRepo>() where TRepo : class, IToUnitOfWorkRepository {
-        if (AttachedRepositories.TryGetValue(typeof(TRepo), out IToUnitOfWorkRepository? cachedRepo) && cachedRepo is TRepo castedCachedRepo) return castedCachedRepo;
+    public virtual TRepo GetRepository<TRepo>() where TRepo : class, IUnitOfWorkRepository {
+        if (AttachedRepositories.TryGetValue(typeof(TRepo), out IUnitOfWorkRepository? cachedRepo) && cachedRepo is TRepo castedCachedRepo) return castedCachedRepo;
 
         // Cache miss so we create a new instance
         var repo = serviceScope.ServiceProvider.GetRequiredService<TRepo>();

tests/Tests.CodeOfChaos.Types.UnitOfWork/UnitOfWorkTests.cs

@@ -193,14 +193,14 @@ public async Task GetDbContextAsync_ShouldThrowForUnsupportedDbContextType() {
     [Test]
     public async Task GetRepository_ShouldRetrieveRepositoryFromServiceProvider() {
         // Arrange
-        var mockRepository = new Mock<IToUnitOfWorkRepository>();
+        var mockRepository = new Mock<IUnitOfWorkRepository>();
         _serviceProvider
-            .Setup(sp => sp.GetService(typeof(IToUnitOfWorkRepository)))
+            .Setup(sp => sp.GetService(typeof(IUnitOfWorkRepository)))
             .Returns(mockRepository.Object);
 
 
         // Act
-        var repository = _unitOfWork.GetRepository<IToUnitOfWorkRepository>();
+        var repository = _unitOfWork.GetRepository<IUnitOfWorkRepository>();
 
         // Assert
         await Assert.That(repository).IsNotNull();