fix: fixed related repos used by tests

nmshd · Dec 18, 2024 · 4c08578 · 4c08578
1 parent 4f35ca7
commit 4c08578
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/Modules/Devices/src/Devices.Infrastructure/Persistence/Repository/IdentitiesRepository.cs b/Modules/Devices/src/Devices.Infrastructure/Persistence/Repository/IdentitiesRepository.cs
@@ -16,14 +16,14 @@ namespace Backbone.Modules.Devices.Infrastructure.Persistence.Repository;
 
 public class IdentitiesRepository : IIdentitiesRepository
 {
-    private readonly DbSet<Identity> _identities;
-    private readonly IQueryable<Identity> _readonlyIdentities;
     private readonly DevicesDbContext _dbContext;
     private readonly DbSet<Device> _devices;
-    private readonly IQueryable<Device> _readonlyDevices;
-    private readonly UserManager<ApplicationUser> _userManager;
+    private readonly DbSet<Identity> _identities;
     private readonly DbSet<IdentityDeletionProcessAuditLogEntry> _identityDeletionProcessAuditLogs;
+    private readonly IQueryable<Device> _readonlyDevices;
+    private readonly IQueryable<Identity> _readonlyIdentities;
     private readonly IQueryable<IdentityDeletionProcessAuditLogEntry> _readonlyIdentityDeletionProcessAuditLogs;
+    private readonly UserManager<ApplicationUser> _userManager;
 
     public IdentitiesRepository(DevicesDbContext dbContext, UserManager<ApplicationUser> userManager)
     {
@@ -173,6 +173,7 @@ public async Task<IEnumerable<Identity>> Find(Expression<Func<Identity, bool>> f
     {
         return await (track ? _identities : _readonlyIdentities)
             .IncludeAll(_dbContext)
+            .AsSplitQuery()
             .Where(filter)
             .ToListAsync(cancellationToken);
     }

diff --git a/...essages/src/Messages.Infrastructure/Persistence/Database/Repository/MessagesRepository.cs b/...essages/src/Messages.Infrastructure/Persistence/Database/Repository/MessagesRepository.cs
@@ -13,9 +13,9 @@ namespace Backbone.Modules.Messages.Infrastructure.Persistence.Database.Reposito
 
 public class MessagesRepository : IMessagesRepository
 {
+    private readonly MessagesDbContext _dbContext;
     private readonly DbSet<Message> _messages;
     private readonly IQueryable<Message> _readOnlyMessages;
-    private readonly MessagesDbContext _dbContext;
 
     public MessagesRepository(MessagesDbContext dbContext)
     {
@@ -80,7 +80,9 @@ public async Task Update(Message message)
 
     public async Task<IEnumerable<Message>> Find(Expression<Func<Message, bool>> expression, CancellationToken cancellationToken)
     {
-        return await _messages.IncludeAll(_dbContext)
+        return await _messages
+            .IncludeAll(_dbContext)
+            .AsSplitQuery()
             .Where(expression)
             .ToListAsync(cancellationToken);
     }